rb-commons 0.1.17__py3-none-any.whl → 0.1.19__py3-none-any.whl

rb_commons/configs/injections.py

@@ -1,6 +1,6 @@
 from typing import Annotated
 
-from rb_commons.permissions import IsAdmin, IsCustomer
+from rb_commons.permissions.role_permissions import IsAdmin, IsCustomer
 from rb_commons.schemes.jwt import Claims
 from fastapi import Request, Depends
 

rb_commons/orm/managers.py

@@ -1,8 +1,11 @@
-from typing import TypeVar, Type, Generic, Optional, List, Dict
+import uuid
+from typing import TypeVar, Type, Generic, Optional, List, Dict, Literal, Union
 from sqlalchemy import select, delete, update
 from sqlalchemy.exc import IntegrityError, SQLAlchemyError, NoResultFound
 from sqlalchemy.ext.asyncio import AsyncSession
 from sqlalchemy.orm import declarative_base
+
+from rb_commons.http.exceptions import NotFoundException
 from rb_commons.orm.exceptions import DatabaseException, InternalException
 
 ModelType = TypeVar('ModelType', bound=declarative_base())
@@ -12,27 +15,55 @@ class BaseManager(Generic[ModelType]):
 
     def __init__(self, session: AsyncSession) -> None:
         self.session = session
+        self.data = None
+        self.filters = {}
 
-    async def get(self, **kwargs) -> Optional[ModelType]:
+    async def get(self, pk: Union[str, int, uuid.UUID]) -> Optional[ModelType]:
         """
            get object based on conditions
        """
-        query = select(self.model).filter_by(**kwargs)
+        query = select(self.model).filter_by(id=pk)
         result = await self.session.execute(query)
-        return result.scalar_one_or_none()
+        instance = result.scalar_one_or_none()
+
+        if instance is None:
+            raise NotFoundException(
+                message="Object does not exist",
+                status=404,
+                code="0001",
+            )
+
+        return instance
 
-    async def filter(self, **kwargs) -> List[ModelType]:
+    def filter(self, **kwargs) -> 'BaseManager[ModelType]':
         """
            Filter objects based on conditions
        """
-        query = select(self.model).filter_by(**kwargs)
+        self.filters.update(kwargs)
+        return self
+
+    async def all(self) -> List[ModelType]:
+        """Return all filtered results."""
+        query = select(self.model).filter_by(**self.filters)
         result = await self.session.execute(query)
         return list(result.scalars().all())
 
+    async def first(self) -> Optional[ModelType]:
+        """Return the first matching object, or None."""
+        query = select(self.model).filter_by(**self.filters)
+        result = await self.session.execute(query)
+        return result.scalars().first()
+
+    async def count(self) -> int:
+        """Return the count of matching records."""
+        query = select(self.model).filter_by(**self.filters)
+        result = await self.session.execute(query)
+        return len(result.scalars().all())
+
     async def create(self, **kwargs) -> ModelType:
         """
                Create a new object
-       """
+        """
         obj = self.model(**kwargs)
 
         try:
@@ -52,7 +83,7 @@ class BaseManager(Generic[ModelType]):
             raise InternalException(f"Unexpected error during creation: {str(e)}") from e
 
 
-    async def delete(self, id: Optional[int] = None, **filters) -> bool | None:
+    async def delete(self, id: Optional[int] = None, **filters):
         """
         Delete object(s) with flexible filtering options
 

rb_commons/permissions/__init__.py

@@ -1,2 +0,0 @@
-from role_permissions import *
-from http_exceptions import *

rb_commons/permissions/role_permissions.py

@@ -1,7 +1,7 @@
 from fastapi import Depends
 
 from rb_commons.configs.injections import get_claims
-from rb_commons.permissions.http_exceptions import ForbiddenException
+from rb_commons.http.exceptions import ForbiddenException
 from rb_commons.schemes.jwt import Claims, UserRole
 
 

