From c758059fe431905ef7a14d78ed18a7060de3c594 Mon Sep 17 00:00:00 2001
From: CaptainB <bin@fit2cloud.com>
Date: Wed, 11 Jun 2025 15:07:38 +0800
Subject: [PATCH] feat: add migration to insert default data for ToolFolder and
 KnowledgeFolder

---
 apps/application/migrations/0001_initial.py |  4 ++--
 apps/knowledge/migrations/0001_initial.py   |  8 ++++++++
 apps/tools/migrations/0001_initial.py       | 12 ++++++++++++
 3 files changed, 22 insertions(+), 2 deletions(-)

diff --git a/apps/application/migrations/0001_initial.py b/apps/application/migrations/0001_initial.py
index 52dadda82..bd35e0d31 100644
--- a/apps/application/migrations/0001_initial.py
+++ b/apps/application/migrations/0001_initial.py
@@ -12,8 +12,8 @@ from django.db import migrations, models
 
 def insert_default_data(apps, schema_editor):
     # 创建一个根模块（没有父节点）
-    QuerySet(application.models.application.ApplicationFolder).create(id='root', name='根目录',
-                                                                      user_id='f0dd8f71-e4ee-11ee-8c84-a8a1595801ab')
+    QuerySet(application.models.application.ApplicationFolder).create(id='default', name='根目录',
+                                                                      user_id='f0dd8f71-e4ee-11ee-8c84-a8a1595801ab', workspace_id='default')
 
 
 class Migration(migrations.Migration):
diff --git a/apps/knowledge/migrations/0001_initial.py b/apps/knowledge/migrations/0001_initial.py
index fcac6c11b..8f3d0dd71 100644
--- a/apps/knowledge/migrations/0001_initial.py
+++ b/apps/knowledge/migrations/0001_initial.py
@@ -7,6 +7,13 @@ import mptt.fields
 import uuid_utils.compat
 from django.db import migrations, models
 
+from knowledge.models import KnowledgeFolder
+
+
+def insert_default_data(apps, schema_editor):
+    # 创建一个根模块（没有父节点）
+    KnowledgeFolder.objects.create(id='default', name='根目录', user_id='f0dd8f71-e4ee-11ee-8c84-a8a1595801ab', workspace_id='default')
+
 
 class Migration(migrations.Migration):
 
@@ -168,4 +175,5 @@ class Migration(migrations.Migration):
                 'db_table': 'problem_paragraph_mapping',
             },
         ),
+        migrations.RunPython(insert_default_data),
     ]
diff --git a/apps/tools/migrations/0001_initial.py b/apps/tools/migrations/0001_initial.py
index da547af9e..6b629180d 100644
--- a/apps/tools/migrations/0001_initial.py
+++ b/apps/tools/migrations/0001_initial.py
@@ -1,10 +1,20 @@
 # Generated by Django 5.2 on 2025-05-15 04:04
+import os
 
 import django.db.models.deletion
 import mptt.fields
 import uuid_utils.compat
 from django.db import migrations, models
 
+from common.utils.common import get_file_content
+from maxkb.conf import PROJECT_DIR
+from tools.models import ToolFolder
+
+
+def insert_default_data(apps, schema_editor):
+    # 创建一个根模块（没有父节点）
+    ToolFolder.objects.create(id='default', name='根目录', user_id='f0dd8f71-e4ee-11ee-8c84-a8a1595801ab', workspace_id='default')
+
 
 class Migration(migrations.Migration):
 
@@ -35,6 +45,7 @@ class Migration(migrations.Migration):
                 'db_table': 'tool_folder',
             },
         ),
+        migrations.RunPython(insert_default_data),
         migrations.CreateModel(
             name='Tool',
             fields=[
@@ -60,4 +71,5 @@ class Migration(migrations.Migration):
                 'db_table': 'tool',
             },
         ),
+        migrations.RunSQL(get_file_content(os.path.join(PROJECT_DIR, "apps", "tools", 'migrations', 'internal_tool.sql')))
     ]