SARepo 0.1.0__py3-none-any.whl → 0.1.3__py3-none-any.whl

SARepo/repo.py

@@ -1,14 +1,16 @@
 
-from typing import Generic, TypeVar, Type, Optional, Any, Protocol
+from typing import Generic, List, TypeVar, Type, Optional, Any, Protocol
 from .base import Page, PageRequest
 
 T = TypeVar("T")
 
 class CrudRepository(Protocol, Generic[T]):
     model: Type[T]
+    def getAll(self, limit: Optional[int]) -> List[T]: ...
     def get(self, id_: Any) -> T: ...
     def try_get(self, id_: Any) -> Optional[T]: ...
     def add(self, entity: T) -> T: ...
     def update(self, entity: T) -> T: ...
     def remove(self, entity: T) -> None: ...
+    def delete_by_id(self, id_: Any) -> bool: ...
     def page(self, page: PageRequest, spec=None, order_by=None) -> Page[T]: ...

SARepo/sa_repo.py

@@ -1,8 +1,8 @@
 
-from typing import Type, Generic, TypeVar, Optional, Sequence, Any, Callable
+from typing import List, Type, Generic, TypeVar, Optional, Sequence, Any, Callable
 from sqlalchemy.orm import Session
 from sqlalchemy.ext.asyncio import AsyncSession
-from sqlalchemy import select, func
+from sqlalchemy import inspect, select, func
 from .base import PageRequest, Page, NotFoundError
 
 T = TypeVar("T")
@@ -17,6 +17,13 @@ class SARepository(Generic[T]):
     def _select(self):
         return select(self.model)
 
+    def getAll(self, limit: Optional[int] = None) -> List[T]:
+        stmt = select(self.model)
+        if limit is not None:
+            stmt = stmt.limit(limit)
+        result = self.session.execute(stmt)
+        return result.scalars().all()
+    
     def get(self, id_: Any) -> T:
         obj = self.session.get(self.model, id_)
         if not obj:
@@ -38,11 +45,26 @@ class SARepository(Generic[T]):
         return entity
 
     def remove(self, entity: T) -> None:
+        insp = inspect(entity, raiseerr=False)
+        if not (insp and (insp.persistent or insp.pending)):
+            pk = getattr(entity, "id", None)
+            if pk is None:
+                raise ValueError("remove() needs a persistent entity or an entity with a primary key set")
+            entity = self.session.get(self.model, pk)
+            if entity is None:
+                return
         if hasattr(entity, "is_deleted"):
             setattr(entity, "is_deleted", True)
         else:
             self.session.delete(entity)
 
+    def delete_by_id(self, id_: Any) -> bool:
+        obj = self.session.get(self.model, id_)
+        if not obj:
+            return False
+        self.remove(obj)
+        return True
+
     def page(self, page: PageRequest, spec: Optional[Spec] = None, order_by=None) -> Page[T]:
         stmt = self._select()
         if spec:
@@ -62,6 +84,13 @@ class SAAsyncRepository(Generic[T]):
     def __init__(self, model: Type[T], session: AsyncSession):
         self.model = model
         self.session = session
+        
+    async def getAll(self, limit: Optional[int] = None) -> List[T]:
+        stmt = select(self.model)
+        if limit is not None:
+            stmt = stmt.limit(limit)
+        result = await self.session.execute(stmt)
+        return result.scalars().all()
 
     def _select(self):
         return select(self.model)
@@ -87,11 +116,26 @@ class SAAsyncRepository(Generic[T]):
         return entity
 
     async def remove(self, entity: T) -> None:
+        insp = inspect(entity, raiseerr=False)
+        if not (insp and (insp.persistent or insp.pending)):
+            pk = getattr(entity, "id", None)
+            if pk is None:
+                raise ValueError("remove() needs a persistent entity or an entity with a primary key set")
+            entity = await self.session.get(self.model, pk)
+            if entity is None:
+                return
         if hasattr(entity, "is_deleted"):
             setattr(entity, "is_deleted", True)
         else:
             await self.session.delete(entity)
 
