rb-commons 0.1.7__tar.gz → 0.1.8__tar.gz

{rb_commons-0.1.7 → rb_commons-0.1.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: rb-commons
-Version: 0.1.7
+Version: 0.1.8
 Summary: Commons of project and simplified orm based on sqlalchemy.
 Home-page: https://github.com/Abdulvoris101/rb-commons
 Author: Abdulvoris

rb_commons-0.1.8/rb_commons/schemes/jwt.py

@@ -0,0 +1,53 @@
+import uuid
+from enum import Enum
+from typing import Optional
+
+from pydantic import BaseModel, Field, ConfigDict
+
+
+class UserRole(str, Enum):
+    ADMIN = "admin"
+    CUSTOMER = "customer"
+    GUEST = "guest"
+
+class Claims(BaseModel):
+    model_config = ConfigDict(extra="ignore")
+
+    class Claims(BaseModel):
+        model_config = ConfigDict(extra="ignore")
+
+        user_id: Optional[int] = Field(None, alias="x-user-id")
+        user_role: UserRole = Field(UserRole.GUEST, alias="x-user-role")
+        shop_id: Optional[uuid.UUID] = Field(None, alias="x-shop-id")
+
+        @classmethod
+        def from_headers(cls, headers: dict) -> 'Claims':
+            raw_claims = {
+                "x-user-id": headers.get("x-user-id"),
+                "x-user-role": headers.get("x-user-role"),
+                "x-shop-id": headers.get("x-shop-id")
+            }
+
+            try:
+                if raw_claims["x-user-id"]:
+                    try:
+                        raw_claims["x-user-id"] = int(raw_claims["x-user-id"])
+                    except ValueError as e:
+                        raise ValueError(f"Invalid user_id format: {e}")
+
+                if raw_claims["x-shop-id"]:
+                    try:
+                        raw_claims["x-shop-id"] = uuid.UUID(raw_claims["x-shop-id"])
+                    except ValueError as e:
+                        raise ValueError(f"Invalid shop_id format: {e}")
+
+                if raw_claims["x-user-role"]:
+                    try:
+                        UserRole(raw_claims["x-user-role"].lower())
+                    except ValueError as e:
+                        raise ValueError(f"Invalid user_role: {e}")
+
+                return cls(**raw_claims)
+
+            except Exception as e:
+                raise ValueError(f"Failed to parse claims: {str(e)}")

{rb_commons-0.1.7 → rb_commons-0.1.8}/rb_commons.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: rb-commons
-Version: 0.1.7
+Version: 0.1.8
 Summary: Commons of project and simplified orm based on sqlalchemy.
 Home-page: https://github.com/Abdulvoris101/rb-commons
 Author: Abdulvoris

{rb_commons-0.1.7 → rb_commons-0.1.8}/setup.py

@@ -5,7 +5,7 @@ with open("README.md", "r", encoding="utf-8") as fh:
 
 setup(
     name="rb-commons",
-    version="0.1.7",
+    version="0.1.8",
     author="Abdulvoris",
     author_email="erkinovabdulvoris101@gmail.com",
     description="Commons of project and simplified orm based on sqlalchemy.",

rb_commons-0.1.7/rb_commons/schemes/jwt.py

@@ -1,16 +0,0 @@
-import uuid
-from enum import Enum
-from pydantic import BaseModel, Field, ConfigDict
-
-
-class UserRole(str, Enum):
-    ADMIN = "admin"
-    CUSTOMER = "customer"
-    GUEST = "guest"
-
-class Claims(BaseModel):
-    model_config = ConfigDict(extra="ignore")
-
-    user_id: str = Field(None, alias="x-user-id")
-    user_role: UserRole = Field(UserRole.GUEST, alias="x-user-role")
-    shop_id: uuid.UUID = Field(None, alias="x-shop-id")