SARepo 0.1.6__py3-none-any.whl → 0.1.7__py3-none-any.whl

SARepo/repo.py

@@ -1,4 +1,3 @@
-
 from typing import Generic, List, TypeVar, Type, Optional, Any, Protocol
 
 from SARepo.sa_repo import Spec
@@ -6,45 +5,70 @@ from .base import Page, PageRequest
 
 T = TypeVar("T")
 
+
 class CrudRepository(Protocol, Generic[T]):
     model: Type[T]
-    def getAll(self, limit: Optional[int] = None, *, include_deleted: bool = False) -> List[T]: ...
-    def get(self, id_: Any, *, include_deleted: bool = False) -> T: ...
-    def try_get(self, id_: Any, *, include_deleted: bool = False) -> Optional[T]: ...
+
+    def getAll(self,
+               limit: Optional[int] = None,
+               *,
+               include_deleted: bool = False,
+               order_by=None,
+               **filters) -> List[T]: ...
+
+    def get(self, id_: Any = None, *, include_deleted: bool = False, **filters) -> T: ...
+
+    def try_get(self, id_: Any = None, *, include_deleted: bool = False, **filters) -> Optional[T]: ...
+
     def add(self, entity: T) -> T: ...
+
     def update(self, entity: T) -> T: ...
+
     def remove(self, entity: Optional[T] = None, id: Optional[Any] = None) -> bool: ...
+
     def delete_by_id(self, id_: Any) -> bool: ...
-    def page(self, page: PageRequest, spec: Optional[Spec] = None, order_by=None, *, include_deleted: bool = False) -> Page[T]: ...
+
+    def page(self, page: PageRequest, spec: Optional[Spec] = None, order_by=None, *, include_deleted: bool = False) -> \
+    Page[T]: ...
+
     def get_all_by_column(
-    self,
-    column_name: str,
-    value: Any,
-    *,
-    limit: Optional[int] = None,
-    order_by=None,
-    include_deleted: bool = False,
-    **extra_filters
-) -> list[T]: ...
+            self,
+            column_name: str,
+            value: Any,
+            *,
+            limit: Optional[int] = None,
+            order_by=None,
+            include_deleted: bool = False,
+            **extra_filters
+    ) -> list[T]: ...
+
     def find_all_by_column(
-    self,
-    column_name: str,
-    value: Any,
-    *,
-    limit: Optional[int] = None,
-    order_by=None,
-    include_deleted: bool = False,
-    **extra_filters
-) -> list[T]: ...
+            self,
+            column_name: str,
+            value: Any,
+            *,
+            limit: Optional[int] = None,
+            order_by=None,
+            include_deleted: bool = False,
+            **extra_filters
+    ) -> list[T]: ...
+
     def get_or_create(
-        self,
-        defaults: Optional[dict] = None,
-        **unique_filters
+            self,
+            defaults: Optional[dict] = None,
+            **unique_filters
     ) -> tuple[T, bool]: ...
+
     def raw_query(self, sql: str, params: Optional[dict] = None) -> list[dict]: ...
+
     def aggregate_avg(self, column_name: str, **filters) -> Optional[float]: ...
+
     def aggregate_min(self, column_name: str, **filters): ...
+
     def aggregate_max(self, column_name: str, **filters): ...
-    def aggregate_sum(self, column_name: str, **filters): ... 
+
+    def aggregate_sum(self, column_name: str, **filters): ...
+
     def count(self, **filters) -> int: ...
-    def restore(self, id_: Any) -> bool: ...
+
+    def restore(self, id_: Any) -> bool: ...

SARepo/sa_repo.py

@@ -1,4 +1,3 @@
-
 from typing import List, Type, Generic, TypeVar, Optional, Sequence, Any, Callable
 from sqlalchemy.orm import Session
 from sqlalchemy.ext.asyncio import AsyncSession
@@ -8,8 +7,10 @@ from .base import PageRequest, Page, NotFoundError
 T = TypeVar("T")
 Spec = Callable
 
+
 class SARepository(Generic[T]):
     """Synchronous repository implementation for SQLAlchemy 2.x."""
+
     def __init__(self, model: Type[T], session: Session):
         self.model = model
         self.session = session
@@ -31,35 +32,52 @@ class SARepository(Generic[T]):
     def _has_soft_delete(self) -> bool:
         return hasattr(self.model, "is_deleted")
 
-    def _apply_alive_filter(self, stmt, include_deleted: bool) :
+    def _apply_alive_filter(self, stmt, include_deleted: bool):
         if self._has_soft_delete() and not include_deleted:
             stmt = stmt.where(self.model.is_deleted == False)  # noqa: E712
         return stmt
