chore(api): migrate import for db

api/controllers/console/files.py

@@ -20,9 +20,9 @@ from controllers.console.wraps import (
     cloud_edition_billing_resource_check,
     setup_required,
 )
+from extensions.ext_database import db
 from fields.file_fields import file_fields, upload_config_fields
 from libs.login import login_required
-from models import db
 from services.file_service import FileService
 
 PREVIEW_WORDS_LIMIT = 3000

api/controllers/console/remote_files.py

@@ -14,8 +14,8 @@ from controllers.common.errors import (
 )
 from core.file import helpers as file_helpers
 from core.helper import ssrf_proxy
+from extensions.ext_database import db
 from fields.file_fields import file_fields_with_signed_url, remote_file_info_fields
-from models import db
 from models.account import Account
 from services.file_service import FileService
 

api/controllers/files/image_preview.py

@@ -7,7 +7,7 @@ from werkzeug.exceptions import NotFound
 import services
 from controllers.common.errors import UnsupportedFileTypeError
 from controllers.files import files_ns
-from models import db
+from extensions.ext_database import db
 from services.account_service import TenantService
 from services.file_service import FileService
 

api/controllers/service_api/app/file.py

@@ -12,8 +12,9 @@ from controllers.common.errors import (
 )
 from controllers.service_api import service_api_ns
 from controllers.service_api.wraps import FetchUserArg, WhereisUserArg, validate_app_token
+from extensions.ext_database import db
 from fields.file_fields import build_file_model
-from models import App, EndUser, db
+from models import App, EndUser
 from services.file_service import FileService
 
 

api/controllers/web/files.py

@@ -11,8 +11,8 @@ from controllers.common.errors import (
 )
 from controllers.web import web_ns
 from controllers.web.wraps import WebApiResource
+from extensions.ext_database import db
 from fields.file_fields import build_file_model
-from models import db
 from services.file_service import FileService
 
 

api/controllers/web/remote_files.py

@@ -14,8 +14,8 @@ from controllers.web import web_ns
 from controllers.web.wraps import WebApiResource
 from core.file import helpers as file_helpers
 from core.helper import ssrf_proxy
+from extensions.ext_database import db
 from fields.file_fields import build_file_with_signed_url_model, build_remote_file_info_model
-from models import db
 from services.file_service import FileService
 
 

api/tests/integration_tests/tasks/test_remove_app_and_related_data_task.py

@@ -5,7 +5,8 @@ import pytest
 from sqlalchemy import delete
 
 from core.variables.segments import StringSegment
-from models import Tenant, db
+from extensions.ext_database import db
+from models import Tenant
 from models.enums import CreatorUserRole
 from models.model import App, UploadFile
 from models.workflow import WorkflowDraftVariable, WorkflowDraftVariableFile
@@ -259,6 +260,7 @@ class TestDeleteDraftVariablesWithOffloadIntegration:
 
         # Create WorkflowDraftVariableFile records
         from core.variables.types import SegmentType
+
         var_file1 = WorkflowDraftVariableFile(
             tenant_id=tenant.id,
             app_id=app.id,

api/tests/integration_tests/test_offload.py

@@ -3,9 +3,9 @@ import uuid
 import pytest
 from sqlalchemy.orm import Session, joinedload, selectinload
 
+from extensions.ext_database import db
 from libs.datetime_utils import naive_utc_now
 from libs.uuid_utils import uuidv7
-from models import db
 from models.enums import CreatorUserRole
 from models.model import UploadFile
 from models.workflow import WorkflowNodeExecutionModel, WorkflowNodeExecutionOffload, WorkflowNodeExecutionTriggeredFrom