# -*- coding: utf-8 -*-
import io

sql_path = r'C:\Users\jcxiong\Documents\Codex\MyProject\trafficAudit\docs\init.sql'
with io.open(sql_path, encoding='utf-8') as f:
    sql = f.read()

old = """    enterprise_code VARCHAR(50) COMMENT '统一社会信用代码',
    enterprise_name VARCHAR(200) COMMENT '企业名称',
"""
new = """    enterprise_code VARCHAR(50) COMMENT '企业代码',
    enterprise_name VARCHAR(200) COMMENT '企业名称',
    unified_credit_code VARCHAR(50) COMMENT '统一社会信用代码',
    report_unit VARCHAR(200) COMMENT '填报单位',
"""
assert old in sql, 'anchor1 not found'
sql = sql.replace(old, new)

old2 = """    -- 填报信息
    contact_person VARCHAR(50) COMMENT '填表人',
"""
new2 = """    -- 填报信息
    unit_leader VARCHAR(50) COMMENT '单位负责人',
    stats_leader VARCHAR(50) COMMENT '统计负责人',
    contact_person VARCHAR(50) COMMENT '填表人',
"""
assert old2 in sql, 'anchor2 not found'
sql = sql.replace(old2, new2)

with io.open(sql_path, 'w', encoding='utf-8', newline='') as f:
    f.write(sql)
print('init.sql updated')

md_path = r'C:\Users\jcxiong\Documents\Codex\MyProject\trafficAudit\docs\database.md'
with io.open(md_path, encoding='utf-8') as f:
    md = f.read()

old3 = """| enterprise_code | varchar(50) | 企业社会信用代码 |
| enterprise_name | varchar(200) | 企业名称 |
"""
new3 = """| enterprise_code | varchar(50) | 企业代码 |
| enterprise_name | varchar(200) | 企业名称 |
| unified_credit_code | varchar(50) | 统一社会信用代码 |
| report_unit | varchar(200) | 填报单位 |
"""
assert old3 in md, 'md anchor1 not found'
md = md.replace(old3, new3)

old4 = """| contact_person | varchar(50) | 填表人 |
| contact_phone | varchar(20) | 联系电话 |
"""
new4 = """| unit_leader | varchar(50) | 单位负责人 |
| stats_leader | varchar(50) | 统计负责人 |
| contact_person | varchar(50) | 填表人 |
| contact_phone | varchar(20) | 联系电话 |
"""
assert old4 in md, 'md anchor2 not found'
md = md.replace(old4, new4)

with io.open(md_path, 'w', encoding='utf-8', newline='') as f:
    f.write(md)
print('database.md updated')