-    
-    def getAll(self, limit: Optional[int] = None, *, include_deleted: bool = False) -> List[T]:
+
+    def getAll(
+            self,
+            limit: Optional[int] = None,
+            *,
+            include_deleted: bool = False,
+            order_by=None,
+            **filters
+    ) -> List[T]:
         stmt = select(self.model)
+        stmt = self._apply_filters(stmt, **filters)
         stmt = self._apply_alive_filter(stmt, include_deleted)
+        if order_by is not None:
+            stmt = stmt.order_by(order_by)
         if limit is not None:
             stmt = stmt.limit(limit)
         result = self.session.execute(stmt)
-        return result.scalars().all()    
-    
-    def get(self, id_: Any, *, include_deleted: bool = False) -> T:
-        obj = self.session.get(self.model, id_)
-        if not obj:
-            raise NotFoundError(f"{self.model.__name__}({id_}) not found")
+        return result.scalars().all()
+
+    def get(self, id_: Any = None, *, include_deleted: bool = False, **filters) -> T:
+        if id_ is not None:
+            obj = self.session.get(self.model, id_)
+            if not obj:
+                raise NotFoundError(f"{self.model.__name__}({id_}) not found")
+        else:
+            stmt = select(self.model)
+            stmt = self._apply_filters(stmt, **filters)
+            stmt = self._apply_alive_filter(stmt, include_deleted)
+            res = self.session.execute(stmt)
+            obj = res.scalars().first()
+            if not obj:
+                raise NotFoundError(f"{self.model.__name__} not found for {filters}")
         if self._has_soft_delete() and not include_deleted and getattr(obj, "is_deleted", False):
-            raise NotFoundError(f"{self.model.__name__}({id_}) not found")  # скрываем как «нет»
+            raise NotFoundError(f"{self.model.__name__}({getattr(obj, 'id', '?')}) deleted")
         return obj
 
-    def try_get(self, id_: Any, *, include_deleted: bool = False) -> Optional[T]:
-        obj = self.session.get(self.model, id_)
-        if not obj:
-            return None
-        if self._has_soft_delete() and not include_deleted and getattr(obj, "is_deleted", False):
+    def try_get(self, id_: Any = None, *, include_deleted: bool = False, **filters) -> Optional[T]:
+        try:
+            return self.get(id_=id_, include_deleted=include_deleted, **filters)
+        except NotFoundError:
             return None
-        return obj
-    
+
     def add(self, entity: T) -> T:
         self.session.add(entity)
         self.session.flush()
@@ -74,10 +92,10 @@ class SARepository(Generic[T]):
     def remove(self, entity: Optional[T] = None, id: Optional[Any] = None) -> bool:
         if entity is None and id is None:
             raise ValueError("remove() requires either entity or id")
-    
+
         if id is not None:
             return self._delete_by_id(id)
-        
+
         insp = inspect(entity, raiseerr=False)
         if not (insp and (insp.persistent or insp.pending)):
             pk = getattr(entity, "id", None)
@@ -90,7 +108,7 @@ class SARepository(Generic[T]):
             setattr(entity, "is_deleted", True)
         else:
             self.session.delete(entity)
-            
+
         return True
 
     def _delete_by_id(self, id_: Any) -> bool:
@@ -100,7 +118,8 @@ class SARepository(Generic[T]):
         self.remove(obj)
         return True
 
-    def page(self, page: PageRequest, spec: Optional[Spec] = None, order_by=None, *, include_deleted: bool = False) -> Page[T]:
+    def page(self, page: PageRequest, spec: Optional[Spec] = None, order_by=None, *, include_deleted: bool = False) -> \
+    Page[T]:
         base = self._select()
         if spec:
             base = spec(base)
@@ -116,16 +135,16 @@ class SARepository(Generic[T]):
             base.offset(page.page * page.size).limit(page.size)
         ).scalars().all()
         return Page(items, total, page.page, page.size)
-    
+
     def get_all_by_column(
-        self,
-        column_name: str,
-        value: Any,
-        *,
-        limit: Optional[int] = None,
-        order_by=None,
-        include_deleted: bool = False,
-        **extra_filters
+            self,
+            column_name: str,
+            value: Any,
+            *,
+            limit: Optional[int] = None,
+            order_by=None,
+            include_deleted: bool = False,
+            **extra_filters
     ) -> list[T]:
         col = self._resolve_column(column_name)
         stmt = select(self.model).where(col == value)
@@ -137,15 +156,15 @@ class SARepository(Generic[T]):
             stmt = stmt.limit(limit)
         res = self.session.execute(stmt)
         return res.scalars().all()
-        
+
     # Alias 
     def find_all_by_column(self, *args, **kwargs):
         return self.get_all_by_column(*args, **kwargs)
