urban-investment-research/CodeMakeUtils.py

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