first

utils/get_user_workspace.py

@@ -0,0 +1,24 @@
+from apps.workspace.models import Workspace as WorkspaceModel
+from apps.cuser.models import CuserRole as CuserRoleModel
+from apps.customer.models import CustomerResponsible as CustomerResponsibleModel, Customer as CustomerModel
+
+def get_user_workspace(request) -> WorkspaceModel:
+    jwt_payload = request.auth.payload
+    cuser_role_uuid = jwt_payload.get('cuser_role_uuid')
+    # use_redis
+    cuser_role = CuserRoleModel.objects.get(uuid=cuser_role_uuid)
+    role_type_name = cuser_role.role_type.role_name_en
+
+    if role_type_name == 'workspace_operator':
+        operator = CustomerResponsibleModel.objects.get(cuser_role=cuser_role)
+        workspace = operator.sys_ws
+        return workspace
+    else:
+        customer = CustomerModel.objects.get(cuser_role=cuser_role)
+        return customer.sys_ws
+
+
+def get_user_workspace_uuid(request) -> str:
+    jwt_payload = request.auth.payload
+    workspace_uuid = jwt_payload.get('workspace_uuid')
+    return str(workspace_uuid)