|
26 | 26 | from common.utils.split_model import group_by |
27 | 27 | from knowledge.models import Knowledge |
28 | 28 | from maxkb.conf import PROJECT_DIR |
| 29 | +from maxkb.settings import edition |
29 | 30 | from models_provider.models import Model |
30 | 31 | from system_manage.models import WorkspaceUserResourcePermission, AuthTargetType |
31 | 32 | from tools.models import Tool |
@@ -109,13 +110,14 @@ def auth_resource(self, resource_id: str): |
109 | 110 | user_id = self.data.get('user_id') |
110 | 111 | wurp = QuerySet(WorkspaceUserResourcePermission).filter(auth_target_type=auth_target_type, |
111 | 112 | workspace_id=workspace_id).first() |
112 | | - auth_type = wurp.auth_type if wurp else ResourceAuthType.RESOURCE_PERMISSION_GROUP |
| 113 | + auth_type = wurp.auth_type if wurp else ( |
| 114 | + ResourceAuthType.RESOURCE_PERMISSION_GROUP if edition == 'CE' else ResourceAuthType.ROLE) |
113 | 115 | # 自动授权给创建者 |
114 | 116 | WorkspaceUserResourcePermission( |
115 | 117 | target=resource_id, |
116 | 118 | auth_target_type=auth_target_type, |
117 | 119 | permission_list=[ResourcePermission.VIEW, |
118 | | - ResourcePermission.MANAGE] if auth_type == ResourceAuthType.RESOURCE_PERMISSION_GROUP else [ |
| 120 | + ResourcePermission.MANAGE] if auth_type == ResourceAuthType.RESOURCE_PERMISSION_GROUP else [ |
119 | 121 | ResourcePermissionRole.ROLE], |
120 | 122 | workspace_id=workspace_id, |
121 | 123 | user_id=user_id, |
|
0 commit comments