diff --git a/mooc/models/__init__.py b/mooc/models/__init__.py index 1ea8c3d..df63918 100644 --- a/mooc/models/__init__.py +++ b/mooc/models/__init__.py @@ -1,7 +1,10 @@ -# 显式导入所有模型模块 +# 显式导入所有模型模块,添加新的模型类所在模块(users模块)的导入 from mooc.models.admin import * # noqa from mooc.models.account import * # noqa from mooc.models.uni_account import * # noqa +from mooc.models.core import * # noqa +from mooc.models.userapi import * # noqa +from mooc.models.users import * # noqa # 新增导入,确保能识别users.py里的模型类所在模块 # 导出常用的模型类 from mooc.models.account import ( @@ -31,11 +34,57 @@ from mooc.models.uni_account import ( UniVerifycode ) +from mooc.models.core import ( + ImsCoreCache, + ImsCoreCron, + ImsCoreCronRecord, + ImsCoreJob, + ImsCoreMenu, + ImsCoreMenuShortcut, + ImsCorePaylog, + ImsCorePerformance, + ImsCoreQueue, + ImsCoreRefundlog, + ImsCoreResource, + ImsCoreSendsmsLog, + ImsCoreSessions, + ImsCoreSettings +) + +from mooc.models.userapi import ( + ImsUserapiCache, + ImsUserapiReply +) + +from mooc.models.users import ( + ImsUsers, + ImsUsersBind, + ImsUsersCreateGroup, + ImsUsersGroup, + ImsUsersInvitation, + ImsUsersLastUse, + ImsUsersLoginLogs, + ImsUsersOperateHistory, + ImsUsersOperateStar, + ImsUsersPermission, + ImsUsersProfile, + ImsUsersExtraGroup, + ImsUsersExtraLimit, + ImsUsersExtraModules, + ImsUsersExtraTemplates, + ImsUsersFailedLogin, + ImsUsersFounderGroup, + ImsUsersFounderOwnCreateGroups, + ImsUsersFounderOwnUniGroups, + ImsUsersFounderOwnUsers, + ImsUsersFounderOwnUsersGroups +) + def verify_all_models(): """验证所有模型是否已正确注册到Base.metadata""" from mooc.db.database import Base - # 所有预期的表名 + # 所有预期的表名,添加新模型对应的表名 expected_tables = { 'ims_goouc_fullexam_admin', # Account 相关表 @@ -61,7 +110,50 @@ def verify_all_models(): 'ims_uni_link_uniacid', 'ims_uni_modules', 'ims_uni_settings', - 'ims_uni_verifycode' + 'ims_uni_verifycode', + + # ImsCore 相关表名 + "ims_core_cache", + "ims_core_cron", + "ims_core_cron_record", + "ims_core_job", + "ims_core_menu", + "ims_core_menu_shortcut", + "ims_core_paylog", + "ims_core_performance", + "ims_core_queue", + "ims_core_refundlog", + "ims_core_resource", + "ims_core_sendsms_log", + "ims_core_sessions", + "ims_core_settings", + + # ImsUserapi 相关表名 + "ims_userapi_cache", + "ims_userapi_reply", + + # ImsUsers 相关表名 + "ims_users", + "ims_users_bind", + "ims_users_create_group", + "ims_users_group", + "ims_users_invitation", + "ims_users_lastuse", + "ims_users_login_logs", + "ims_users_operate_history", + "ims_users_operate_star", + "ims_users_permission", + "ims_users_profile", + "ims_users_extra_group", + "ims_users_extra_limit", + "ims_users_extra_modules", + "ims_users_extra_templates", + "ims_users_failed_login", + "ims_users_founder_group", + "ims_users_founder_own_create_groups", + "ims_users_founder_own_uni_groups", + "ims_users_founder_own_users", + "ims_users_founder_own_users_groups" } actual_tables = set(Base.metadata.tables.keys()) @@ -79,4 +171,4 @@ def verify_all_models(): def get_all_table_names(): """获取所有已注册的表名""" from mooc.db.database import Base - return sorted(Base.metadata.tables.keys()) + return sorted(Base.metadata.tables.keys()) \ No newline at end of file