# -*- coding: utf-8 -*- """按数据库表结构生成 Java 实体 + Mapper 骨架。 生成规则与项目既有实体保持一致： - 表名为全名（MyBatis-Plus 的 table-prefix 对显式 @TableName 无效） - int/tinyint -> Integer，bigint -> Long，decimal -> BigDecimal（ToStringSerializer，PHP 输出字符串） - double -> Double（PhpFloatSerializer，对齐 PHP 1.0 -> 1） - create_time / update_time -> @JsonIgnore Integer + 输出格式化字符串的 getter - is_delete -> @TableLogic(value="0", delval="1")（本库统一 0/1） """ import re import sys import pymysql from pathlib import Path from _root import ROOT YML = str(ROOT / "yoshop-java/src/main/resources/application.yml") ROOT = Path(str(ROOT / "yoshop-java/src/main/java/cn/yoshop")) pwd = re.findall(r"password:\s*(.+)", open(YML, encoding="utf-8").read())[0].strip().strip("\"'") # 表名 -> (类名, 说明, 需转义的列) TABLES = { "yoshop_category": ("Category", "商品分类"), "yoshop_goods_service": ("GoodsService", "商品服务（保障服务）"), "yoshop_upload_group": ("UploadGroup", "文件库分组"), "yoshop_region": ("Region", "地区（省/市/区）"), "yoshop_user": ("User", "商城会员"), "yoshop_user_grade": ("UserGrade", "会员等级"), "yoshop_page": ("PageEntity", "页面（首页/自定义页）"), "yoshop_article_category": ("ArticleCategory", "文章分类"), "yoshop_article": ("Article", "文章"), "yoshop_help": ("Help", "帮助"), "yoshop_coupon": ("Coupon", "优惠券"), "yoshop_delivery": ("Delivery", "配送模板"), "yoshop_express": ("Express", "物流公司"), "yoshop_printer": ("Printer", "打印机"), "yoshop_store_address": ("StoreAddress", "商城地址（发货/退货）"), "yoshop_store_menu": ("StoreMenu", "后台菜单"), "yoshop_store_role": ("StoreRole", "角色"), "yoshop_store_role_menu": ("StoreRoleMenu", "角色-菜单关联"), "yoshop_store_user_role": ("StoreUserRole", "管理员-角色关联"), "yoshop_store_setting": ("StoreSetting", "商城设置项"), "yoshop_comment": ("Comment", "商品评价"), "yoshop_order": ("Order", "订单"), "yoshop_order_goods": ("OrderGoods", "订单商品"), "yoshop_order_address": ("OrderAddress", "订单收货地址"), "yoshop_payment_trade": ("PaymentTrade", "第三方支付交易"), "yoshop_order_delivery": ("OrderDelivery", "订单发货单"), "yoshop_user_balance_log": ("UserBalanceLog", "余额变动记录"), "yoshop_recharge_plan": ("RechargePlan", "充值套餐"), "yoshop_order_refund": ("OrderRefund", "订单退款单"), "yoshop_delivery_rule": ("DeliveryRule", "配送模板规则"), } RESERVED = {"describe", "values", "key", "interval", "condition", "order", "group", "system"} INT_TYPES = ("int", "tinyint", "smallint", "mediumint") def camel(name, upper_first=False): parts = name.split("_") out = "".join(p[:1].upper() + p[1:] for p in parts) if not upper_first: out = out[:1].lower() + out[1:] return out def java_type(col_type): t = col_type.lower() # 形如 "int(10) unsigned" -> "int"；"decimal(10,2) unsigned" -> "decimal" base = t.split("(")[0].split()[0] if base == "bigint": return "Long" if base in INT_TYPES: return "Integer" if base == "decimal": return "BigDecimal" if base in ("double", "float"): return "Double" return "String" def gen_entity(table, cls, comment, cols): pk = None body = [] imports = { "import com.baomidou.mybatisplus.annotation.IdType;", "import com.baomidou.mybatisplus.annotation.TableId;", "import com.baomidou.mybatisplus.annotation.TableName;", "import lombok.Data;", } if any(c[1].lower().startswith("decimal") for c in cols): imports.add("import com.fasterxml.jackson.databind.annotation.JsonSerialize;") imports.add("import com.fasterxml.jackson.databind.ser.std.ToStringSerializer;") imports.add("import java.math.BigDecimal;") if any(c[1].lower().startswith("double") for c in cols): imports.add("import com.fasterxml.jackson.databind.annotation.JsonSerialize;") imports.add("import cn.yoshop.common.util.PhpFloatSerializer;") has_delete = any(c[0] == "is_delete" for c in cols) has_time = any(c[0] in ("create_time", "update_time") for c in cols) if has_delete: imports.add("import com.baomidou.mybatisplus.annotation.TableLogic;") for name, ctype, extra, cmt in cols: jt = java_type(ctype) field = camel(name) base = ctype.lower().split("(")[0].split()[0] if name == "id" and table == "yoshop_region": pk = name body.append(" @TableId(value = \"id\", type = IdType.AUTO)") body.append(f" private {jt} {field};") body.append("") continue if name.endswith("_id") and pk is None and name == table.replace("yoshop_", "") + "_id": pk = name body.append(f" @TableId(value = \"{name}\", type = IdType.AUTO)") elif name.endswith("_id") and pk is None and name in ( "category_id", "service_id", "group_id", "grade_id", "page_id", "article_id", "help_id", "coupon_id", "delivery_id", "express_id", "printer_id", "address_id", "menu_id", "role_id", "comment_id", "order_id", "order_goods_id", "trade_id", "log_id", "plan_id", "refund_id", "rule_id"): pk = name body.append(f" @TableId(value = \"{name}\", type = IdType.AUTO)") elif name in ("create_time", "update_time"): body.append(f" @JsonIgnore") body.append(f" private Integer {field};") body.append("") continue elif name == "is_delete": body.append(" @TableLogic(value = \"0\", delval = \"1\")") elif name in RESERVED: body.append(f" @TableField(\"`{name}`\")") imports.add("import com.baomidou.mybatisplus.annotation.TableField;") annotations = [] if jt == "BigDecimal": annotations.append(" @JsonSerialize(using = ToStringSerializer.class)") elif jt == "Double": annotations.append(" @JsonSerialize(using = PhpFloatSerializer.class)") for a in annotations: body.append(a) body.append(f" private {jt} {field};" + (f" // {cmt}" if cmt else "")) body.append("") # 时间字段输出 if has_time: imports.add("import com.fasterxml.jackson.annotation.JsonIgnore;") imports.add("import com.fasterxml.jackson.annotation.JsonProperty;") imports.add("import cn.yoshop.common.util.TimeUtil;") if any(c[0] == "create_time" for c in cols): body.append(" @JsonProperty(\"create_time\")") body.append(" public String getCreateTimeText() {") body.append(" return TimeUtil.format(createTime);") body.append(" }") body.append("") if any(c[0] == "update_time" for c in cols): body.append(" @JsonProperty(\"update_time\")") body.append(" public String getUpdateTimeText() {") body.append(" return TimeUtil.format(updateTime);") body.append(" }") body.append("") src = "package cn.yoshop.common.entity;\n\n" src += "\n".join(sorted(imports)) + "\n\n" src += f"/**\n * {comment}（{table}），字段与库内结构一一对应。\n */\n" src += "@Data\n" src += f"@TableName(\"{table}\")\n" src += f"public class {cls} {{\n\n" src += "\n".join(body).rstrip() + "\n" src += "}\n" (ROOT / "common" / "entity" / f"{cls}.java").write_text(src, encoding="utf-8") mapper = ( "package cn.yoshop.common.mapper;\n\n" "import cn.yoshop.common.entity." + cls + ";\n" "import com.baomidou.mybatisplus.core.mapper.BaseMapper;\n" "import org.apache.ibatis.annotations.Mapper;\n\n" f"@Mapper\npublic interface {cls}Mapper extends BaseMapper<{cls}> {{\n}}\n" ) (ROOT / "common" / "mapper" / f"{cls}Mapper.java").write_text(mapper, encoding="utf-8") def main(): conn = pymysql.connect(host="192.168.1.177", port=3306, user="root", password=pwd, database="shop") cur = conn.cursor() for table, (cls, comment) in TABLES.items(): cur.execute( "SELECT column_name,column_type,extra,column_comment FROM information_schema.columns " "WHERE table_schema='shop' AND table_name=%s ORDER BY ordinal_position", (table,)) cols = cur.fetchall() if not cols: print("!! 表不存在:", table) continue gen_entity(table, cls, comment, cols) print("ok", cls, len(cols), "列") conn.close() if __name__ == "__main__": main()