caebf90438
- flask_app.py 1166行→33行纯入口
- 新建 db.py（配置+连接+SQL工具）
- 新建 helpers.py（attach_common/monthly_finance/add_file_index）
- 新建 routes.py（全路由 Blueprint + 装饰器 + TABLES）
- 新建 migrations/seed_data.py（seed_db 搬迁）
- migrations/{tables,columns,data_fixes,seed}.py 改 import 为 from db
- 删除死代码 init_db（228行）+ latest_followup（10行）
- 反向依赖消除：migrations 不再 import flask_app
- 前端零改动，URL 不变
82 lines
4.3 KiB
Python
82 lines
4.3 KiB
Python
"""migrations/columns.py — 加列迁移（老表补字段，幂等）"""
|
|
|
|
|
|
def _add_column_if_missing(conn, table, column, ddl):
|
|
"""检查列是否存在，不存在才加（幂等）"""
|
|
from db import _exec, mysql, logger
|
|
|
|
cur = conn.cursor(dictionary=True)
|
|
cur.execute(f"SHOW COLUMNS FROM {table} LIKE %s", (column,))
|
|
exists = cur.fetchone()
|
|
cur.close()
|
|
if not exists:
|
|
try:
|
|
_exec(conn, ddl)
|
|
print(f"[migrate] {table}.{column} 列已添加")
|
|
except mysql.connector.Error as e:
|
|
logger.debug(f"add column {table}.{column} skipped: {e}")
|
|
|
|
|
|
def migrate_add_columns():
|
|
"""为老表补齐后续新增的字段"""
|
|
from db import db
|
|
|
|
conn = db()
|
|
try:
|
|
# tenant 字段（多工作台支持）
|
|
for table in ["sales_leads", "follow_up_records", "business_proposals",
|
|
"operation_projects", "product_versions", "finance_records",
|
|
"project_tasks"]:
|
|
_add_column_if_missing(conn, table, "tenant",
|
|
f"ALTER TABLE {table} ADD COLUMN tenant VARCHAR(100) NOT NULL DEFAULT '科普·无界'")
|
|
|
|
# business_proposals 扩展字段
|
|
_add_column_if_missing(conn, "business_proposals", "proposal_type",
|
|
"ALTER TABLE business_proposals ADD COLUMN proposal_type VARCHAR(100) NOT NULL DEFAULT '业务方案'")
|
|
_add_column_if_missing(conn, "business_proposals", "notes",
|
|
"ALTER TABLE business_proposals ADD COLUMN notes VARCHAR(2000) NOT NULL DEFAULT ''")
|
|
|
|
# product_versions 扩展字段
|
|
_add_column_if_missing(conn, "product_versions", "priority",
|
|
"ALTER TABLE product_versions ADD COLUMN priority VARCHAR(10) NOT NULL DEFAULT 'P2'")
|
|
_add_column_if_missing(conn, "product_versions", "start_date",
|
|
"ALTER TABLE product_versions ADD COLUMN start_date VARCHAR(30) NOT NULL DEFAULT ''")
|
|
_add_column_if_missing(conn, "product_versions", "plan_date",
|
|
"ALTER TABLE product_versions ADD COLUMN plan_date VARCHAR(30) NOT NULL DEFAULT ''")
|
|
_add_column_if_missing(conn, "product_versions", "dev_done_date",
|
|
"ALTER TABLE product_versions ADD COLUMN dev_done_date VARCHAR(30) NOT NULL DEFAULT ''")
|
|
_add_column_if_missing(conn, "product_versions", "test_date",
|
|
"ALTER TABLE product_versions ADD COLUMN test_date VARCHAR(30) NOT NULL DEFAULT ''")
|
|
_add_column_if_missing(conn, "product_versions", "devs",
|
|
"ALTER TABLE product_versions ADD COLUMN devs VARCHAR(500) NOT NULL DEFAULT ''")
|
|
_add_column_if_missing(conn, "product_versions", "testers",
|
|
"ALTER TABLE product_versions ADD COLUMN testers VARCHAR(500) NOT NULL DEFAULT ''")
|
|
|
|
# project_tasks 扩展字段
|
|
_add_column_if_missing(conn, "project_tasks", "status",
|
|
"ALTER TABLE project_tasks ADD COLUMN status VARCHAR(50) NOT NULL DEFAULT '未开始'")
|
|
_add_column_if_missing(conn, "project_tasks", "sort_order",
|
|
"ALTER TABLE project_tasks ADD COLUMN sort_order INT NOT NULL DEFAULT 0")
|
|
_add_column_if_missing(conn, "project_tasks", "priority",
|
|
"ALTER TABLE project_tasks ADD COLUMN priority VARCHAR(10) NOT NULL DEFAULT 'P2'")
|
|
|
|
# project_finances 12 个月度预算字段（确收/毛利/回款/费用）
|
|
for m in ["01","02","03","04","05","06","07","08","09","10","11","12"]:
|
|
for field in ["rev", "gross", "payment", "cost"]:
|
|
col = f"{field}_2026_{m}"
|
|
_add_column_if_missing(conn, "project_finances", col,
|
|
f"ALTER TABLE project_finances ADD COLUMN {col} DOUBLE NOT NULL DEFAULT 0")
|
|
|
|
# project_finances 总视图字段：已回款 / 应付 / 已付
|
|
_add_column_if_missing(conn, "project_finances", "total_payment",
|
|
"ALTER TABLE project_finances ADD COLUMN total_payment DOUBLE NOT NULL DEFAULT 0")
|
|
_add_column_if_missing(conn, "project_finances", "total_cost",
|
|
"ALTER TABLE project_finances ADD COLUMN total_cost DOUBLE NOT NULL DEFAULT 0")
|
|
_add_column_if_missing(conn, "project_finances", "total_paid",
|
|
"ALTER TABLE project_finances ADD COLUMN total_paid DOUBLE NOT NULL DEFAULT 0")
|
|
|
|
conn.commit()
|
|
print("[migrate] 加列迁移完成")
|
|
finally:
|
|
conn.close()
|