-    
+
     def get_or_create(
-        self,
-        defaults: Optional[dict] = None,
-        **unique_filters
+            self,
+            defaults: Optional[dict] = None,
+            **unique_filters
     ) -> tuple[T, bool]:
         """
         Возвращает (obj, created). unique_filters определяют уникальность.
@@ -228,12 +247,14 @@ class SARepository(Generic[T]):
             return True
         return False
 
+
 class SAAsyncRepository(Generic[T]):
     """Async repository implementation for SQLAlchemy 2.x."""
+
     def __init__(self, model: Type[T], session: AsyncSession):
         self.model = model
         self.session = session
-    
+
     def _resolve_column(self, column_name: str):
         try:
             return getattr(self.model, column_name)
@@ -244,41 +265,71 @@ class SAAsyncRepository(Generic[T]):
         if filters:
             stmt = stmt.filter_by(**filters)
         return stmt
-    
+
     def _select(self):
         return select(self.model)
-    
+
     def _has_soft_delete(self) -> bool:
         return hasattr(self.model, "is_deleted")
 
-    def _apply_alive_filter(self, stmt, include_deleted: bool) :
+    def _apply_alive_filter(self, stmt, include_deleted: bool):
         if self._has_soft_delete() and not include_deleted:
             stmt = stmt.where(self.model.is_deleted == False)  # noqa: E712
         return stmt
 
-    async def getAll(self, limit: Optional[int] = None, *, include_deleted: bool = False) -> List[T]:
+    async def getAll(
+            self,
+            limit: Optional[int] = None,
+            *,
+            include_deleted: bool = False,
+            order_by=None,
+            **filters
+    ) -> List[T]:
+        """
+        Получить все записи с возможностью фильтрации (username='foo', age__gt=20, и т.д.)
+        """
         stmt = select(self.model)
+        stmt = self._apply_filters(stmt, **filters)
         stmt = self._apply_alive_filter(stmt, include_deleted)
+        if order_by is not None:
+            stmt = stmt.order_by(order_by)
         if limit is not None:
             stmt = stmt.limit(limit)
         result = await self.session.execute(stmt)
         return result.scalars().all()
 
-    async def get(self, id_: Any, *, include_deleted: bool = False) -> T:
-        obj = await self.session.get(self.model, id_)
-        if not obj:
-            raise NotFoundError(f"{self.model.__name__}({id_}) not found")
+    async def get(self, id_: Any = None, *, include_deleted: bool = False, **filters) -> T:
+        """
+        Получить один объект по id или по произвольным фильтрам.
+        Пример:
+            await repo.get(id_=1)
+            await repo.get(username='ibrahim')
+            await repo.get(username__ilike='%rah%')
+        """
+        if id_ is not None:
+            obj = await self.session.get(self.model, id_)
+            if not obj:
+                raise NotFoundError(f"{self.model.__name__}({id_}) not found")
+        else:
+            stmt = select(self.model)
+            stmt = self._apply_filters(stmt, **filters)
+            stmt = self._apply_alive_filter(stmt, include_deleted)
+            res = await self.session.execute(stmt)
+            obj = res.scalars().first()
+            if not obj:
+                raise NotFoundError(f"{self.model.__name__} not found for {filters}")
         if self._has_soft_delete() and not include_deleted and getattr(obj, "is_deleted", False):
-            raise NotFoundError(f"{self.model.__name__}({id_}) not found")
+            raise NotFoundError(f"{self.model.__name__}({getattr(obj, 'id', '?')}) deleted")
         return obj
 
-    async def try_get(self, id_: Any, *, include_deleted: bool = False) -> Optional[T]:
-        obj = await self.session.get(self.model, id_)
-        if not obj:
-            return None
-        if self._has_soft_delete() and not include_deleted and getattr(obj, "is_deleted", False):
+    async def try_get(self, id_: Any = None, *, include_deleted: bool = False, **filters) -> Optional[T]:
+        """
+        Как get(), но не выбрасывает исключение при отсутствии объекта.
+        """
+        try:
+            return await self.get(id_=id_, include_deleted=include_deleted, **filters)
+        except NotFoundError:
             return None
-        return obj
 
     async def add(self, entity: T) -> T:
         self.session.add(entity)
@@ -294,10 +345,10 @@ class SAAsyncRepository(Generic[T]):
     async def remove(self, entity: Optional[T] = None, id: Optional[Any] = None) -> bool:
         if entity is None and id is None:
             raise ValueError("remove() requires either entity or id")
-    
+
         if id is not None:
             return await self._delete_by_id(id)
-        
+
         insp = inspect(entity, raiseerr=False)
         if not (insp and (insp.persistent or insp.pending)):
             pk = getattr(entity, "id", None)
@@ -310,10 +361,9 @@ class SAAsyncRepository(Generic[T]):
             setattr(entity, "is_deleted", True)
         else:
             await self.session.delete(entity)
-            
+
         return True
 
-    
     async def _delete_by_id(self, id_: Any) -> bool:
         obj = await self.session.get(self.model, id_)
         if not obj:
@@ -321,7 +371,8 @@ class SAAsyncRepository(Generic[T]):
         await self.remove(obj)
         return True
 
-    async def page(self, page: PageRequest, spec: Optional[Spec] = None, order_by=None, *, include_deleted: bool = False) -> Page[T]: # type: ignore
+    async def page(self, page: PageRequest, spec: Optional[Spec] = None, order_by=None, *,
+                   include_deleted: bool = False) -> Page[T]:  # type: ignore
         base = self._select()
         if spec:
             base = spec(base)
@@ -340,14 +391,14 @@ class SAAsyncRepository(Generic[T]):
         return Page(items, total, page.page, page.size)
 
     async def get_all_by_column(
-        self,
-        column_name: str,
-        value: Any,
-        *,
-        limit: Optional[int] = None,
-        order_by=None,
-        include_deleted: bool = False,
-        **extra_filters
+            self,
+            column_name: str,
+            value: Any,
+            *,
+            limit: Optional[int] = None,
+            order_by=None,
+            include_deleted: bool = False,
+            **extra_filters
     ) -> list[T]:
         col = self._resolve_column(column_name)
         stmt = select(self.model).where(col == value)
@@ -363,11 +414,11 @@ class SAAsyncRepository(Generic[T]):
     # Alias
     async def find_all_by_column(self, *args, **kwargs):
         return await self.get_all_by_column(*args, **kwargs)
-    
+
     async def get_or_create(
-        self,
-        defaults: Optional[dict] = None,
-        **unique_filters
+            self,
+            defaults: Optional[dict] = None,
+            **unique_filters
     ) -> tuple[T, bool]:
         stmt = select(self.model).filter_by(**unique_filters)
         if hasattr(self.model, "is_deleted"):
@@ -437,4 +488,4 @@ class SAAsyncRepository(Generic[T]):
             setattr(obj, "is_deleted", False)
             await self.session.flush()
             return True
-        return False
+        return False

{sarepo-0.1.6.dist-info → sarepo-0.1.7.dist-info}/METADATA

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

sarepo-0.1.7.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=2MsR21GFMYstXRimFVqShC9foG7KmAuzTFtqsjK3a-Q,2129
+SARepo/sa_repo.py,sha256=fUae3B1aDxN3E-oKuthi0joWnrpCAOaUFHCi9H3be8c,19052
+SARepo/specs.py,sha256=e-X1cCkva3e71M37hcfbjNDMezZAaOkkRaPToUzhEeU,687
+SARepo/uow.py,sha256=yPlvi7PoH9x2pLNt6hEin9zT5qG9axUKn_TkZcIuz1s,859
+sarepo-0.1.7.dist-info/licenses/LICENSE,sha256=px90NOQCrnZ77qoFT8IvAiNS1CXQ2OU27uVBKbki9DM,131
+sarepo-0.1.7.dist-info/METADATA,sha256=b3pMl6G85wJ8qN1LtB5dc-mGwj0vmQXWBaL66x2WazY,2691
+sarepo-0.1.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+sarepo-0.1.7.dist-info/top_level.txt,sha256=0k952UYVZLIIv3kZzFlxI0yzBMusZo3-XCQtxms_kS0,7
+sarepo-0.1.7.dist-info/RECORD,,

sarepo-0.1.6.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=q99K9odCVuxJ9HIKwgVpu3p9-hyzxmGlE-wkiCp9tbg,1820
-SARepo/sa_repo.py,sha256=iPJan94ALJ3N58aU1D6rgO6uhqLDUxcHXvORpQrAt9c,17258
-SARepo/specs.py,sha256=e-X1cCkva3e71M37hcfbjNDMezZAaOkkRaPToUzhEeU,687
-SARepo/uow.py,sha256=yPlvi7PoH9x2pLNt6hEin9zT5qG9axUKn_TkZcIuz1s,859
-sarepo-0.1.6.dist-info/licenses/LICENSE,sha256=px90NOQCrnZ77qoFT8IvAiNS1CXQ2OU27uVBKbki9DM,131
-sarepo-0.1.6.dist-info/METADATA,sha256=4Yfl0Eh0h8Sb2Lrs_DxZRfPJuZzGTg7-I9blRPva9aw,2691
-sarepo-0.1.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-sarepo-0.1.6.dist-info/top_level.txt,sha256=0k952UYVZLIIv3kZzFlxI0yzBMusZo3-XCQtxms_kS0,7
-sarepo-0.1.6.dist-info/RECORD,,