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")
|
||
|