1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
# -*- 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')