+    async def delete_by_id(self, id_: Any) -> bool:
+        obj = await self.session.get(self.model, id_)
+        if not obj:
+            return False
+        await self.remove(obj)
+        return True
+
     async def page(self, page: PageRequest, spec: Optional[Spec] = None, order_by=None) -> Page[T]:
         stmt = self._select()
         if spec:

{sarepo-0.1.0.dist-info → sarepo-0.1.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: SARepo
-Version: 0.1.0
+Version: 0.1.3
 Summary: Minimal, explicit Repository & Unit-of-Work layer over SQLAlchemy 2.x
 Author: nurbergenovv
 License: MIT

sarepo-0.1.3.dist-info/RECORD

@@ -0,0 +1,12 @@
+SARepo/__init__.py,sha256=tNYbuUDloC1qVnXq7uomS3jRcaPUvy0VEZiMdYR6x1M,183
+SARepo/base.py,sha256=UbAdZ9WYh_o93mrCVL3D8Q0tY_8mvm0HspO_L5m0GTQ,874
+SARepo/models.py,sha256=ypSmbKAijvNH4WjSeJBgyNT8mKa_e_7-I5kiNisjGAI,570
+SARepo/repo.py,sha256=5Af4SLfbKOMJdX7MOev4OcGIGQ0HzN6QDUktURjasyU,597
+SARepo/sa_repo.py,sha256=feuwBHAEdW_STJ6fHxYlSO0KvSruoqiB_v1sTCRy6JE,5332
+SARepo/specs.py,sha256=e-X1cCkva3e71M37hcfbjNDMezZAaOkkRaPToUzhEeU,687
+SARepo/uow.py,sha256=yPlvi7PoH9x2pLNt6hEin9zT5qG9axUKn_TkZcIuz1s,859
+sarepo-0.1.3.dist-info/licenses/LICENSE,sha256=px90NOQCrnZ77qoFT8IvAiNS1CXQ2OU27uVBKbki9DM,131
+sarepo-0.1.3.dist-info/METADATA,sha256=FXf9N-SsghCIQGucviE8GVVxh1viF0BV0FhKkMeg2HQ,2691
+sarepo-0.1.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+sarepo-0.1.3.dist-info/top_level.txt,sha256=0k952UYVZLIIv3kZzFlxI0yzBMusZo3-XCQtxms_kS0,7
+sarepo-0.1.3.dist-info/RECORD,,

sarepo-0.1.0.dist-info/RECORD

@@ -1,12 +0,0 @@
-SARepo/__init__.py,sha256=tNYbuUDloC1qVnXq7uomS3jRcaPUvy0VEZiMdYR6x1M,183
-SARepo/base.py,sha256=UbAdZ9WYh_o93mrCVL3D8Q0tY_8mvm0HspO_L5m0GTQ,874
-SARepo/models.py,sha256=ypSmbKAijvNH4WjSeJBgyNT8mKa_e_7-I5kiNisjGAI,570
-SARepo/repo.py,sha256=vCVQmjYifzlO_ggOFwj3Xdyqt4_7ILDoUtHUAFnFyZU,482
-SARepo/sa_repo.py,sha256=mdTe3RyrGklG3z50jr4pOv4aidWxOZ_6dtdLgr83Osk,3617
-SARepo/specs.py,sha256=e-X1cCkva3e71M37hcfbjNDMezZAaOkkRaPToUzhEeU,687
-SARepo/uow.py,sha256=yPlvi7PoH9x2pLNt6hEin9zT5qG9axUKn_TkZcIuz1s,859
-sarepo-0.1.0.dist-info/licenses/LICENSE,sha256=px90NOQCrnZ77qoFT8IvAiNS1CXQ2OU27uVBKbki9DM,131
-sarepo-0.1.0.dist-info/METADATA,sha256=lHz7KBK2U2CAvoxwUB1ytK9d1LeNoiV0ODlvM7BFjgE,2691
-sarepo-0.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-sarepo-0.1.0.dist-info/top_level.txt,sha256=0k952UYVZLIIv3kZzFlxI0yzBMusZo3-XCQtxms_kS0,7
-sarepo-0.1.0.dist-info/RECORD,,