rb-commons 0.1.16__py3-none-any.whl → 0.1.18__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/http/exceptions.py

@@ -0,0 +1,41 @@
+from fastapi import HTTPException, status
+
+
+class CustomHTTPException(HTTPException):
+    def __init__(self, message: str, status: int, code: str = None, additional_info: dict = None):
+        super().__init__(status_code=status, detail=message)
+
+        self.code = code
+        self.message = message
+        self.status = status
+        self.additional_info = additional_info
+
+
+class NotAuthorizedException(CustomHTTPException):
+    def __init__(self, message: str = "You are not authorized", status: int = 401, code: str = None,
+                 additional_info: dict = None):
+        super().__init__(message=message, status=status, code=code, additional_info=additional_info)
+
+
+class ForbiddenException(CustomHTTPException):
+    def __init__(self, message: str = "Permission denied", status: int = 403, code: str = None,
+                 additional_info: dict = None):
+        super().__init__(message=message, status=status, code=code, additional_info=additional_info)
+
+
+class BadRequestException(CustomHTTPException):
+    def __init__(self, message: str = "Bad request", status: int = 400, code: str = None,
+                 additional_info: dict = None):
+        super().__init__(message=message, status=status, code=code, additional_info=additional_info)
+
+
+class NotFoundException(CustomHTTPException):
+    def __init__(self, message: str = "Not found", status: int = 404, code: str = None,
+                 additional_info: dict = None):
+        super().__init__(message=message, status=status, code=code, additional_info=additional_info)
+
+
+class InternalException(CustomHTTPException):
+    def __init__(self, message: str = "Internal exception", status: int = 500, code: str = None,
+                 additional_info: dict = None):
+        super().__init__(message=message, status=status, code=code, additional_info=additional_info)

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]:
+    async 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 +0,0 @@
-from role_permissions import *

rb_commons/permissions/role_permissions.py

@@ -1,13 +1,14 @@
-from fastapi import Depends, HTTPException
+from fastapi import Depends
 
 from rb_commons.configs.injections import get_claims
+from rb_commons.http.exceptions import ForbiddenException
 from rb_commons.schemes.jwt import Claims, UserRole
 
 
 class BasePermission:
     def __call__(self, claims: Claims = Depends(get_claims)):
         if not self.has_permission(claims):
-            raise HTTPException(status_code=403, detail="Permission Denied")
+            raise ForbiddenException(message=f"Access denied", status=401, code="0000")
 
     def has_permission(self, claims: Claims) -> bool:
         return False

{rb_commons-0.1.16.dist-info → rb_commons-0.1.18.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: rb-commons
-Version: 0.1.16
+Version: 0.1.18
 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.18.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=RDWDOJ0wGXbGErDi2clzJF_KDF0lPMZK_j0KQy_AMhg,7761
+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.18.dist-info/METADATA,sha256=W415WB-6LHzNtSL0xYVeuWs-DCA1ZIU74HWzSCtVma0,6533
+rb_commons-0.1.18.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
+rb_commons-0.1.18.dist-info/top_level.txt,sha256=HPx_WAYo3_fbg1WCeGHsz3wPGio1ucbnrlm2lmqlJog,11
+rb_commons-0.1.18.dist-info/RECORD,,

rb_commons-0.1.16.dist-info/RECORD

@@ -1,15 +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=2AWODOWWC_p7ThC6kilzSFm4vZbekgk3Tk8hFOHzyTM,30
-rb_commons/permissions/role_permissions.py,sha256=SSLToBcz6OLGBo2Jw9L8BcfZlGsTy_kB1VRrVTizKy0,735
-rb_commons/schemes/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-rb_commons/schemes/jwt.py,sha256=F66JJDhholuOPPzlKeoC6f1TL4gXg4oRUrV5yheNpyo,1675
-rb_commons-0.1.16.dist-info/METADATA,sha256=k_eJ4IUYbN5y9wz4FC0uZDwtZ4DbQwsAofGLQO84wJI,6533
-rb_commons-0.1.16.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
-rb_commons-0.1.16.dist-info/top_level.txt,sha256=HPx_WAYo3_fbg1WCeGHsz3wPGio1ucbnrlm2lmqlJog,11
-rb_commons-0.1.16.dist-info/RECORD,,