545 lines
26 KiB
Python
545 lines
26 KiB
Python
"""数据库结构自动迁移：检查并补全缺失的列/表
|
||
|
||
用法：
|
||
venv/bin/python3 migrate_db.py
|
||
|
||
每次代码升级后自动执行，确保生产数据库结构与代码一致。
|
||
"""
|
||
|
||
import sys
|
||
import os
|
||
|
||
# 切换到项目根目录
|
||
os.chdir(os.path.dirname(os.path.abspath(__file__)))
|
||
sys.path.insert(0, os.getcwd())
|
||
|
||
from services.db import query
|
||
|
||
|
||
def table_exists(name):
|
||
rows = query(
|
||
"SELECT TABLE_NAME FROM information_schema.TABLES "
|
||
"WHERE TABLE_SCHEMA=DATABASE() AND TABLE_NAME=%s",
|
||
(name,)
|
||
)
|
||
return len(rows or []) > 0
|
||
|
||
|
||
def column_exists(table, column):
|
||
rows = query(
|
||
"SELECT COLUMN_NAME FROM information_schema.COLUMNS "
|
||
"WHERE TABLE_SCHEMA=DATABASE() AND TABLE_NAME=%s AND COLUMN_NAME=%s",
|
||
(table, column)
|
||
)
|
||
return len(rows or []) > 0
|
||
|
||
def index_exists(table, index_name):
|
||
rows = query(
|
||
"SELECT INDEX_NAME FROM information_schema.STATISTICS "
|
||
"WHERE TABLE_SCHEMA=DATABASE() AND TABLE_NAME=%s AND INDEX_NAME=%s LIMIT 1",
|
||
(table, index_name)
|
||
)
|
||
return len(rows or []) > 0
|
||
|
||
|
||
def add_column(table, column, definition):
|
||
sql = f"ALTER TABLE `{table}` ADD COLUMN `{column}` {definition}"
|
||
query(sql)
|
||
print(f" ✅ {table}.{column}")
|
||
|
||
|
||
def rename_column(table, old_column, new_column):
|
||
query(f"ALTER TABLE `{table}` RENAME COLUMN `{old_column}` TO `{new_column}`")
|
||
print(f" ✅ {table}.{old_column} → {new_column}")
|
||
|
||
|
||
def create_table(name, definition):
|
||
query(f"CREATE TABLE `{name}` ({definition}) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci")
|
||
print(f" ✅ 新建表 {name}")
|
||
|
||
|
||
# ─── 需要检查并自动创建的表 ───
|
||
TABLES = {
|
||
'contract_records': """
|
||
`id` bigint NOT NULL AUTO_INCREMENT,
|
||
`contract_no` varchar(80) NOT NULL,
|
||
`supplier_id` int DEFAULT NULL,
|
||
`supplier_name` varchar(255) NOT NULL DEFAULT '',
|
||
`buyer_name` varchar(255) NOT NULL DEFAULT '',
|
||
`order_type` varchar(32) NOT NULL DEFAULT '',
|
||
`template_id` int DEFAULT NULL,
|
||
`template_name` varchar(100) NOT NULL DEFAULT '',
|
||
`order_count` int NOT NULL DEFAULT 0,
|
||
`product_count` int NOT NULL DEFAULT 0,
|
||
`total_amount` decimal(18,2) NOT NULL DEFAULT 0,
|
||
`status` varchar(32) NOT NULL DEFAULT '已生成',
|
||
`signer` varchar(100) NOT NULL DEFAULT '',
|
||
`signed_at` date DEFAULT NULL,
|
||
`effective_date` date DEFAULT NULL,
|
||
`expiry_date` date DEFAULT NULL,
|
||
`payment_terms` text NULL,
|
||
`delivery_terms` text NULL,
|
||
`remark` text NULL,
|
||
`order_ids_json` longtext NULL,
|
||
`orders_json` longtext NULL,
|
||
`current_file_path` varchar(500) NOT NULL DEFAULT '',
|
||
`current_file_name` varchar(255) NOT NULL DEFAULT '',
|
||
`created_by` int DEFAULT NULL,
|
||
`created_by_name` varchar(100) NOT NULL DEFAULT '',
|
||
`created_at` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP,
|
||
`updated_by` int DEFAULT NULL,
|
||
`updated_by_name` varchar(100) NOT NULL DEFAULT '',
|
||
`updated_at` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
|
||
`is_deleted` tinyint NOT NULL DEFAULT 0,
|
||
PRIMARY KEY (`id`),
|
||
UNIQUE KEY `uk_contract_no` (`contract_no`),
|
||
KEY `idx_contract_supplier` (`supplier_name`),
|
||
KEY `idx_contract_status` (`status`, `created_at`),
|
||
KEY `idx_contract_type` (`order_type`, `created_at`)
|
||
""",
|
||
'contract_files': """
|
||
`id` bigint NOT NULL AUTO_INCREMENT,
|
||
`contract_id` bigint NOT NULL,
|
||
`version` int NOT NULL,
|
||
`file_path` varchar(500) NOT NULL,
|
||
`file_name` varchar(255) NOT NULL DEFAULT '',
|
||
`generated_at` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP,
|
||
`generated_by` int DEFAULT NULL,
|
||
`generated_by_name` varchar(100) NOT NULL DEFAULT '',
|
||
`is_current` tinyint NOT NULL DEFAULT 0,
|
||
`remark` varchar(500) NOT NULL DEFAULT '',
|
||
PRIMARY KEY (`id`),
|
||
UNIQUE KEY `uk_contract_file_version` (`contract_id`, `version`),
|
||
KEY `idx_contract_file_current` (`contract_id`, `is_current`)
|
||
""",
|
||
'contract_events': """
|
||
`id` bigint NOT NULL AUTO_INCREMENT,
|
||
`contract_id` bigint NOT NULL,
|
||
`action` varchar(80) NOT NULL,
|
||
`detail` text NULL,
|
||
`user_id` int DEFAULT NULL,
|
||
`user_name` varchar(100) NOT NULL DEFAULT '',
|
||
`created_at` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP,
|
||
PRIMARY KEY (`id`),
|
||
KEY `idx_contract_event` (`contract_id`, `created_at`)
|
||
""",
|
||
'planning_suggestion_snapshots': """
|
||
`id` bigint NOT NULL AUTO_INCREMENT,
|
||
`snapshot_date` date NOT NULL COMMENT '快照日期',
|
||
`product_code` varchar(100) NOT NULL COMMENT '货品编号',
|
||
`product_name` varchar(300) NOT NULL DEFAULT '' COMMENT '货品名称',
|
||
`suggested_qty` int NOT NULL DEFAULT 0 COMMENT '建议下单数量',
|
||
`consecutive_days` int NOT NULL DEFAULT 0 COMMENT '连续建议天数',
|
||
`created_at` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP,
|
||
PRIMARY KEY (`id`),
|
||
UNIQUE KEY `uk_planning_snapshot_date_product` (`snapshot_date`, `product_code`),
|
||
KEY `idx_planning_snapshot_product_date` (`product_code`, `snapshot_date`)
|
||
""",
|
||
'spot_authorization_materials': """
|
||
`id` bigint NOT NULL AUTO_INCREMENT,
|
||
`merchant_code` varchar(100) NOT NULL DEFAULT '' COMMENT '商家编码',
|
||
`product_name` varchar(300) NOT NULL DEFAULT '' COMMENT '货品名称',
|
||
`platform` varchar(100) NOT NULL DEFAULT '' COMMENT '平台',
|
||
`operator` varchar(100) NOT NULL DEFAULT '' COMMENT '负责运营',
|
||
`authorization_letter` varchar(255) NOT NULL DEFAULT '' COMMENT '授权书文件名',
|
||
`testing_report` varchar(255) NOT NULL DEFAULT '' COMMENT '检测报告文件名',
|
||
`factory_qualification` text NOT NULL COMMENT '工厂资质文件名（单个文件名或JSON数组）',
|
||
`trademark` text NOT NULL COMMENT '商标文件名（单个文件名或JSON数组）',
|
||
`remark` text NULL COMMENT '备注',
|
||
`created_by` int DEFAULT NULL COMMENT '创建人账号ID',
|
||
`created_at` timestamp NULL DEFAULT CURRENT_TIMESTAMP,
|
||
`updated_at` timestamp NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
|
||
PRIMARY KEY (`id`),
|
||
KEY `idx_spot_authorization_merchant` (`merchant_code`),
|
||
KEY `idx_spot_authorization_name` (`product_name`)
|
||
""",
|
||
'sales_management_records': """
|
||
`id` bigint NOT NULL AUTO_INCREMENT,
|
||
`c01` text NULL COMMENT '批次',
|
||
`c02` text NULL COMMENT '货品编码',
|
||
`c03` text NULL COMMENT '货品名称',
|
||
`c04` text NULL COMMENT '确权人',
|
||
`c05` text NULL COMMENT '确权平台',
|
||
`c06` text NULL COMMENT '库存区间',
|
||
`c07` text NULL COMMENT '建档时间',
|
||
`c08` text NULL COMMENT '盘盈盘亏',
|
||
`c09` text NULL COMMENT '差异合计',
|
||
`c10` text NULL COMMENT '生产日期',
|
||
`c11` text NULL COMMENT '过期时间',
|
||
`c12` text NULL COMMENT '成本价',
|
||
`c13` text NULL COMMENT '总库存',
|
||
`c14` text NULL COMMENT '可发库存',
|
||
`c15` text NULL COMMENT '可支撑天数',
|
||
`c16` text NULL COMMENT '30天销量',
|
||
`c17` text NULL COMMENT '仓库存合计',
|
||
`c18` text NULL COMMENT '29青鸟海外仓',
|
||
`c19` text NULL COMMENT '27启程仓',
|
||
`c20` text NULL COMMENT '26海外抖音仓',
|
||
`c21` text NULL COMMENT '25海外拼多多保税仓',
|
||
`c22` text NULL COMMENT '1蝶雅仓',
|
||
`c23` text NULL COMMENT '2青鸟仓',
|
||
`c24` text NULL COMMENT '4百橙仓',
|
||
`c25` text NULL COMMENT '5东升仓',
|
||
`c26` text NULL COMMENT '6CK仓',
|
||
`c27` text NULL COMMENT '9天猫菜鸟仓',
|
||
`c28` text NULL COMMENT '10唯品会海外仓',
|
||
`c29` text NULL COMMENT '京东海外11仓',
|
||
`c30` text NULL COMMENT '京东海外12仓',
|
||
`c31` text NULL COMMENT '京东13仓',
|
||
`c32` text NULL COMMENT '京东14仓',
|
||
`c33` text NULL COMMENT '16蝶雅南阳仓',
|
||
`c34` text NULL COMMENT '分类',
|
||
`c35` text NULL COMMENT '戴总确认意见',
|
||
`c36` text NULL COMMENT '运营核实更改',
|
||
`c37` text NULL COMMENT '运营核对备注',
|
||
`created_at` timestamp NULL DEFAULT CURRENT_TIMESTAMP,
|
||
`updated_at` timestamp NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
|
||
PRIMARY KEY (`id`),
|
||
KEY `idx_sales_management_code` (`c02`(100)),
|
||
KEY `idx_sales_management_name` (`c03`(150))
|
||
""",
|
||
'brand_information': """
|
||
`id` int NOT NULL AUTO_INCREMENT,
|
||
`brand` varchar(200) NOT NULL COMMENT '品牌名称',
|
||
`category` varchar(20) NOT NULL DEFAULT '' COMMENT '国内/海外，空表示未分类',
|
||
`created_at` timestamp NULL DEFAULT CURRENT_TIMESTAMP,
|
||
`updated_at` timestamp NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
|
||
PRIMARY KEY (`id`),
|
||
UNIQUE KEY `uk_brand_information_brand` (`brand`)
|
||
""",
|
||
'spot_applications': """
|
||
`id` int NOT NULL AUTO_INCREMENT,
|
||
`applicant_user_id` int NOT NULL COMMENT '申请人账号ID',
|
||
`application_date` date NOT NULL COMMENT '申请日期',
|
||
`department` varchar(100) NOT NULL DEFAULT '' COMMENT '部门（组织架构导入后补充）',
|
||
`brand` varchar(200) NOT NULL DEFAULT '' COMMENT '品牌',
|
||
`product_name` varchar(300) NOT NULL COMMENT '产品名称',
|
||
`manufacturer` varchar(300) NOT NULL DEFAULT '' COMMENT '厂家',
|
||
`image_path` varchar(500) NOT NULL DEFAULT '' COMMENT '产品图片路径',
|
||
`product_link` varchar(1000) NOT NULL DEFAULT '' COMMENT '产品链接',
|
||
`sample_quantity` int NOT NULL DEFAULT 0 COMMENT '寄样数量',
|
||
`barcode` varchar(100) NOT NULL DEFAULT '' COMMENT '条形码',
|
||
`recipient_info` text COMMENT '收件信息',
|
||
`created_at` timestamp NULL DEFAULT CURRENT_TIMESTAMP,
|
||
`updated_at` timestamp NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
|
||
PRIMARY KEY (`id`),
|
||
KEY `idx_spot_application_user` (`applicant_user_id`),
|
||
KEY `idx_spot_application_date` (`application_date`),
|
||
CONSTRAINT `fk_spot_application_user` FOREIGN KEY (`applicant_user_id`) REFERENCES `dashboard_users` (`id`)
|
||
""",
|
||
'spot_confirmations': """
|
||
`id` int NOT NULL AUTO_INCREMENT,
|
||
`application_id` int NOT NULL COMMENT '现货申请ID',
|
||
`sample_status` varchar(50) NOT NULL DEFAULT '' COMMENT '样品状态',
|
||
`sample_remark` text COMMENT '样品备注',
|
||
`oa_approval_no` varchar(100) NOT NULL DEFAULT '' COMMENT 'OA审批单号',
|
||
`express_tracking_no` varchar(100) NOT NULL DEFAULT '' COMMENT '寄样快递单号',
|
||
`arrival_date` date DEFAULT NULL COMMENT '到样日期',
|
||
`barcode` varchar(100) NOT NULL DEFAULT '' COMMENT '可编辑条形码',
|
||
`remark` text COMMENT '备注',
|
||
`receiving_department` varchar(100) NOT NULL DEFAULT '' COMMENT '领样部门',
|
||
`recipient` varchar(100) NOT NULL DEFAULT '' COMMENT '领样人',
|
||
`sample_confirmed` varchar(50) NOT NULL DEFAULT '' COMMENT '样品确认',
|
||
`sample_confirmation_reason` text COMMENT '样品确认原因',
|
||
`sample_location` varchar(200) NOT NULL DEFAULT '' COMMENT '样品位置',
|
||
`product_code` varchar(100) NOT NULL DEFAULT '' COMMENT '产品编码',
|
||
`final_product_name` varchar(300) NOT NULL DEFAULT '' COMMENT '货品名称（最终名称）',
|
||
`platform` varchar(200) NOT NULL DEFAULT '' COMMENT '平台',
|
||
`created_at` timestamp NULL DEFAULT CURRENT_TIMESTAMP,
|
||
`updated_at` timestamp NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
|
||
PRIMARY KEY (`id`),
|
||
UNIQUE KEY `uk_spot_confirmation_application` (`application_id`),
|
||
KEY `idx_spot_confirmation_status` (`sample_status`),
|
||
CONSTRAINT `fk_spot_confirmation_application` FOREIGN KEY (`application_id`) REFERENCES `spot_applications` (`id`)
|
||
""",
|
||
'user_positions': """
|
||
`id` int NOT NULL AUTO_INCREMENT,
|
||
`position_code` varchar(80) NOT NULL COMMENT '岗位编码',
|
||
`position_name` varchar(100) NOT NULL COMMENT '岗位名称',
|
||
`sort_order` int NOT NULL DEFAULT 0 COMMENT '排序值',
|
||
`status` tinyint NOT NULL DEFAULT 1 COMMENT '1=启用 0=停用',
|
||
`created_at` timestamp NULL DEFAULT CURRENT_TIMESTAMP,
|
||
`updated_at` timestamp NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
|
||
PRIMARY KEY (`id`),
|
||
UNIQUE KEY `uk_position_code` (`position_code`),
|
||
UNIQUE KEY `uk_position_name` (`position_name`)
|
||
""",
|
||
'user_column_widths': """
|
||
`id` int NOT NULL AUTO_INCREMENT,
|
||
`user_id` int NOT NULL,
|
||
`page` varchar(50) NOT NULL COMMENT '页面标识',
|
||
`field_name` varchar(100) NOT NULL COMMENT '稳定字段名',
|
||
`width` int NOT NULL COMMENT '列宽(px)',
|
||
`created_at` timestamp NULL DEFAULT CURRENT_TIMESTAMP,
|
||
`updated_at` timestamp NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
|
||
PRIMARY KEY (`id`),
|
||
UNIQUE KEY `uk_user_page_field` (`user_id`,`page`,`field_name`),
|
||
KEY `idx_user_page` (`user_id`,`page`)
|
||
""",
|
||
}
|
||
|
||
|
||
# ─── 需要保持与前端字段一致的列重命名 ───
|
||
COLUMN_RENAMES = {
|
||
'domestic_orders': [
|
||
('编码', '货品编号'),
|
||
('名称', '货品名称'),
|
||
],
|
||
}
|
||
|
||
|
||
# ─── 需要检查的列 ───
|
||
SCHEMA = {
|
||
'spot_applications': [
|
||
('status', "varchar(50) NOT NULL DEFAULT '待处理' COMMENT '申请状态'"),
|
||
('reason', "varchar(500) NOT NULL DEFAULT '' COMMENT '申请原因'"),
|
||
('applicant_name', "varchar(200) NOT NULL DEFAULT '' COMMENT '申请人姓名（历史导入文本）'"),
|
||
],
|
||
'dashboard_users': [
|
||
('must_change_password', "tinyint DEFAULT 1 COMMENT '1=下次登录必须修改密码'"),
|
||
('department', "varchar(50) NOT NULL DEFAULT '' COMMENT '部门'"),
|
||
],
|
||
'spot_confirmations': [
|
||
('barcode', "varchar(100) NOT NULL DEFAULT '' COMMENT '可编辑条形码'"),
|
||
],
|
||
'packaging_orders': [
|
||
('计划下单时间', "varchar(50) DEFAULT NULL COMMENT '计划下单日期'"),
|
||
('新旧品', "varchar(20) DEFAULT NULL COMMENT '新品、旧品或无'"),
|
||
('是否审核价格', "varchar(50) DEFAULT '未审核' COMMENT '包材价格审核状态'"),
|
||
('来源产品编码', "varchar(100) DEFAULT '' COMMENT '来源成品编码'"),
|
||
],
|
||
'domestic_orders': [
|
||
('金蝶入库', "varchar(50) DEFAULT '否' COMMENT '金蝶是否入库'"),
|
||
('未到货数', "int DEFAULT 0 COMMENT '未到货数量'"),
|
||
('完结时间', "date DEFAULT NULL COMMENT '订单完结日期'"),
|
||
],
|
||
'pending_orders': [
|
||
('product_orderer', "varchar(200) DEFAULT '' COMMENT '成品订单下单人（国内/海外订单同步）'"),
|
||
('packaging_orderer', "varchar(200) DEFAULT '' COMMENT '包材订单下单人同步'"),
|
||
('sales_snapshot', "longtext DEFAULT NULL COMMENT '计划预测点击下单时的销量信息快照'"),
|
||
('sales_snapshot_at', "datetime DEFAULT NULL COMMENT '销量信息快照时间'"),
|
||
],
|
||
'arrival_batches': [
|
||
('order_type', "varchar(20) NOT NULL DEFAULT 'domestic' COMMENT '订单类型'"),
|
||
('operator_name', "varchar(100) NOT NULL DEFAULT '' COMMENT '操作人'"),
|
||
('operated_at', 'datetime DEFAULT NULL COMMENT "操作时间"'),
|
||
],
|
||
'new_product_bom': [
|
||
('usage_quantity', "decimal(12,4) NOT NULL DEFAULT '1.0000' COMMENT 'BOM用量'"),
|
||
],
|
||
'products': [
|
||
# (列名, 定义)
|
||
('deleted_at', 'datetime DEFAULT NULL COMMENT "软删除时间"'),
|
||
('link', "varchar(500) DEFAULT '' COMMENT '链接'"),
|
||
('sku_type', "varchar(50) DEFAULT '' COMMENT '产品类型(爆品/日销/新品)'"),
|
||
('remark', "varchar(500) DEFAULT '' COMMENT '批注'"),
|
||
],
|
||
'warehouses': [
|
||
('warehouse_property', "varchar(100) DEFAULT NULL COMMENT '仓库属性(云仓,外仓)'"),
|
||
('category', "varchar(50) DEFAULT NULL COMMENT '仓库分类'"),
|
||
('disabled', "varchar(10) DEFAULT NULL COMMENT '是否禁用'"),
|
||
('remark', "varchar(500) DEFAULT NULL COMMENT '备注'"),
|
||
],
|
||
'new_products': [
|
||
('procurement_planner', "varchar(100) NOT NULL DEFAULT '' COMMENT '采购计划'"),
|
||
],
|
||
'suppliers': [
|
||
('data_status', "varchar(50) DEFAULT NULL COMMENT '数据状态'"),
|
||
('disabled_status', "varchar(50) DEFAULT NULL COMMENT '禁用状态'"),
|
||
('production_type', "varchar(200) DEFAULT NULL COMMENT '生产类型'"),
|
||
('review_date', "date DEFAULT NULL COMMENT '审核日期'"),
|
||
('reviewer', "varchar(100) DEFAULT NULL COMMENT '审核人'"),
|
||
('short_name', "varchar(200) DEFAULT NULL COMMENT '简称'"),
|
||
('supplier_group', "varchar(200) DEFAULT NULL COMMENT '供应商分组'"),
|
||
('procurement_follower', "varchar(100) DEFAULT NULL COMMENT '跟单小组'"),
|
||
('procurement_planner', "varchar(100) DEFAULT NULL COMMENT '采购计划'"),
|
||
],
|
||
'user_permissions': [
|
||
('updated_at', "timestamp DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP COMMENT '更新时间'"),
|
||
],
|
||
'role_permissions': [
|
||
('updated_at', "timestamp DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP COMMENT '更新时间'"),
|
||
],
|
||
'bom_packaging': [
|
||
('pkg_remark', "varchar(500) DEFAULT '' COMMENT '包材备注'"),
|
||
],
|
||
'plan_hidden': [
|
||
('created_at', "datetime DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间'"),
|
||
],
|
||
'contract_templates': [
|
||
('supplier_id', 'int DEFAULT NULL COMMENT "供应商ID"'),
|
||
],
|
||
}
|
||
|
||
|
||
def main():
|
||
print("")
|
||
print("数据库结构检查 & 自动迁移")
|
||
print("=" * 40)
|
||
print("")
|
||
|
||
changed = 0
|
||
created = 0
|
||
for table, definition in TABLES.items():
|
||
if table_exists(table):
|
||
continue
|
||
try:
|
||
create_table(table, definition)
|
||
created += 1
|
||
except Exception as e:
|
||
print(f" ❌ 表 {table} 创建失败: {e}")
|
||
|
||
# 初始化岗位字典，并把历史用户中出现过但尚未进入字典的岗位补齐。
|
||
# 只新增字典记录，不修改或删除 dashboard_users 中的历史数据。
|
||
if table_exists('user_positions'):
|
||
default_positions = [
|
||
('procurement_planner', '采购计划', 10),
|
||
('procurement_follower', '采购跟单', 20),
|
||
('product_developer', '产品开发', 30),
|
||
('operator', '运营专员', 40),
|
||
('bp', 'BP', 50),
|
||
('procurement_trainee', '采购管培生', 60),
|
||
('finance_specialist', '财务专员', 70),
|
||
('采购开发', '采购开发', 80),
|
||
]
|
||
for position_code, position_name, sort_order in default_positions:
|
||
query(
|
||
"INSERT IGNORE INTO user_positions "
|
||
"(position_code, position_name, sort_order, status) VALUES (%s, %s, %s, 1)",
|
||
(position_code, position_name, sort_order)
|
||
)
|
||
if table_exists('dashboard_users'):
|
||
query(
|
||
"INSERT IGNORE INTO user_positions (position_code, position_name, sort_order, status) "
|
||
"SELECT DISTINCT TRIM(position), TRIM(position), 0, 1 "
|
||
"FROM dashboard_users WHERE position IS NOT NULL AND TRIM(position) <> ''"
|
||
)
|
||
print(" ✅ 初始化岗位字典")
|
||
|
||
for table, renames in COLUMN_RENAMES.items():
|
||
if not table_exists(table):
|
||
print(f" ⚠️ 表 {table} 不存在，跳过字段重命名")
|
||
continue
|
||
for old_column, new_column in renames:
|
||
if column_exists(table, new_column):
|
||
continue
|
||
if not column_exists(table, old_column):
|
||
print(f" ⚠️ {table}.{old_column} 不存在，无法重命名")
|
||
continue
|
||
try:
|
||
rename_column(table, old_column, new_column)
|
||
changed += 1
|
||
except Exception as e:
|
||
print(f" ❌ {table}.{old_column} 重命名失败: {e}")
|
||
|
||
# 同步已有账号权限和字段方案中的旧字段名。
|
||
if table_exists('user_permissions'):
|
||
permission_renames = [
|
||
('field:编码:view', 'field:货品编号:view'),
|
||
('field:编码:edit', 'field:货品编号:edit'),
|
||
('field:名称:view', 'field:货品名称:view'),
|
||
('field:名称:edit', 'field:货品名称:edit'),
|
||
]
|
||
for old_operation, new_operation in permission_renames:
|
||
query("""INSERT IGNORE INTO user_permissions (user_id, page, operation, permission)
|
||
SELECT user_id, page, %s, permission
|
||
FROM user_permissions
|
||
WHERE page='orders_domestic' AND operation=%s""",
|
||
(new_operation, old_operation))
|
||
query("""UPDATE user_permissions AS target
|
||
JOIN user_permissions AS source
|
||
ON source.user_id=target.user_id AND source.page=target.page
|
||
SET target.permission=GREATEST(target.permission, source.permission)
|
||
WHERE target.page='orders_domestic' AND target.operation=%s
|
||
AND source.operation=%s""",
|
||
(new_operation, old_operation))
|
||
query("DELETE FROM user_permissions WHERE page='orders_domestic' AND operation=%s",
|
||
(old_operation,))
|
||
if table_exists('user_column_schemes'):
|
||
query("""UPDATE user_column_schemes
|
||
SET columns_json=REPLACE(REPLACE(REPLACE(REPLACE(columns_json,
|
||
'"field":"编码"', '"field":"货品编号"'),
|
||
'"field": "编码"', '"field": "货品编号"'),
|
||
'"field":"名称"', '"field":"货品名称"'),
|
||
'"field": "名称"', '"field": "货品名称"')
|
||
WHERE page='orders_domestic'""")
|
||
|
||
added = 0
|
||
for table, columns in SCHEMA.items():
|
||
if not table_exists(table):
|
||
print(f" ⚠️ 表 {table} 不存在，跳过")
|
||
continue
|
||
|
||
for col_name, col_def in columns:
|
||
if column_exists(table, col_name):
|
||
continue
|
||
try:
|
||
add_column(table, col_name, col_def)
|
||
added += 1
|
||
except Exception as e:
|
||
print(f" ❌ {table}.{col_name} 添加失败: {e}")
|
||
|
||
# 国内订单的既有字段也要统一默认值和历史派生值，避免仅新增数据库生效。
|
||
if table_exists('domestic_orders'):
|
||
query("ALTER TABLE domestic_orders MODIFY COLUMN `金蝶入库` varchar(50) DEFAULT '否'")
|
||
query("UPDATE domestic_orders SET `金蝶入库`='否' WHERE `金蝶入库` IS NULL OR `金蝶入库`=''")
|
||
query("""UPDATE domestic_orders
|
||
SET `未到货数`=COALESCE(`数量`,0)-COALESCE(`实际到货数量`,0)
|
||
WHERE `未到货数` IS NULL
|
||
OR `未到货数`<>COALESCE(`数量`,0)-COALESCE(`实际到货数量`,0)""")
|
||
print(" ✅ 国内订单默认值及未到货数已同步")
|
||
|
||
# pending_orders 需支持同一商品多次下单（多条 ordered/rolled_back 历史），
|
||
# 移除 product_code 唯一索引并改为普通索引，否则回退订单会因唯一约束冲突失败。
|
||
if table_exists('pending_orders'):
|
||
if index_exists('pending_orders', 'uk_product_code'):
|
||
try:
|
||
query("ALTER TABLE pending_orders DROP INDEX uk_product_code")
|
||
print(" ✅ pending_orders 移除唯一索引 uk_product_code")
|
||
changed += 1
|
||
except Exception as e:
|
||
print(f" ❌ 移除唯一索引 uk_product_code 失败: {e}")
|
||
if not index_exists('pending_orders', 'idx_product_code'):
|
||
try:
|
||
query("ALTER TABLE pending_orders ADD INDEX idx_product_code (product_code)")
|
||
print(" ✅ pending_orders 新增普通索引 idx_product_code")
|
||
except Exception as e:
|
||
print(f" ❌ 新增普通索引 idx_product_code 失败: {e}")
|
||
|
||
# 产品业务编码允许在名称不同（成品/包材分别建档）时重复。
|
||
if table_exists('products'):
|
||
if index_exists('products', 'uk_product_code'):
|
||
try:
|
||
query("ALTER TABLE products DROP INDEX uk_product_code")
|
||
print(" ✅ products 移除唯一索引 uk_product_code")
|
||
changed += 1
|
||
except Exception as e:
|
||
print(f" ❌ 移除 products 唯一索引 uk_product_code 失败: {e}")
|
||
if not index_exists('products', 'uk_product_code_name'):
|
||
try:
|
||
query("ALTER TABLE products ADD UNIQUE INDEX uk_product_code_name (product_code, product_name)")
|
||
print(" ✅ products 新增唯一索引 uk_product_code_name")
|
||
changed += 1
|
||
except Exception as e:
|
||
print(f" ❌ 新增 products 组合唯一索引失败: {e}")
|
||
if not index_exists('products', 'idx_product_code'):
|
||
try:
|
||
query("ALTER TABLE products ADD INDEX idx_product_code (product_code)")
|
||
print(" ✅ products 新增普通索引 idx_product_code")
|
||
changed += 1
|
||
except Exception as e:
|
||
print(f" ❌ 新增 products 普通索引失败: {e}")
|
||
|
||
print("")
|
||
if created or changed or added:
|
||
print(f"🎉 共新建 {created} 表，重命名 {changed} 列，新增 {added} 列")
|
||
else:
|
||
print("✅ 所有列已存在，无需迁移")
|
||
print("")
|
||
|
||
return 0 if created == 0 and changed == 0 and added == 0 else 1 # 1 表示有变更
|
||
|
||
|
||
if __name__ == '__main__':
|
||
sys.exit(main())
|