expose ExpressionField, use __slots__ for several class

beanie/__init__.py

@@ -26,6 +26,7 @@
     BackLink,
     BeanieObjectId,
     DeleteRules,
+    ExpressionField,
     Indexed,
     Link,
     PydanticObjectId,
@@ -74,6 +75,7 @@
     "BackLink",
     "WriteRules",
     "DeleteRules",
+    "ExpressionField",
     # Custom Types
     "DecimalAnnotation",
     "BsonBinary",

beanie/odm/bulk.py

@@ -43,6 +43,8 @@ class Config:
 
 
 class BulkWriter:
+    __slots__ = ("operations", "session", "ordered")
+
     def __init__(
         self,
         session: Optional[AsyncIOMotorClientSession] = None,

beanie/odm/cache.py

@@ -14,6 +14,8 @@ class CachedItem(BaseModel):
 
 
 class LRUCache:
+    __slots__ = ("capacity", "expiration_time", "cache")
+
     def __init__(self, capacity: int, expiration_time: timedelta):
         self.capacity: int = capacity
         self.expiration_time: timedelta = expiration_time

beanie/odm/fields.py

@@ -29,15 +29,7 @@
 from pymongo import ASCENDING, IndexModel
 
 from beanie.odm.enums import SortDirection
-from beanie.odm.operators.find.comparison import (
-    GT,
-    GTE,
-    LT,
-    LTE,
-    NE,
-    Eq,
-    In,
-)
+from beanie.odm.operators.find.comparison import GT, GTE, LT, LTE, NE, Eq, In
 from beanie.odm.registry import DocsRegistry
 from beanie.odm.utils.parsing import parse_obj
 from beanie.odm.utils.pydantic import (
@@ -138,6 +130,8 @@ class PydanticObjectId(ObjectId):
     Object Id field. Compatible with Pydantic.
     """
 
+    __slots__ = ()
+
     @classmethod
     def __get_validators__(cls):
         yield cls.validate
@@ -206,6 +200,8 @@ def __modify_schema__(cls, field_schema):
 
 
 class ExpressionField(str):
+    __slots__ = ()
+
     def __getitem__(self, item):
         """
         Get sub field
@@ -511,6 +507,8 @@ def to_dict(self):
 
 
 class IndexModelField:
+    __slots__ = ("index", "name", "fields", "options")
+
     def __init__(self, index: IndexModel):
         self.index = index
         self.name = index.document["name"]