{rb_commons-0.1.17.dist-info → rb_commons-0.1.19.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: rb-commons
-Version: 0.1.17
+Version: 0.1.19
 Summary: Commons of project and simplified orm based on sqlalchemy.
 Home-page: https://github.com/RoboSell-organization/rb-commons
 Author: Abdulvoris

rb_commons-0.1.19.dist-info/RECORD

@@ -0,0 +1,17 @@
+rb_commons/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+rb_commons/configs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+rb_commons/configs/config.py,sha256=tpqC1V9PXx88m0N5L13WqkoalPUk6SthjxgL_lmf-lE,1439
+rb_commons/configs/injections.py,sha256=LcwYQWwmAKApID0xnpt9IUorpFXBo679b-ODrG19jvA,465
+rb_commons/http/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+rb_commons/http/exceptions.py,sha256=EGRMr1cRgiJ9Q2tkfANbf0c6-zzXf1CD6J3cmCaT_FA,1885
+rb_commons/orm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+rb_commons/orm/exceptions.py,sha256=1aMctiEwrPjyehoXVX1l6ML5ZOhmDkmBISzlTD5ey1Y,509
+rb_commons/orm/managers.py,sha256=2RJyiW4uUYtgE_BzOS6Z3-TdHgXE0PPcFuQYt4Ze3MQ,7755
+rb_commons/permissions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+rb_commons/permissions/role_permissions.py,sha256=XOv81nZpX8Iwj9A0eLfirPfwsha7TeVcnUiEDTS_Qfw,790
+rb_commons/schemes/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+rb_commons/schemes/jwt.py,sha256=F66JJDhholuOPPzlKeoC6f1TL4gXg4oRUrV5yheNpyo,1675
+rb_commons-0.1.19.dist-info/METADATA,sha256=KP2FaMIl2eaVVohkEKHA6arUyHCwV5jLfDnxT-FSu9Q,6533
+rb_commons-0.1.19.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
+rb_commons-0.1.19.dist-info/top_level.txt,sha256=HPx_WAYo3_fbg1WCeGHsz3wPGio1ucbnrlm2lmqlJog,11
+rb_commons-0.1.19.dist-info/RECORD,,

rb_commons-0.1.17.dist-info/RECORD

@@ -1,16 +0,0 @@
-rb_commons/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-rb_commons/configs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-rb_commons/configs/config.py,sha256=tpqC1V9PXx88m0N5L13WqkoalPUk6SthjxgL_lmf-lE,1439
-rb_commons/configs/injections.py,sha256=jOnabKVNj490nYqGuqw5jeSFFvq1lM8vfmx-R1NJobc,448
-rb_commons/orm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-rb_commons/orm/exceptions.py,sha256=1aMctiEwrPjyehoXVX1l6ML5ZOhmDkmBISzlTD5ey1Y,509
-rb_commons/orm/managers.py,sha256=s0MeGfuDmQA287IV57s8kk467AJwskAoG4HvgQxKTYc,6711
-rb_commons/permissions/__init__.py,sha256=Gd1yBLBTBph8NdHSA_-a-wN5ys4Uol647cKCwBFcsRY,61
-rb_commons/permissions/http_exceptions.py,sha256=EGRMr1cRgiJ9Q2tkfANbf0c6-zzXf1CD6J3cmCaT_FA,1885
-rb_commons/permissions/role_permissions.py,sha256=8oD2bwvugW-tV07J-T6EELjDeJqYNqnIYrgSKg7cgbk,802
-rb_commons/schemes/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-rb_commons/schemes/jwt.py,sha256=F66JJDhholuOPPzlKeoC6f1TL4gXg4oRUrV5yheNpyo,1675
-rb_commons-0.1.17.dist-info/METADATA,sha256=zAw74i9HyF57I2mWMpDHgxxfIu__w_OEu77qzv4T6kA,6533
-rb_commons-0.1.17.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
-rb_commons-0.1.17.dist-info/top_level.txt,sha256=HPx_WAYo3_fbg1WCeGHsz3wPGio1ucbnrlm2lmqlJog,11
-rb_commons-0.1.17.dist-info/RECORD,,