Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
22 changes: 16 additions & 6 deletions apps/system_manage/migrations/0005_resourcemapping.py
Original file line number Diff line number Diff line change
@@ -1,13 +1,14 @@
# Generated by Django 5.2.8 on 2025-12-19 09:37
from concurrent.futures import ThreadPoolExecutor

import uuid_utils.compat
from django.db import migrations, models
from django.db.models import QuerySet

from knowledge.models import Knowledge


def resource_mapping(apps, schema_editor):
from system_manage.models.resource_mapping import ResourceMapping
def get_initialization_resource_mapping():
from django.db.models import QuerySet
from application.flow.tools import get_workflow_resource, get_node_handle_callback, \
get_instance_resource
Expand All @@ -28,7 +29,7 @@ def resource_mapping(apps, schema_editor):
resource_mapping_list += workflow_mapping
resource_mapping_list += instance_mapping
except:
continue
pass
knowledge_ids = list(Knowledge.objects.values_list('id', flat=True))
for knowledge_id in knowledge_ids:
try:
Expand All @@ -46,8 +47,17 @@ def resource_mapping(apps, schema_editor):

resource_mapping_list += instance_mapping
except:
continue
QuerySet(ResourceMapping).bulk_create(resource_mapping_list)
pass

return resource_mapping_list


def resource_mapping(apps, schema_editor):
from system_manage.models.resource_mapping import ResourceMapping
with ThreadPoolExecutor(max_workers=3) as executor:
future = executor.submit(get_initialization_resource_mapping)
resource_mapping_list = future.result()
QuerySet(ResourceMapping).bulk_create(resource_mapping_list)


class Migration(migrations.Migration):
Expand Down Expand Up @@ -79,5 +89,5 @@ class Migration(migrations.Migration):
'db_table': 'resource_mapping',
},
),
migrations.RunPython(resource_mapping)
migrations.RunPython(resource_mapping, atomic=False)
]
Loading