from read_log_times import connect
|
from pathlib import Path
|
import json
|
import re
|
|
HERE = Path(__file__).resolve().parent
|
ROOT = HERE.parents[1]
|
ADDITIONS = {
|
'type': "int DEFAULT 1 COMMENT '杂项类型：mes_wm_misc_receipt_type' AFTER `name`",
|
'source_doc_type': "varchar(64) DEFAULT NULL COMMENT '来源单据类型' AFTER `type`",
|
'source_doc_id': "bigint DEFAULT NULL COMMENT '来源单据ID' AFTER `source_doc_type`",
|
'source_doc_code': "varchar(100) DEFAULT NULL COMMENT '来源单据编码' AFTER `source_doc_id`",
|
}
|
|
with connect() as c:
|
with c.cursor() as q:
|
q.execute('SHOW CREATE TABLE mes_wm_misc_receipt')
|
old_schema = q.fetchall()
|
q.execute('SHOW COLUMNS FROM mes_wm_misc_receipt')
|
columns = {r['Field'] for r in q.fetchall()}
|
q.execute('SELECT * FROM mes_wm_misc_receipt ORDER BY id')
|
before = q.fetchall()
|
missing = [name for name in ADDITIONS if name not in columns]
|
backup = HERE/'misc-receipt-schema-before.json'
|
if missing:
|
with backup.open('x', encoding='utf-8') as f:
|
json.dump(dict(target='36.213.150.237:10000/mom-hm', schema=old_schema,
|
rows=before, missing=missing), f, ensure_ascii=False, indent=2, default=str)
|
sql = 'ALTER TABLE mes_wm_misc_receipt '+', '.join(
|
'ADD COLUMN `'+name+'` '+ADDITIONS[name] for name in missing)
|
q.execute(sql)
|
# 验证三个数据对象的全部映射字段都能查询。
|
counts = {}
|
for suffix, class_name in [('', 'MesWmMiscReceiptDO'), ('_line', 'MesWmMiscReceiptLineDO'),
|
('_detail', 'MesWmMiscReceiptDetailDO')]:
|
source = (ROOT / ('yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/dal/dataobject/wm/miscreceipt/'+class_name+'.java')).read_text(encoding='utf-8')
|
fields = re.findall(r'private\s+\w+\s+(\w+);', source)
|
mapped = [re.sub(r'(?<!^)([A-Z])', r'_\1', name).lower() for name in fields]
|
mapped += ['create_time', 'update_time', 'creator', 'updater', 'deleted']
|
table = 'mes_wm_misc_receipt'+suffix
|
q.execute('SELECT '+','.join('`'+field+'`' for field in mapped)+
|
' FROM '+table+' WHERE deleted=0 ORDER BY id DESC LIMIT 10')
|
counts[table] = len(q.fetchall())
|
q.execute('SELECT * FROM mes_wm_misc_receipt ORDER BY id')
|
after = q.fetchall()
|
assert [{k: row[k] for k in columns} for row in after] == before, 'Existing data changed'
|
result = dict(target='36.213.150.237:10000/mom-hm', added_columns=missing,
|
original_data_unchanged=True, query_verified=counts, backup=str(backup))
|
(HERE/'misc-receipt-schema-result.json').write_text(json.dumps(result, ensure_ascii=False, indent=2), encoding='utf-8')
|
print(json.dumps(result, ensure_ascii=False))
|
