# -*- coding: utf-8 -*- import re from typing import List from app.common.constant import GenConstant from app.config.setting import settings from app.utils.string_util import StringUtil from app.api.v1.module_generator.gencode.schema import GenTableOutSchema, GenTableSchema, GenTableColumnSchema class GenUtils: """代码生成器工具类""" @classmethod def init_table(cls, gen_table: GenTableSchema) -> None: """ 初始化表信息 参数: - gen_table (GenTableSchema): 业务表对象。 返回: - None """ # 只有当字段为None时才设置默认值 gen_table.class_name = cls.convert_class_name(gen_table.table_name or "") gen_table.package_name = settings.package_name gen_table.module_name = settings.package_name.split('.')[-1] gen_table.business_name = gen_table.table_name.split('_')[-1] gen_table.function_name = re.sub(r'(?:表|测试)', '', gen_table.table_comment or "") @classmethod def init_column_field(cls, column: GenTableColumnSchema, table: GenTableOutSchema) -> None: """ 初始化列属性字段 参数: - column (GenTableColumnSchema): 业务表字段对象。 - table (GenTableOutSchema): 业务表对象。 返回: - None """ data_type = cls.get_db_type(column.column_type or "") column_name = column.column_name or "" column.table_id = table.id column.python_field = cls.to_camel_case(column_name) # 只有当python_type为None时才设置默认类型 column.python_type = StringUtil.get_mapping_value_by_key_ignore_case(GenConstant.DB_TO_PYTHON, data_type) # 查询类型：优先根据字段语义（如以name结尾走LIKE），否则默认EQ column.query_type = GenConstant.QUERY_LIKE # 确保is_pk等字段为字符串格式 # 将布尔值或其他类型转换为字符串'1'或'0' if column.is_pk is not None and not isinstance(column.is_pk, str): column.is_pk = '1' if bool(column.is_pk) else '0' if column.is_increment is not None and not isinstance(column.is_increment, str): column.is_increment = '1' if bool(column.is_increment) else '0' if column.is_required is not None and not isinstance(column.is_required, str): column.is_required = '1' if bool(column.is_required) else '0' if column.is_unique is not None and not isinstance(column.is_unique, str): column.is_unique = '1' if bool(column.is_unique) else '0' # 确保None值默认为'0' column.is_pk = column.is_pk or '0' column.is_increment = column.is_increment or '0' column.is_required = column.is_required or '0' column.is_unique = column.is_unique or '0' # 确保column_length和column_default字段有默认值 if column.column_length is None: column.column_length = '' if column.column_default is None: column.column_default = '' if column.html_type is None: if cls.arrays_contains(GenConstant.COLUMNTYPE_STR, data_type) or cls.arrays_contains( GenConstant.COLUMNTYPE_TEXT, data_type ): # 字符串长度超过500设置为文本域 column_length = cls.get_column_length(column.column_type or "") html_type = ( GenConstant.HTML_TEXTAREA if column_length >= 500 or cls.arrays_contains(GenConstant.COLUMNTYPE_TEXT, data_type) else GenConstant.HTML_INPUT ) column.html_type = html_type elif cls.arrays_contains(GenConstant.COLUMNTYPE_TIME, data_type): column.html_type = GenConstant.HTML_DATETIME elif cls.arrays_contains(GenConstant.COLUMNTYPE_NUMBER, data_type): column.html_type = GenConstant.HTML_INPUT elif column_name.lower().endswith("status"): column.html_type = GenConstant.HTML_RADIO elif column_name.lower().endswith("type") or column_name.lower().endswith("sex"): column.html_type = GenConstant.HTML_SELECT elif column_name.lower().endswith("image"): column.html_type = GenConstant.HTML_IMAGE_UPLOAD elif column_name.lower().endswith("file"): column.html_type = GenConstant.HTML_FILE_UPLOAD elif column_name.lower().endswith("content"): column.html_type = GenConstant.HTML_EDITOR else: column.html_type = GenConstant.HTML_INPUT # 只有当is_insert为None时才设置插入字段（默认所有字段都需要插入） if column.is_insert is None: column.is_insert = GenConstant.REQUIRE else: # 确保is_insert为字符串格式，并且值为'0'或'1' column.is_insert = '1' if (column.is_insert is True or str(column.is_insert).lower() in ('1', 'true', 'yes')) else '0' # 只有当is_edit为None时才设置编辑字段 if column.is_edit is None: if not cls.arrays_contains(GenConstant.COLUMNNAME_NOT_EDIT, column_name) and column.is_pk != '1': column.is_edit = GenConstant.REQUIRE else: column.is_edit = '0' else: # 确保is_edit为字符串格式，并且值为'0'或'1' column.is_edit = '1' if (column.is_edit is True or str(column.is_edit).lower() in ('1', 'true', 'yes')) else '0' # 只有当is_list为None时才设置列表字段 if column.is_list is None: if not cls.arrays_contains(GenConstant.COLUMNNAME_NOT_LIST, column_name) and column.is_pk != '1': column.is_list = GenConstant.REQUIRE else: column.is_list = '0' else: # 确保is_list为字符串格式，并且值为'0'或'1' column.is_list = '1' if (column.is_list is True or str(column.is_list).lower() in ('1', 'true', 'yes')) else '0' # 只有当is_query为None时才设置查询字段 if column.is_query is None: if not cls.arrays_contains(GenConstant.COLUMNNAME_NOT_QUERY, column_name) and column.is_pk != '1': column.is_query = GenConstant.REQUIRE else: column.is_query = '0' else: # 确保is_query为字符串格式，并且值为'0'或'1' column.is_query = '1' if (column.is_query is True or str(column.is_query).lower() in ('1', 'true', 'yes')) else '0' @classmethod def arrays_contains(cls, arr: List[str], target_value: str) -> bool: """ 校验数组是否包含指定值 param arr: 数组 param target_value: 需要校验的值 :return: 校验结果 """ return target_value in arr @classmethod def convert_class_name(cls, table_name: str) -> str: """ 表名转换成 Python 类名 参数: - table_name (str): 业务表名。 返回: - str: Python 类名。 """ auto_remove_pre = settings.auto_remove_pre table_prefix = settings.table_prefix if auto_remove_pre and table_prefix: search_list = table_prefix.split(',') table_name = cls.replace_first(table_name, search_list) return StringUtil.convert_to_camel_case(table_name) @classmethod def replace_first(cls, input_string: str, search_list: List[str]) -> str: """ 批量替换前缀 参数: - input_string (str): 需要被替换的字符串。 - search_list (List[str]): 可替换的字符串列表。 返回: - str: 替换后的字符串。 """ for search_string in search_list: if input_string.startswith(search_string): return input_string.replace(search_string, '', 1) return input_string @classmethod def get_db_type(cls, column_type: str) -> str: """ 获取数据库类型字段 参数: - column_type (str): 字段类型。 返回: - str: 数据库类型。 """ if '(' in column_type: return column_type.split('(')[0] return column_type @classmethod def get_column_length(cls, column_type: str) -> int: """ 获取字段长度 参数: - column_type (str): 字段类型，例如 'varchar(255)' 或 'decimal(10,2)' 返回: - int: 字段长度（优先取第一个长度值，无法解析时返回0）。 """ if '(' in column_type: length = len(column_type.split('(')[1].split(')')[0]) return length return 0 @classmethod def split_column_type(cls, column_type: str) -> List[str]: """ 拆分列类型 参数: - column_type (str): 字段类型。 返回: - List[str]: 拆分结果。 """ if '(' in column_type and ')' in column_type: return column_type.split('(')[1].split(')')[0].split(',') return [] @classmethod def to_camel_case(cls, text: str) -> str: """ 将字符串转换为驼峰命名 param text: 需要转换的字符串 :return: 驼峰命名 """ parts = text.split('_') return parts[0] + ''.join(word.capitalize() for word in parts[1:])