update mooc/models/__init__.py. 加入core表,userapi表和users表

Signed-off-by: 雨过 <zxx1747362695@qq.com>
This commit is contained in:
雨过 2025-01-05 07:46:42 +00:00 committed by Gitee
parent bfa1a65a0e
commit dcc79ecd88
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F

View File

@ -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())