rb-commons 0.5.17__py3-none-any.whl → 0.5.18__py3-none-any.whl

rb_commons/permissions/role_permissions.py

@@ -23,7 +23,8 @@ class IsAdmin(BasePermission):
 
 class IsCustomer(BasePermission):
     def has_permission(self, claims: Claims) -> bool:
-        return claims.user_role == UserRole.CUSTOMER and claims.user_id is not None and claims.shop_id is not None
+        return claims.user_role == UserRole.CUSTOMER and claims.user_id is not None and claims.shop_id is not None \
+        and claims.customer_id is not None
 
 
 IsAdminDep = Annotated[Claims, Depends(IsAdmin())]

rb_commons/schemes/jwt.py

@@ -14,6 +14,7 @@ class Claims(BaseModel):
     model_config = ConfigDict(extra="ignore")
 
     user_id: Optional[int] = Field(None, alias="x-user-id")
+    customer_id: Optional[int] = Field(None, alias="x-customer-id")
     user_role: UserRole = Field(UserRole.GUEST, alias="x-user-role")
     shop_id: Optional[uuid.UUID] = Field(None, alias="x-shop-id")
     jwt_token: Optional[str] = Field(None, alias="x-jwt-token")
@@ -22,6 +23,7 @@ class Claims(BaseModel):
     def from_headers(cls, headers: dict) -> 'Claims':
         raw_claims = {
             "x-user-id": headers.get("x-user-id"),
+            "x-customer-id": headers.get("x-customer-id"),
             "x-user-role": headers.get("x-user-role", "admin"),
             "x-shop-id": headers.get("x-shop-id"),
             "x-jwt-token": headers.get("x-jwt-token")
@@ -34,6 +36,12 @@ class Claims(BaseModel):
                 except ValueError as e:
                     raise ValueError(f"Invalid user_id format: {e}")
 
+            if raw_claims["x-customer-id"]:
+                try:
+                    raw_claims["x-customer-id"] = int(raw_claims["x-customer-id"])
+                except ValueError as e:
+                    raise ValueError(f"Invalid customer_id format: {e}")
+
             if raw_claims["x-shop-id"]:
                 try:
                     raw_claims["x-shop-id"] = uuid.UUID(raw_claims["x-shop-id"])

{rb_commons-0.5.17.dist-info → rb_commons-0.5.18.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rb-commons
-Version: 0.5.17
+Version: 0.5.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.5.17.dist-info → rb_commons-0.5.18.dist-info}/RECORD

@@ -16,13 +16,13 @@ rb_commons/orm/exceptions.py,sha256=1aMctiEwrPjyehoXVX1l6ML5ZOhmDkmBISzlTD5ey1Y,
 rb_commons/orm/managers.py,sha256=VMu3v-JzSVZ9pN2d4ZcAF6qZyqmBeXZ62kiHUY73v20,17163
 rb_commons/orm/services.py,sha256=71eRcJ4TxZvzNz-hLXo12X4U7PGK54ZfbLAb27AjZi8,1589
 rb_commons/permissions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-rb_commons/permissions/role_permissions.py,sha256=kDvAUvUPt5iUV6ZrdeVPYNJjv4a_Gau7VR8oC_GaaGU,1050
+rb_commons/permissions/role_permissions.py,sha256=4dV89z6ggzLqCCiFYlMp7kQVJRESu6MHpkT5ZNjLo6A,1096
 rb_commons/schemes/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-rb_commons/schemes/jwt.py,sha256=4MLw3J1-LiU2CXIgm0kP1mlGoDmNfMXmj9iav2s-1Iw,2056
+rb_commons/schemes/jwt.py,sha256=5J-VTAgRWnRrOU6cuumc8eDJQDyTm5y8cn-kLSLan_g,2453
 rb_commons/schemes/pagination.py,sha256=8VZW1wZGJIPR9jEBUgppZUoB4uqP8ORudHkMwvEJSxg,1866
 rb_commons/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 rb_commons/utils/media.py,sha256=J2Zi0J28DhcVQVzt-myNNVuzj9Msaetul53VjZtdDdc,820
-rb_commons-0.5.17.dist-info/METADATA,sha256=fE2fDxCw2vdzr3-1VWvsrya3Nnw6sp2RAdXco07y3LM,6571
-rb_commons-0.5.17.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
-rb_commons-0.5.17.dist-info/top_level.txt,sha256=HPx_WAYo3_fbg1WCeGHsz3wPGio1ucbnrlm2lmqlJog,11
-rb_commons-0.5.17.dist-info/RECORD,,
+rb_commons-0.5.18.dist-info/METADATA,sha256=K4LUgXImE93aU2C4Sl9t0ttSj-WP3TLsTPV1hl9Ziws,6571
+rb_commons-0.5.18.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
+rb_commons-0.5.18.dist-info/top_level.txt,sha256=HPx_WAYo3_fbg1WCeGHsz3wPGio1ucbnrlm2lmqlJog,11
+rb_commons-0.5.18.dist-info/RECORD,,