c91fee75d2
- models.py: 迁移25个模型类，使用统一的Mapped类型注解 * 包含: ChatStreams, Messages, PersonInfo, LLMUsage等 * 新增: PermissionNodes, UserPermissions, UserRelationships * 654行纯模型定义代码，无初始化逻辑 - migration.py: 重构数据库迁移逻辑 * check_and_migrate_database: 自动检查和迁移表结构 * create_all_tables: 快速创建所有表 * drop_all_tables: 测试用删除所有表 * 使用新架构的engine和models - __init__.py: 完善导出清单 * 导出所有25个模型类 * 导出迁移函数 * 导出Base和工具函数 - 辅助脚本: * extract_models.py: 自动提取模型定义 * cleanup_models.py: 清理非模型代码 核心层现已完整，下一步进入优化层实现
50 lines
1.3 KiB
Python
50 lines
1.3 KiB
Python
#!/usr/bin/env python3
|
||
"""清理 core/models.py，只保留模型定义"""
|
||
|
||
import os
|
||
|
||
# 文件路径
|
||
models_file = os.path.join(
|
||
os.path.dirname(os.path.dirname(__file__)),
|
||
"src",
|
||
"common",
|
||
"database",
|
||
"core",
|
||
"models.py"
|
||
)
|
||
|
||
print(f"正在清理文件: {models_file}")
|
||
|
||
# 读取文件
|
||
with open(models_file, "r", encoding="utf-8") as f:
|
||
lines = f.readlines()
|
||
|
||
# 找到最后一个模型类的结束位置（MonthlyPlan的 __table_args__ 结束）
|
||
# 我们要保留到第593行（包含）
|
||
keep_lines = []
|
||
found_end = False
|
||
|
||
for i, line in enumerate(lines, 1):
|
||
keep_lines.append(line)
|
||
|
||
# 检查是否到达 MonthlyPlan 的 __table_args__ 结束
|
||
if i > 580 and line.strip() == ")":
|
||
# 再检查前一行是否有 Index 相关内容
|
||
if "idx_monthlyplan" in "".join(lines[max(0, i-5):i]):
|
||
print(f"找到模型定义结束位置: 第 {i} 行")
|
||
found_end = True
|
||
break
|
||
|
||
if not found_end:
|
||
print("❌ 未找到模型定义结束标记")
|
||
exit(1)
|
||
|
||
# 写回文件
|
||
with open(models_file, "w", encoding="utf-8") as f:
|
||
f.writelines(keep_lines)
|
||
|
||
print(f"✅ 文件清理完成")
|
||
print(f"保留行数: {len(keep_lines)}")
|
||
print(f"原始行数: {len(lines)}")
|
||
print(f"删除行数: {len(lines) - len(keep_lines)}")
|