sharedkernel 2.9.1__py3-none-any.whl → 2.9.3__py3-none-any.whl

sharedkernel/enum/user_role.py

@@ -0,0 +1,16 @@
+from enum import Enum
+
+
+class Role(str, Enum):
+    ADMIN = "admin"
+    MANAGER = "manager"
+    EDITOR = "editor"
+    VIEWER = "viewer"
+    OPERATOR = "operator"
+
+class RoleFa(str, Enum):
+    ADMIN = "ادمین"
+    MANAGER = "مدیر"
+    EDITOR = "ویرایشگر"
+    VIEWER = "بیننده"
+    OPERATOR = "اپراتور"

sharedkernel/objects/user_workspace.py

@@ -0,0 +1,13 @@
+from sharedkernel.objects.base_document import BaseDocument
+from pydantic import BaseModel, RootModel
+from sharedkernel.enum.user_role import Role
+
+
+class UserWorkspace(BaseDocument):
+    user_id: str
+    workspace_id: str | None = None
+    is_owner: bool | None = None
+    role: Role = Role.ADMIN
+
+class UserWorkspaces(RootModel[list[UserWorkspace]]):
+    pass

sharedkernel/toolbox_service.py

@@ -1,16 +1,21 @@
 from typing import Type
 from pydantic import BaseModel
 import aiohttp
+from typing import List
 
 from sharedkernel.objects.result import Result
 from sharedkernel.enum import FeatureKey, PlanType
 from sharedkernel.enum import ErrorCode
 from sharedkernel.objects.subscription import Subscription
+from sharedkernel.objects.user_workspace import UserWorkspace
 from sharedkernel.exception.exception import BusinessException
+from sharedkernel.database.cache.cache_repository import CacheRepository
+from sharedkernel.enum.user_role import Role
+from sharedkernel.objects.user_info import current_user_info
 
 
 class ToolboxService:
-    def __init__(self, cache):
+    def __init__(self, cache: CacheRepository | None = None):
         self.cache = cache
 
     async def send_request(
@@ -32,6 +37,10 @@ class ToolboxService:
                 headers=headers,
             ) as response:
 
