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

rb_commons/configs/injections.py

@@ -1,4 +1,6 @@
 from typing import Annotated
+
+from rb_commons.permissions import IsAdmin, IsCustomer
 from rb_commons.schemes.jwt import Claims
 from fastapi import Request, Depends
 
@@ -6,4 +8,6 @@ from fastapi import Request, Depends
 async def get_claims(request: Request) -> Claims:
     return Claims.from_headers(dict(request.headers))
 
-ClaimsDep = Annotated[Claims, Depends(get_claims)]
+ClaimsDep = Annotated[Claims, Depends(get_claims)]
+IsAdminDep = Annotated[IsAdmin, Depends(IsAdmin())]
+IsCustomerDep = Annotated[IsCustomer, Depends(IsCustomer())]

rb_commons/permissions/__init__.py

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

rb_commons/permissions/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/permissions/role_permissions.py

@@ -0,0 +1,24 @@
+from fastapi import Depends
+
+from rb_commons.configs.injections import get_claims
+from rb_commons.permissions.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 ForbiddenException(message=f"Access denied", status=401, code="0000")
+
+    def has_permission(self, claims: Claims) -> bool:
+        return False
+
+
+class IsAdmin(BasePermission):
+    def has_permission(self, claims: Claims) -> bool:
+        return claims.user_role == UserRole.ADMIN
+
+
+class IsCustomer(BasePermission):
+    def has_permission(self, claims: Claims) -> bool:
+        return claims.user_role == UserRole.CUSTOMER

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: rb-commons
-Version: 0.1.15
+Version: 0.1.17
 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.15.dist-info → rb_commons-0.1.17.dist-info}/RECORD

@@ -1,13 +1,16 @@
 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=6B1EOgIGnkWv3UrFaV9PRgG0-CJAbLu1UZ3kq-SjPVU,273
+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.15.dist-info/METADATA,sha256=jRHiJGnvpJQg8LHPrL05VpeZMCY0fZ-FQbE7EDmi9r0,6533
-rb_commons-0.1.15.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
-rb_commons-0.1.15.dist-info/top_level.txt,sha256=HPx_WAYo3_fbg1WCeGHsz3wPGio1ucbnrlm2lmqlJog,11
-rb_commons-0.1.15.dist-info/RECORD,,
+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,,