Skip to content

Commit 36b63d7

Browse files
committed
fix: Migration
1 parent 66684dd commit 36b63d7

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

apps/system_manage/migrations/0003_alter_workspaceuserresourcepermission_target.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
# Generated by Django 5.2.6 on 2025-10-11 02:54
2+
from concurrent.futures import ThreadPoolExecutor
23
from functools import reduce
34

45
from django.db import migrations, models
@@ -8,6 +9,13 @@
89
from common.utils.common import group_by
910

1011

12+
def workspace_user_role_mapping_model_exists(workspace_user_role_mapping_model):
13+
try:
14+
QuerySet(workspace_user_role_mapping_model).first()
15+
except Exception as e:
16+
return False
17+
return False
18+
1119
def delete_auth(apps,folder_model):
1220
workspace_user_resource_permission_model = apps.get_model('system_manage', 'WorkspaceUserResourcePermission')
1321
QuerySet(workspace_user_resource_permission_model).filter(target__in=QuerySet(folder_model).values_list('id')).delete()
@@ -35,6 +43,13 @@ def auth_folder(apps, schema_editor):
3543
workspace_user_resource_permission_model = apps.get_model('system_manage', 'WorkspaceUserResourcePermission')
3644

3745
workspace_user_role_mapping_model = DatabaseModelManage.get_model("workspace_user_role_mapping")
46+
47+
with ThreadPoolExecutor(max_workers=3) as executor:
48+
future = executor.submit(workspace_user_role_mapping_model_exists, workspace_user_role_mapping_model)
49+
exists = future.result()
50+
if not exists:
51+
workspace_user_role_mapping_model = None
52+
3853
if workspace_user_role_mapping_model is None:
3954
workspace_user_role_mapping_model_workspace_dict = {
4055
'default': [WorkspaceUserRoleMapping('default', '', u.id) for u in QuerySet(user_model).all()]}

0 commit comments

Comments
 (0)