+                if not response.ok:
+                    print(f"API Request to {endpoint} Failed: {response.status} {response.text}")
+                    raise BusinessException(error_code=ErrorCode.Internal_Server)
+
                 response_data = await response.json()
                 return Result(
                     **response_data,
@@ -67,4 +76,11 @@ class ToolboxService:
             if _feature.feature_key == feature:
                 if _feature.remaining - quantity < 0:
                     raise BusinessException(error_code=feature_errors[feature])
-                break
+                break
+
+    def check_access_level(self, allowed_roles: List[Role], user_workspace: UserWorkspace):
+        if user_workspace is None:
+            raise BusinessException(ErrorCode.Workspace_Access_Denied)
+
+        if user_workspace.role not in allowed_roles and user_workspace.role != Role.ADMIN:
+            raise BusinessException(ErrorCode.User_Access_Denied)

{sharedkernel-2.9.1.dist-info → sharedkernel-2.9.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sharedkernel
-Version: 2.9.1
+Version: 2.9.3
 Summary: sharekernel is a shared package between all python projects
 Author: Smilinno
 Description-Content-Type: text/markdown
@@ -32,8 +32,10 @@ Dynamic: summary
 this is a shared kernel package
 
 # Change Log
+### Version 2.9.3
+- add token to user info
 ### Version 2.9.1
-- Add token to user info
+- add check_access_level to toolbox
 ### Version 2.9.0
 - Add toolbox_service
 ### Version 2.8.0

{sharedkernel-2.9.1.dist-info → sharedkernel-2.9.3.dist-info}/RECORD

@@ -10,7 +10,7 @@ sharedkernel/multipart_upload.py,sha256=JVlCBlznB9dWh2_spjAqzLOqQT1CHUTvrR4m7ug8
 sharedkernel/regex_masking.py,sha256=zQrgteP8Cuq1EC9B7QUJqAXUxK9ISD9kWMYK2AbRfw0,3288
 sharedkernel/s3_uploader.py,sha256=VWgN-RVHmLXMDuxgZux3M-iFwWk5zhRnUIECKS0auW4,3637
 sharedkernel/string_extentions.py,sha256=ld02W06gd0Ql80GQU6nlqPAeUSfOe2Yr8cCzf3lJgQY,98
-sharedkernel/toolbox_service.py,sha256=l4nPm7DiXCudeQd-JRYyS-T52g7q_YCsTJ9Sz9iDKAY,2440
+sharedkernel/toolbox_service.py,sha256=-vIJfKiAMzEcLPrBTqpErygZ56LaC1Edxdza6HyoMdo,3326
 sharedkernel/chunker/chunk_rule.py,sha256=djxeQ3UvMx3_suH-heAQB0VwINT7uMfV7rnuuJChmwU,2531
 sharedkernel/chunker/text_chunker.py,sha256=QL8rQjJX-7U9TXMWiQBPldb9h1EeEECmFwLF1JLoGAM,3146
 sharedkernel/database/__init__.py,sha256=AtIbU7pDKwY6YCg_J8PX62WcYld1lAvBcVFW62MdiFg,241
@@ -32,6 +32,7 @@ sharedkernel/enum/plan_interval.py,sha256=Fn_E6NyswYa00OETE4E8wV8o4gNE7WfP6tsDxY
 sharedkernel/enum/plan_type.py,sha256=ExcPl8wKTbRu1gYB7jZqweaj-K1F8HTXlCuxEOIvl3g,139
 sharedkernel/enum/redis_mode_enum.py,sha256=v_n9g7_rq_pdMiZrC8mMveRcjgF3eBv3K2tsaV1Pidc,136
 sharedkernel/enum/sort_order.py,sha256=T_1AknuVg2OSJMlIDkEiCUNHPW0yDnFvbzoNpTgG3-I,97
+sharedkernel/enum/user_role.py,sha256=CnHCMHqgjsSNsldV8v6DNDWwoLKnHbnCeZE80dkX2rA,329
 sharedkernel/exception/__init__.py,sha256=Yjkd1VsLFuzfWcMj-v8YYTA-vjtUlF4ESglBFeN1zvs,315
 sharedkernel/exception/exception.py,sha256=TOxnymWJfhsrMmuLsG-amqJlDxmzznEfRnqDBe1vizk,1016
 sharedkernel/exception/exception_handlers.py,sha256=VHEYaf1quaf63w_JCsNWFbMs2TxqmQqqlL4yGt1OjXI,3302
@@ -52,7 +53,8 @@ sharedkernel/objects/jwt_model.py,sha256=XQHQhTbg7PT8XiUh5fd9MwRH4ldPsesI_hfbjaS
 sharedkernel/objects/result.py,sha256=I_9hX5TPEO1oStzuFLjFh1rtimXorz7ml-OaW_2BMvc,680
 sharedkernel/objects/subscription.py,sha256=9MT2Vu-cf8ozO4pwh00ovk6aRNMjbJboKP8NkJgeSRI,1043
 sharedkernel/objects/user_info.py,sha256=A6b0M1aQJLnhtEIkgnPpo_mWJ7v6ZqafT7tOhRGrExA,315
-sharedkernel-2.9.1.dist-info/METADATA,sha256=PlJBgQAA8pdzim5c0bs4hCWTVcjhjE1tLk110DCFHFU,4238
-sharedkernel-2.9.1.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
-sharedkernel-2.9.1.dist-info/top_level.txt,sha256=TVTOnV1MItSSlpSjqkiijuHkoVsGHS4CArpsM-lylkE,13
-sharedkernel-2.9.1.dist-info/RECORD,,
+sharedkernel/objects/user_workspace.py,sha256=npmkAdfLQkPhkINbSAGfmnSa279I-YjgtBR7zboXTds,362
+sharedkernel-2.9.3.dist-info/METADATA,sha256=gnoPEJ4EXFVpno4TJKP2rF_Cyqz1878iGXWKcQCP2VA,4292
+sharedkernel-2.9.3.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+sharedkernel-2.9.3.dist-info/top_level.txt,sha256=TVTOnV1MItSSlpSjqkiijuHkoVsGHS4CArpsM-lylkE,13
+sharedkernel-2.9.3.dist-info/RECORD,,