From 7c29c9830b46f6f852fe975745be2506a88fe7de Mon Sep 17 00:00:00 2001
From: SZCJW <792430652@qq.com>
Date: Sat, 7 Mar 2026 03:14:16 +0800
Subject: [PATCH] x

---
 scripts/migrate_multi_upload.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/scripts/migrate_multi_upload.py b/scripts/migrate_multi_upload.py
index 73bd628..071a9c6 100644
--- a/scripts/migrate_multi_upload.py
+++ b/scripts/migrate_multi_upload.py
@@ -11,13 +11,16 @@
 """
 import asyncio
 import sys
+import os
 from pathlib import Path
 
-sys.path.insert(0, str(Path(__file__).parent.parent))
+project_root = Path(__file__).parent.parent
+sys.path.insert(0, str(project_root))
+sys.path.insert(0, str(project_root / "src"))
 
 from sqlalchemy import text
-from database.database import async_engine, get_db_session
-from utils.logger import get_logger
+from src.database.database import async_engine
+from src.utils.logger import get_logger
 
 logger = get_logger(__name__)