320 lines
11 KiB
Python
320 lines
11 KiB
Python
|
import shutil
|
|||
|
from pathlib import Path
|
|||
|
import os
|
|||
|
import string
|
|||
|
|
|||
|
|
|||
|
def code_make(dir_path, replace_dic={}, save_path="temp"):
|
|||
|
"""
|
|||
|
增删改查代码生成
|
|||
|
@param dir_path: 代码模板
|
|||
|
@param replace_dic: 需要替换的名称
|
|||
|
@param save_path: 保存路径
|
|||
|
"""
|
|||
|
code_path = Path(dir_path)
|
|||
|
if not code_path.exists():
|
|||
|
os.makedirs(code_path)
|
|||
|
if code_path.is_dir():
|
|||
|
save_path = Path(save_path)
|
|||
|
shutil.copytree(code_path, save_path)
|
|||
|
for root, dirs, files in os.walk(save_path):
|
|||
|
for file in files:
|
|||
|
file_path = Path(os.path.join(root, file))
|
|||
|
item_code_str = file_path.read_text(encoding="utf-8")
|
|||
|
new_code_str = item_code_str
|
|||
|
for replace_word, replaced_word in replace_dic.items():
|
|||
|
new_code_str = new_code_str.replace(replace_word, replaced_word)
|
|||
|
with open(file_path, "w", encoding="utf-8") as f:
|
|||
|
f.write(new_code_str)
|
|||
|
|
|||
|
|
|||
|
def to_snake(name):
|
|||
|
new_name = ""
|
|||
|
for i in range(len(name)):
|
|||
|
k = name[i]
|
|||
|
if name[i] in string.ascii_letters:
|
|||
|
if name[i].isupper():
|
|||
|
if i == 0:
|
|||
|
k = name[i].lower()
|
|||
|
else:
|
|||
|
k = '_' + name[i].lower()
|
|||
|
|
|||
|
new_name += k
|
|||
|
return new_name
|
|||
|
|
|||
|
|
|||
|
def to_upper_camel_case(name):
|
|||
|
items = name.split("_")
|
|||
|
new_name = ""
|
|||
|
for item in items:
|
|||
|
new_name += item[0].upper() + item[1:]
|
|||
|
return new_name
|
|||
|
|
|||
|
def table_str_convert(table_str, key='id'):
|
|||
|
replace_dic = {
|
|||
|
'varchar': 'String',
|
|||
|
'char': 'String',
|
|||
|
'date': 'Date',
|
|||
|
'datetime': 'DateTime',
|
|||
|
'text': 'Text',
|
|||
|
'int': 'Integer',
|
|||
|
'double': 'Double'
|
|||
|
}
|
|||
|
new_lines = []
|
|||
|
for line in table_str.split("\n"):
|
|||
|
if line:
|
|||
|
cols = line.split("\t")
|
|||
|
col_name = cols[0].replace(' ', "").replace("-", '_')
|
|||
|
col_type = cols[1].replace(" ", '')
|
|||
|
col_type = col_type.replace(col_type.split('(')[0], replace_dic[col_type.split('(')[0]])
|
|||
|
comment = cols[3]
|
|||
|
comment_str = f', comment="{comment}"' if comment.replace(" ", '') else ""
|
|||
|
primary_key_str = f', primary_key=True' if col_name == key else ''
|
|||
|
new_line = f'{col_name} = Column({col_type}{primary_key_str}{comment_str})'
|
|||
|
new_lines.append(new_line)
|
|||
|
new_content = "\n ".join(new_lines)
|
|||
|
return new_content
|
|||
|
|
|||
|
|
|||
|
def make_info_str(table_str, key='id'):
|
|||
|
replace_dic = {
|
|||
|
'varchar': 'str',
|
|||
|
'char': 'str',
|
|||
|
'text': 'str',
|
|||
|
'double': 'float'
|
|||
|
}
|
|||
|
info_lines = []
|
|||
|
add_info_lines = []
|
|||
|
for line in table_str.split("\n"):
|
|||
|
if line:
|
|||
|
cols = line.split("\t")
|
|||
|
col_name = cols[0].replace(' ', "").replace("-", '_')
|
|||
|
col_type = cols[1].replace(" ", '')
|
|||
|
if col_type.split('(')[0] in replace_dic:
|
|||
|
col_type = replace_dic[col_type.split('(')[0]]
|
|||
|
# comment = col_type[3]
|
|||
|
# comment_str = f',comment="{col_type[3]}"' if comment else ""
|
|||
|
# primary_key_str = f',primary_key=True' if col_name == key else ''
|
|||
|
new_line = f'{col_name}: Optional[{col_type}]'
|
|||
|
info_lines.append(new_line)
|
|||
|
|
|||
|
if col_name != "id":
|
|||
|
add_info_lines.append(new_line)
|
|||
|
|
|||
|
info_content = "\n ".join(info_lines)
|
|||
|
add_info_content = "\n ".join(add_info_lines)
|
|||
|
return info_content, add_info_content
|
|||
|
|
|||
|
|
|||
|
def crud_code_make(table_name: str, chinese_name, table_str, path: str):
|
|||
|
model_content = table_str_convert(table_str)
|
|||
|
info_content, add_info_content = make_info_str(table_str)
|
|||
|
upper_camel_case_name = to_upper_camel_case(table_name)
|
|||
|
code_make("Utils/CodeMake/CrudTemplate",
|
|||
|
{"EditItemType": upper_camel_case_name,
|
|||
|
"edit_item_type": table_name,
|
|||
|
"修改项类型": chinese_name,
|
|||
|
'"""model_content"""': model_content,
|
|||
|
'"""info_content"""': info_content,
|
|||
|
'"""add_info_content"""': add_info_content,
|
|||
|
}
|
|||
|
, f"{path}/{upper_camel_case_name}")
|
|||
|
|
|||
|
|
|||
|
crud_code_make("company_credit_rating",
|
|||
|
"主体评级",
|
|||
|
"""id int unsigned auto_increment
|
|||
|
company_id int 企业ID
|
|||
|
result char(5) 评级结果
|
|||
|
rate_date date 评级日期
|
|||
|
publish_date date 披露日期
|
|||
|
outlook char(5) 展望
|
|||
|
rate_agency varchar(50) 评级机构
|
|||
|
rate_file text 评级报告文件地址链接""",
|
|||
|
"Mods/OtherInformation2/Mods")
|
|||
|
|
|||
|
crud_code_make("company_business_compose",
|
|||
|
"业务构成",
|
|||
|
"""id int unsigned auto_increment
|
|||
|
company_id int 企业ID
|
|||
|
name varchar(255) not null 业务名称
|
|||
|
income double 收入(元)
|
|||
|
total_date date 统计截至时间
|
|||
|
remark text 业务描述""",
|
|||
|
"Mods/OtherInformation2/Mods")
|
|||
|
|
|||
|
crud_code_make("company_business_items",
|
|||
|
"重要项目",
|
|||
|
"""id int unsigned auto_increment
|
|||
|
company_id int 企业ID
|
|||
|
name varchar(255) not null 项目名称
|
|||
|
product varchar(255) 产品类型
|
|||
|
industry varchar(255) 行业类型
|
|||
|
region int 地区
|
|||
|
year int 业务年度
|
|||
|
remark text 项目描述""",
|
|||
|
"Mods/OtherInformation2/Mods")
|
|||
|
|
|||
|
crud_code_make("company_finance_main",
|
|||
|
"财务报表主要指标",
|
|||
|
"""id int unsigned auto_increment
|
|||
|
company_id int 企业ID
|
|||
|
total_date date 统计截至日期
|
|||
|
total_assets double 资产总额
|
|||
|
owners_equity double 所有者权益
|
|||
|
short_term_debt double 短期债务
|
|||
|
long_term_debt double 长期债务
|
|||
|
total_debt double 总债务
|
|||
|
operating_income double 营业收入
|
|||
|
total_profit double 利润总额
|
|||
|
net_profit double 净利润
|
|||
|
ebdita double EBITDA
|
|||
|
operating_net_cash_inflow double 经营性净现金流入
|
|||
|
receivable_turnover_times double 应收款项周转次数(次)
|
|||
|
inventory_turnover_times double 存货周转次数(次)
|
|||
|
total_asset_turnover_times double 总资产周转次数(次)
|
|||
|
cash_to_income_ratio double 现金收入比
|
|||
|
operating_margin double 营业利润率
|
|||
|
long_debt_capitalization_ratio double 长期债务资本化比率
|
|||
|
total_debt_capitalization_ratio double 总债务资本化比率
|
|||
|
asset_liability_ratio double 资产负债率
|
|||
|
current_ratio double 流动比率
|
|||
|
quick_ratio double 速动比率
|
|||
|
ebdita_interest_coverage_ratio double EBITDA利息保障倍数(倍)
|
|||
|
total_debt_ebdita double 总债务/EBITDA(倍)""",
|
|||
|
"Mods/OtherInformation2/Mods")
|
|||
|
|
|||
|
crud_code_make("company_restricted_assets",
|
|||
|
"受限资产",
|
|||
|
"""id int unsigned auto_increment
|
|||
|
company_id int 企业ID
|
|||
|
total_date date 统计截至时间
|
|||
|
name varchar(255) 受限资产明细
|
|||
|
duration varchar(255) 受限期限期限
|
|||
|
amount varchar(255) 资产账面价值
|
|||
|
reason varchar(255) 受限原因
|
|||
|
remark text 受限资产说明""",
|
|||
|
"Mods/OtherInformation2/Mods")
|
|||
|
crud_code_make("company_financial_statement",
|
|||
|
"财务说明",
|
|||
|
"""id int unsigned auto_increment
|
|||
|
company_id int 企业ID
|
|||
|
total_date date 评价截至时间
|
|||
|
assets text 资产说明
|
|||
|
debt text 负债说明
|
|||
|
rights text 权益说明""",
|
|||
|
"Mods/OtherInformation2/Mods")
|
|||
|
crud_code_make("company_credit",
|
|||
|
"授信情况",
|
|||
|
"""id int unsigned auto_increment
|
|||
|
company_id int 企业ID
|
|||
|
total_date date 统计时间
|
|||
|
credit_amount double 授信额度(万)
|
|||
|
used_amount double 使用额度(万)
|
|||
|
unused_amount double 剩余额度(万)
|
|||
|
credit_institution varchar(255) 授信机构""",
|
|||
|
"Mods/OtherInformation2/Mods")
|
|||
|
crud_code_make("company_guarantee",
|
|||
|
"提供担保",
|
|||
|
"""id int unsigned auto_increment
|
|||
|
company_id int 企业ID
|
|||
|
third_name varchar(255) 被担保方
|
|||
|
type varchar(255) 担保类型
|
|||
|
amount double 担保规模(万)
|
|||
|
duration varchar(255) 担保期限
|
|||
|
method varchar(255) 保证方式
|
|||
|
status char(2) 担保状态(存续、完成)
|
|||
|
remark text 担保说明""",
|
|||
|
"Mods/OtherInformation2/Mods")
|
|||
|
crud_code_make("company_bond",
|
|||
|
"发债情况",
|
|||
|
"""id int unsigned auto_increment
|
|||
|
company_id int 企业ID
|
|||
|
name varchar(255) 债券简称
|
|||
|
code varchar(255) 债券代码
|
|||
|
type char(20) 债券类型
|
|||
|
asset varchar(255) 发行规模
|
|||
|
publish_date date 发行日期
|
|||
|
duration varchar(255) 债券期限
|
|||
|
raising_way char(5) 募集方式
|
|||
|
due_date date 到期日期
|
|||
|
coupon varchar(255) 票面利率
|
|||
|
rate_result char(5) 债项评级
|
|||
|
rate_agency varchar(255) 评级机构
|
|||
|
status char(4) 发债状态:存续、到期偿付、违约""",
|
|||
|
"Mods/OtherInformation2/Mods")
|
|||
|
|
|||
|
crud_code_make("company_bank_lease",
|
|||
|
"银行借款",
|
|||
|
"""id int unsigned auto_increment
|
|||
|
company_id int 企业ID
|
|||
|
third_name varchar(255) 借款机构
|
|||
|
amount varchar(255) 借款金额
|
|||
|
start_date date 借款日期
|
|||
|
end_date date 到期日期
|
|||
|
coupon varchar(255) 利率
|
|||
|
method varchar(255) 借款方式(信用、抵押、保证、质押...)
|
|||
|
status char(4) 状态:存续、到期偿付、违约
|
|||
|
remark text 备注""",
|
|||
|
"Mods/OtherInformation2/Mods")
|
|||
|
|
|||
|
crud_code_make("company_trust_lease",
|
|||
|
"信托融资",
|
|||
|
"""id int unsigned auto_increment
|
|||
|
company_id int 企业ID
|
|||
|
third_name varchar(255) 借款机构
|
|||
|
amount varchar(255) 借款金额
|
|||
|
start_date date 借款日期
|
|||
|
end_date date 到期日期
|
|||
|
coupon varchar(255) 利率
|
|||
|
method varchar(255) 借款方式(信用、抵押、保证、质押...)
|
|||
|
stutus char(4) 状态:存续、到期偿付、违约
|
|||
|
remark text 备注""",
|
|||
|
"Mods/OtherInformation2/Mods")
|
|||
|
|
|||
|
crud_code_make("company_rental_lease",
|
|||
|
"租赁融资",
|
|||
|
"""id int unsigned auto_increment
|
|||
|
company_id int 企业ID
|
|||
|
third_name varchar(255) 借款机构
|
|||
|
amount varchar(255) 借款金额
|
|||
|
start_date date 借款日期
|
|||
|
end_date date 到期日期
|
|||
|
coupon varchar(255) 利率
|
|||
|
method varchar(255) 借款方式(信用、抵押、保证、质押...)
|
|||
|
stutus char(4) 状态:存续、到期偿付、违约
|
|||
|
remark text 备注""",
|
|||
|
"Mods/OtherInformation2/Mods")
|
|||
|
|
|||
|
crud_code_make("company_other_lease",
|
|||
|
"其他融资",
|
|||
|
"""id int unsigned auto_increment
|
|||
|
company_id int 企业ID
|
|||
|
type varchar(255) 融资类型
|
|||
|
third_name varchar(255) 借款机构
|
|||
|
amount double 借款金额(万)
|
|||
|
duration varchar(255) 借款期限
|
|||
|
coupon varchar(255) 利率
|
|||
|
method varchar(255) 借款方式(信用、抵押、保证、质押...)
|
|||
|
stutus char(4) 状态:存续、已兑付、违约
|
|||
|
remark text 备注""",
|
|||
|
"Mods/OtherInformation2/Mods")
|
|||
|
|
|||
|
crud_code_make("company_judicial_document",
|
|||
|
"裁判文书",
|
|||
|
"""id int unsigned auto_increment
|
|||
|
company_id int 企业ID
|
|||
|
doc_type varchar(255) 文书类型
|
|||
|
case_date date 裁判日期
|
|||
|
title varchar(255) 案件名称
|
|||
|
code varchar(255) 案号
|
|||
|
type varchar(255) 案件类型
|
|||
|
reason varchar(255) 案由
|
|||
|
amount double 案件金额(万)
|
|||
|
identity varchar(255) 案件身份
|
|||
|
result text 裁判结果
|
|||
|
court varchar(255) 审理法院
|
|||
|
publish_date date 发布日期""",
|
|||
|
"Mods/OtherInformation2/Mods")
|
|||
|
|