瀏覽代碼

chore(api): migrate import for db

tags/2.0.0-beta.1
QuantumGhost 2 月之前
父節點
當前提交
63c035d8a2

+ 1
- 1
api/controllers/console/files.py 查看文件

cloud_edition_billing_resource_check, cloud_edition_billing_resource_check,
setup_required, setup_required,
) )
from extensions.ext_database import db
from fields.file_fields import file_fields, upload_config_fields from fields.file_fields import file_fields, upload_config_fields
from libs.login import login_required from libs.login import login_required
from models import db
from services.file_service import FileService from services.file_service import FileService


PREVIEW_WORDS_LIMIT = 3000 PREVIEW_WORDS_LIMIT = 3000

+ 1
- 1
api/controllers/console/remote_files.py 查看文件

) )
from core.file import helpers as file_helpers from core.file import helpers as file_helpers
from core.helper import ssrf_proxy 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 fields.file_fields import file_fields_with_signed_url, remote_file_info_fields
from models import db
from models.account import Account from models.account import Account
from services.file_service import FileService from services.file_service import FileService



+ 1
- 1
api/controllers/files/image_preview.py 查看文件

import services import services
from controllers.common.errors import UnsupportedFileTypeError from controllers.common.errors import UnsupportedFileTypeError
from controllers.files import files_ns from controllers.files import files_ns
from models import db
from extensions.ext_database import db
from services.account_service import TenantService from services.account_service import TenantService
from services.file_service import FileService from services.file_service import FileService



+ 2
- 1
api/controllers/service_api/app/file.py 查看文件

) )
from controllers.service_api import service_api_ns from controllers.service_api import service_api_ns
from controllers.service_api.wraps import FetchUserArg, WhereisUserArg, validate_app_token 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 fields.file_fields import build_file_model
from models import App, EndUser, db
from models import App, EndUser
from services.file_service import FileService from services.file_service import FileService





+ 1
- 1
api/controllers/web/files.py 查看文件

) )
from controllers.web import web_ns from controllers.web import web_ns
from controllers.web.wraps import WebApiResource from controllers.web.wraps import WebApiResource
from extensions.ext_database import db
from fields.file_fields import build_file_model from fields.file_fields import build_file_model
from models import db
from services.file_service import FileService from services.file_service import FileService





+ 1
- 1
api/controllers/web/remote_files.py 查看文件

from controllers.web.wraps import WebApiResource from controllers.web.wraps import WebApiResource
from core.file import helpers as file_helpers from core.file import helpers as file_helpers
from core.helper import ssrf_proxy 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 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 from services.file_service import FileService





+ 3
- 1
api/tests/integration_tests/tasks/test_remove_app_and_related_data_task.py 查看文件

from sqlalchemy import delete from sqlalchemy import delete


from core.variables.segments import StringSegment 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.enums import CreatorUserRole
from models.model import App, UploadFile from models.model import App, UploadFile
from models.workflow import WorkflowDraftVariable, WorkflowDraftVariableFile from models.workflow import WorkflowDraftVariable, WorkflowDraftVariableFile


# Create WorkflowDraftVariableFile records # Create WorkflowDraftVariableFile records
from core.variables.types import SegmentType from core.variables.types import SegmentType

var_file1 = WorkflowDraftVariableFile( var_file1 = WorkflowDraftVariableFile(
tenant_id=tenant.id, tenant_id=tenant.id,
app_id=app.id, app_id=app.id,

+ 1
- 1
api/tests/integration_tests/test_offload.py 查看文件

import pytest import pytest
from sqlalchemy.orm import Session, joinedload, selectinload from sqlalchemy.orm import Session, joinedload, selectinload


from extensions.ext_database import db
from libs.datetime_utils import naive_utc_now from libs.datetime_utils import naive_utc_now
from libs.uuid_utils import uuidv7 from libs.uuid_utils import uuidv7
from models import db
from models.enums import CreatorUserRole from models.enums import CreatorUserRole
from models.model import UploadFile from models.model import UploadFile
from models.workflow import WorkflowNodeExecutionModel, WorkflowNodeExecutionOffload, WorkflowNodeExecutionTriggeredFrom from models.workflow import WorkflowNodeExecutionModel, WorkflowNodeExecutionOffload, WorkflowNodeExecutionTriggeredFrom

Loading…
取消
儲存