use slots to save memory

Author: mkrd

dictdatabase/configuration.py

@@ -3,7 +3,7 @@
 
 
 
-@dataclass
+@dataclass(slots=True)
 class Confuguration:
 	storage_directory: str = "ddb_storage"
 	indent: int | str | None = "\t"  # eg. "\t" or 4 or None

dictdatabase/indexing.py

@@ -38,6 +38,8 @@ class Indexer:
 		- value_hash: The hash of the value bytes
 	"""
 
+	__slots__ = ("data", "path")
+
 	def __init__(self, db_name: str):
 		# Make path of index file
 		db_name = db_name.replace("/", "___")

dictdatabase/io_unsafe.py

@@ -7,7 +7,7 @@
 from . import config, utils, byte_codes, indexing, io_bytes
 
 
-@dataclass(frozen=True)
+@dataclass(frozen=True, slots=True)
 class PartialDict:
 	prefix: bytes
 	key: str
@@ -17,7 +17,7 @@ class PartialDict:
 	suffix: bytes
 
 
-@dataclass(frozen=True)
+@dataclass(frozen=True, slots=True)
 class PartialFileHandle:
 	db_name: str
 	partial_dict: PartialDict

dictdatabase/locking.py

@@ -79,11 +79,15 @@ class AbstractLock:
 		An abstract lock doesn't do anything by itself. A subclass of it needs to
 		call super().__init__(...) and then only exit __init__ when the lock is aquired.
 	"""
+
+	__slots__ = ("id", "time_ns", "db_name", "need_path", "path", "ddb_dir")
+
 	id: str
 	time_ns: int
 	db_name: str
-	need_path: str = None
-	path: str = None
+	need_path: str
+	path: str
+	ddb_dir: str
 
 	def __init__(self, db_name: str):
 		"""
@@ -101,8 +105,7 @@ def _lock(self):
 	def _unlock(self):
 		for p in ("need_path", "path"):
 			try:
-				path = getattr(self, p, None)
-				if path:
+				if path := getattr(self, p, None):
 					os.unlink(path)
 			except FileNotFoundError:
 				pass

dictdatabase/models.py

@@ -81,9 +81,13 @@ def at(*path, key: str = None, where: Callable[[Any, Any], bool] = None) -> DDBM
 
 
 class DDBMethodChooser:
+
+	__slots__ = ("path", "key", "where", "op_type")
+
 	path: str
 	key: str
 	where: Callable[[Any, Any], bool]
+	op_type: OperationType
 
 
 	def __init__(self, path: tuple, key: str = None, where: Callable[[Any, Any], bool] = None):

dictdatabase/session.py

@@ -18,10 +18,18 @@ class DDBSession(Generic[T]):
 		the changes will be lost after exiting the with statement.
 	"""
 
-	in_session: bool = False
+	__slots__ = ("in_session", "db_name", "as_type", "where", "key", "op_type", "data_handle", "write_lock", "partial_handle", "original_data")
+
+	in_session: bool
+	db_name: str
 	as_type: T
+	where: Callable[[Any, Any], bool]
+	key: str
+	op_type: Any
+
 
 	def __init__(self, db_name: str, op_type, key: str = None, where: Callable[[Any, Any], bool] = None, as_type: T = None):
+		self.in_session = False
 		self.db_name = db_name
 		self.as_type = as_type
 		self.where = where