orm wip

Author: mkrd

dictdatabase/orm.py

@@ -3,6 +3,7 @@
 from typing import get_type_hints, get_origin, get_args, TypeVar, Tuple, Type
 from types import UnionType, NoneType
 from . import io_safe
+from . sessions import SessionFileFull, SessionFileKey, SessionFileWhere, SessionDirFull, SessionDirWhere
 
 
 
@@ -67,40 +68,46 @@ class FileDictModel(ABC):
 	__item_model__: T = None
 
 	@classmethod
-	def get_by_key(cls, key) -> T:
+	def get_at_key(cls, key) -> T:
 		"""
 			Gets an item by key.
 			The data is partially read from the __file__.
 		"""
 		return cls.__item_model__.partial_read_by_key(key)
 
+	@classmethod
+	def session_at_key(cls, key):
+		return cls.__item_model__.session(key)
+
 	@classmethod
 	def items(cls) -> list[Tuple[str, T]]:
 		"""
-			Gets all items.
+			Gets all items as a list of tuples (key, ORM model of value).
 		"""
 		data = io_safe.read(cls.__file__)
-		return [cls.__item_model__(key, value) for key, value in data.items()]
+		return [(k, cls.__item_model__(k, v)) for k, v in data.items()]
 
 	@classmethod
 	def session(cls):
-		...
-
-
-	@classmethod
-	def session_at_key(cls, key):
-		...
-
-
+		"""
+		Enter a session with the file as (session, data) where data is a dict of
+		<key>: <ORM model of value> pairs.
+		"""
+		def make_session_obj_from_dict(data):
+			sess_obj = {}
+			for k, v in data.items():
+				sess_obj[k] = cls.__item_model__.from_key_value(k, v)
+			return sess_obj
+		return SessionFileFull(cls.__file__, make_session_obj_from_dict)
 
 
 	@classmethod
-	def get_where(cls, where: callable) -> list[T]:
+	def get_where(cls, where: callable) -> list[Tuple[str, T]]:
 		"""
 			Gets all items where the where function returns True.
 			The where function takes an object of type __item_model__.
 		"""
-		return [item for item in cls.get_all() if where(item)]
+		return [(k, v) for k, v in cls.items() if where(v)]
 
 	# Not Implemented:
 	# - select by filter callback (file_where): Not implemented because no performance advantage.
@@ -121,6 +128,11 @@ def read_by_key(cls, key) -> T:
 		data = io_safe.partial_read(cls.__file__, key)
 		return cls.from_key_value(key, data)
 
+	@classmethod
+	def session(cls, key):
+		def partial_func(x):
+			return cls.from_key_value(key, x)
+		return SessionFileKey(cls.__file__, key, partial_func)
 
 
 
@@ -131,7 +143,6 @@ def from_dict(cls, data) -> DictModel:
 		obj = cls()
 		return fill_object_from_dict_using_type_hints(obj, cls, data)
 
-
 	def to_dict(self) -> dict:
 		res = {}
 		for var_name in get_type_hints(self).keys():