25 lines
982 B
Python
25 lines
982 B
Python
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)
|