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) 借款方式（信用、抵押、保证、质押...）
|
||
# status 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) 借款方式（信用、抵押、保证、质押...）
|
||
# status 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) 借款方式（信用、抵押、保证、质押...）
|
||
# status 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")
|