用python生成基于lombok 和 hibernate 生成javabean

import pymysql.cursors
import sys
from contextlib import contextmanager
import traceback
import ExcelHelp as excelfrom logbook import Logger, StreamHandlerStreamHandler(sys.stdout).push_application()
log = Logger('FileOperation')"""
pymysql.Connect()参数说明
host(str): MySQL服务器地址
port(int): MySQL服务器端口号
user(str): 用户名
passwd(str): 密码
db(str): 数据库名称
charset(str): 连接编码 connection对象支持的方法
cursor() 使用该连接创建并返回游标
commit() 提交当前事务
rollback() 回滚当前事务
close() 关闭连接 cursor对象支持的方法
execute(op) 执行一个数据库的查询命令
fetchone() 取得结果集的下一行
fetchmany(size) 获取结果集的下几行
fetchall() 获取结果集中的所有行
rowcount() 返回数据条数或影响行数
close() 关闭游标对象"""# charset='utf8'
class Mysql(object):
StreamHandler(sys.stdout).push_application()
logger = Logger('Mysql')
# 连接数据库 # 生产环境数据库
__config = {
'host': '118.190.158.239',
'port': 3306,
'user': 'develop',
'password': 'B06C670823934B0B9838F96E1F8C11AC',
'db': 'faith',
'charset': 'utf8mb4',
'cursorclass': pymysql.cursors.DictCursor,
} # 数据库构造函数，从连接池中取出连接，并生成操作游标
def __init__(self):
"""
配置成员变量
"""
# 连接对象
self.__conn = self.__getConn()
self.__cursor = self.__getCursor() def __getConn(self):
"""
获取con连接
:return: con
"""
self.__conn = pymysql.Connect(**Mysql.__config)
return self.__conn def __getCursor(self):
"""
获取游标
:return: cursor
"""
self.__cursor = self.__conn.cursor()
return self.__cursor @contextmanager
def __con_cursor(self):
"""
1、定义上下文管理器，连接后自动关闭连接
2、元组对象前面如果不带“*”、字典对象如果前面不带“**”，则作为普通的对象传递参数。
:return:
"""
# 打开连接
conn = self.__getConn()
# 打开游标
cursor = self.__getCursor() try:
yield cursor
except Exception as ex:
conn.rollback()
Mysql.logger.error(repr(ex))
finally:
self.__conn.commit()
self.__cursor.close()
self.__conn.close() # ---------- 搜索 ---------- def __query(self, cursor, sql, param=None):
if param:
count = cursor.execute(sql, param)
else:
count = cursor.execute(sql)
return count def getOne(self, sql, param=None):
"""
@summary: 执行查询，并取出第一条
@param sql:查询ＳＱＬ，如果有查询条件，请只指定条件列表，并将条件值使用参数[param]传递进来
@param param: 可选参数，条件列表值（元组/列表）
@return: result list/boolean 查询到的结果集
""" with self.__con_cursor() as cursor:
if self.__query(cursor, sql, param) > 0:
result = cursor.fetchone()
else:
result = False
return result def getMany(self, sql, param=None, num=1):
"""
@summary: 执行查询，并取出num条结果
@param sql:查询ＳＱＬ，如果有查询条件，请只指定条件列表，并将条件值使用参数[param]传递进来
@param num:取得的结果条数
@param param: 可选参数，条件列表值（元组/列表）
@return: result list/boolean 查询到的结果集
"""
with self.__con_cursor() as cursor:
if self.__query(cursor, sql, param) > 0:
result = cursor.fetchmany(num)
else:
result = False
return result def getAll(self, sql, param=None):
"""
@summary: 执行查询，并取出所有结果集
@param sql:查询ＳＱＬ，如果有查询条件，请只指定条件列表，并将条件值使用参数[param]传递进来
@param param: 可选参数，条件列表值（元组/列表）
@return: result list(字典对象)/boolean 查询到的结果集
""" with self.__con_cursor() as cursor:
if self.__query(cursor, sql, param) > 0:
result = cursor.fetchall()
else:
result = False
return result def __select(self, table, cond_dict=None, order=None):
"""
@summary: 执行条件查询，并取出所有结果集
@cond_dict:{'name':'xiaoming'...}
@order:'order by id desc'
@return: result ({"col":"val","":""},{})
"""
consql = ' '
if cond_dict != '':
for k, v in cond_dict.items():
consql = consql + k + '=' + v + ' and'
consql = consql + ' 1=1 '
sql = 'select * from %s where ' % table
sql = sql + consql + order
print
'_select:' + sql
return self._exeCute(sql) # ---------- 更新 ---------- def update(self, sql, param=None):
"""
@summary: 更新数据表记录
@param sql: ＳＱＬ格式及条件，使用(%s,%s)
@param param: 要更新的 值 tuple/list
@return: count 受影响的行数
"""
with self.__con_cursor() as cursor:
return cursor.execute(sql, param) # ---------- 删除 ---------- def delete(self, sql, param=None):
"""
@summary: 删除数据表记录
@param sql: ＳＱＬ格式及条件，使用(%s,%s)
@param param: 要删除的条件 值 tuple/list
@return: count 受影响的行数
""" with self.__con_cursor() as cursor:
return cursor.execute(sql, param) # ---------- 插入 ---------- # 插入一条/多条数据
def insert(self, sql, *param):
"""
@summary: 向数据表插入一条记录
@param sql:要插入的ＳＱＬ格式
@param value:要插入的记录数据tuple/list
@return: insertId
"""
with self.__con_cursor() as cursor:
# 执行插入操作
cursor.executemany(sql, param)
# cursor.execute(sql)
# 获取最后更新的ID
return cursor.lastrowidif __name__ == '__main__':
aaa = Mysql()
# print(aaa.getOne("select * from erp_users"))
# print(aaa.getOne("select * from erp_users WHERE id in(%s)", (19,)))
# for item in aaa.getMany("select * from erp_users WHERE id in(19,39)", None, 5):
# print(item)
# for item in aaa.getAll("select name from erp_users ORDER BY `name` asc"):
# print(item)
# for item in aaa.getAll("select * from erp_users WHERE id in(%s)", (19,)):
# print(item)
print(aaa.update("UPDATE erp_users SET mail =%s WHERE id = %s", ('123@789', 19)))
# print(aaa.delete("DELETE erp_users WHERE id = %s", (19,)))
# print(aaa.getInsertId())
# 插入一条数据
# print(aaa.insert("INSERT `erp_areas` (`areaName`,`charge`,`areaCode`,`is_delete`,`commission`) VALUES (%s,%s,%s,%s,%s)", ('通州片区2222ssssssd', '片区经理3', '0', '0', '0.90')))
# 插入多条数据
# print(aaa.insert("INSERT `erp_areas` (`areaName`,`charge`,`areaCode`,`is_delete`,`commission`) VALUES (%s,%s,%s,%s,%s)", ('通州片区2222ssssssd', '片区经理3', '0', '0', '0.90'), ('通州片区2222ssssssd', '片区经理3', '0',
'0', '0.90')))

import sys
import datetime
from unipath import *sys.path.append("lib")
from MySQLHelper import Mysql
from io import StringIO
from logbook import Logger, StreamHandler# sys.path.append("lib/test")
StreamHandler(sys.stdout).push_application()
log = Logger('ProcessProdect')class StringBuilder(object):
_file_str = None def __init__(self):
self._file_str = StringIO() def Append(self, str):
self._file_str.write(str) def AppendN(self, str):
self._file_str.write(str + "\n") def __str__(self):
return self._file_str.getvalue()class CreateJaveBean: @staticmethod
def createJaveClass(databaseName, tableName, ClassNameSuffix, filepath):
# 类型映射
datatypeMapping = {"varchar": "String", "int": "Integer", "bit": "Boolean", "datetime": "Date"}
selectSql = "SELECT * FROM information_schema. COLUMNS WHERE TABLE_SCHEMA = '{schemaName}' AND table_name = '{tabName}';".format(schemaName=databaseName, tabName=tableName)
mysql = Mysql() resultset = mysql.getAll(selectSql)
log.info(resultset)
sb = StringBuilder()
# entity #####################################################################################
sb.AppendN("package com.shouxin.modules.erp.model;")
sb.AppendN("")
sb.AppendN("import lombok.*;")
sb.AppendN("import org.hibernate.annotations.DynamicUpdate;")
sb.AppendN("import javax.persistence.*;")
sb.AppendN("import java.util.*;")
sb.AppendN("")
sb.AppendN("")
sb.AppendN("/**")
sb.AppendN(" * 保留版权 : CopyRright (c) 2018-2018 于继业 版权所有")
sb.AppendN(" * 文件名称 : " + tableName + "Entity.java:")
sb.AppendN(" * 文件编号 : 1-1-1")
sb.AppendN(" * 功能描述 : ")
sb.AppendN(" * 文件版本 : " + datetime.datetime.now().strftime('%Y-%m-%d') + ".01")
sb.AppendN(" * 创建人名 : 于继业")
sb.AppendN(" * 创建日期 : " + datetime.datetime.now().strftime('%Y-%m-%d'))
sb.AppendN(" * 修改人名 :")
sb.AppendN(" * 修改日期 :")
sb.AppendN(" */")
sb.AppendN("@RequiredArgsConstructor")
sb.AppendN("@Accessors(chain = true)")
sb.AppendN("@ToString")
sb.AppendN("@EqualsAndHashCode")
sb.AppendN("@Entity")
sb.AppendN("@DynamicUpdate")
sb.AppendN("@Table(name = \"" + str(tableName).lower() + "\")")
sb.AppendN("public class " + tableName + ClassNameSuffix + " {")
sb.AppendN("") for data in resultset:
# sb.Append(data["colName"] + "\t")
# 获取注解
sb.AppendN("/** " + data["COLUMN_COMMENT"] + " */")
# PRI主键约束；
# UNI唯一约束；
# MUL可以重复。
if data["COLUMN_KEY"].strip():
sb.AppendN("@Id")
sb.AppendN("@Getter")
sb.AppendN("@Setter")
sb.AppendN("@GeneratedValue(strategy = GenerationType.IDENTITY)")
# 是否有版本控制
elif str(data["COLUMN_NAME"]).lower() == "version":
sb.AppendN("@Getter")
sb.AppendN("@Setter")
sb.AppendN("@Version")
else: # 基本格式
sb.AppendN("@Basic")
sb.AppendN("@Getter")
sb.AppendN("@Setter")
sb.AppendN("@Column(name = \"" + data["COLUMN_NAME"] + "\"" + ("" if data["IS_NULLABLE"] == "YES" else ", nullable = false") + (", length = " + str(data["CHARACTER_MAXIMUM_LENGTH"]) if data["DATA_TYPE"] == "varchar" else "") + ")")
# sb.AppendN("@Getter(onMethod = @_({" + keyinfo + ", @Column(name = \""
# + data["COLUMN_NAME"] + "\""
# + ("" if data["IS_NULLABLE"] == "YES" else ", nullable = false")
# + (", length = " + str(data["CHARACTER_MAXIMUM_LENGTH"]) if data["DATA_TYPE"] == "varchar" else "") + ")}))")
sb.AppendN("private " + datatypeMapping[data["DATA_TYPE"]] + " " + str(data["COLUMN_NAME"][0:1]).lower() + data["COLUMN_NAME"][1:] + ";")
sb.AppendN("}")
log.info(sb)
with open(Path(Path(filepath), tableName + ClassNameSuffix + '.java'), 'w+', encoding='utf-8') as file:
file.write(str(sb)) # dao层接口#####################################################################################
sb = StringBuilder()
sb.AppendN("package com.shouxin.modules.erp.dao;")
sb.AppendN("")
sb.AppendN("import com.shouxin.modules.erp.model." + tableName + "Entity;")
sb.AppendN("import org.hibernate.HibernateException;")
sb.AppendN("")
sb.AppendN("")
sb.AppendN("/**")
sb.AppendN(" * 保留版权 : CopyRright (c) 2018-2018 于继业 版权所有")
sb.AppendN(" * 文件名称 : I" + tableName + "Dao.java:")
sb.AppendN(" * 文件编号 : 1-1-1")
sb.AppendN(" * 功能描述 : ")
sb.AppendN(" * 文件版本 : " + datetime.datetime.now().strftime('%Y-%m-%d') + ".01")
sb.AppendN(" * 创建人名 : 于继业")
sb.AppendN(" * 创建日期 : " + datetime.datetime.now().strftime('%Y-%m-%d'))
sb.AppendN(" * 修改人名 :")
sb.AppendN(" * 修改日期 :")
sb.AppendN(" */")
sb.AppendN("public interface I" + tableName + "Dao extends IBaseDao<" + tableName + "Entity> {")
sb.AppendN("")
sb.AppendN("}")
with open(Path(Path(filepath), "I" + tableName + 'Dao.java'), 'w+', encoding='utf-8') as file:
file.write(str(sb)) # dao层实现#####################################################################################
sb = StringBuilder()
sb.AppendN("package com.shouxin.modules.erp.dao.impl;")
sb.AppendN("")
sb.AppendN("import org.springframework.stereotype.Repository;")
sb.AppendN("import org.hibernate.HibernateException;")
sb.AppendN("import com.shouxin.modules.erp.dao.IRoleDao;")
sb.AppendN("import com.shouxin.modules.erp.model.Role;")
sb.AppendN("")
sb.AppendN("")
sb.AppendN("/**")
sb.AppendN(" * 保留版权 : CopyRright (c) 2018-2018 于继业 版权所有")
sb.AppendN(" * 文件名称 : " + tableName + "DaoImpl.java:")
sb.AppendN(" * 文件编号 : 1-1-1")
sb.AppendN(" * 功能描述 : ")
sb.AppendN(" * 文件版本 : " + datetime.datetime.now().strftime('%Y-%m-%d') + ".01")
sb.AppendN(" * 创建人名 : 于继业")
sb.AppendN(" * 创建日期 : " + datetime.datetime.now().strftime('%Y-%m-%d'))
sb.AppendN(" * 修改人名 :")
sb.AppendN(" * 修改日期 :")
sb.AppendN(" */")
sb.AppendN("@Repository(\"" + str(tableName[0:1]).lower() + tableName[1:] + "Dao\")")
sb.AppendN("public class " + tableName + "DaoImpl extends BaseDaoImpl<" + tableName + "Entity> implements I" + tableName + "Dao {")
sb.AppendN("")
sb.AppendN("}")
with open(Path(Path(filepath), tableName + 'DaoImpl.java'), 'w+', encoding='utf-8') as file:
file.write(str(sb))if __name__ == '__main__':
CreateJaveBean.createJaveClass("faith", "UniqueCode", "Entity", r'C:\Users\Administrator\Desktop\流程信息')

package com.shouxin.modules.erp.model;import com.shouxin.annotation.model.AModelView;
import lombok.*;
import lombok.experimental.Accessors;
import org.hibernate.annotations.DynamicUpdate;import javax.persistence.*;/**
* 保留版权 : CopyRright (c) 2018-2018 于继业 版权所有
* 文件名称 : UniqueCodeEntity.java:
* 文件编号 : 1-1-1
* 功能描述 : 唯一码存储表
* 文件版本 : 2018-01-22.01
* 创建人名 : 于继业
* 创建日期 : 2018-01-22
* 修改人名 :
* 修改日期 :
*/
@RequiredArgsConstructor
@Accessors(chain = true)
@ToString
@EqualsAndHashCode
@Entity
@DynamicUpdate
@AModelView(viewName = "view_uniquecode")
@Table(name = "uniquecode")
public class UniqueCodeEntity { /**
* 唯一码
*/
@Id
@Getter
@Setter
@GeneratedValue(strategy = GenerationType.IDENTITY)
@Column(name = "ID", nullable = false)
private Integer iD;
/**
* GUID
*/
@Basic
@Getter
@Setter
@Column(name = "GUID", nullable = false, length = 36, updatable = false)
private String gUID;
/**
* 明码
*/
@Basic
@Getter
@Setter
@Column(name = "PublicCode", nullable = false, length = 36, updatable = false)
private String publicCode;
/**
* 暗码
*/
@Basic
@Getter
@Setter
@Column(name = "PrivateCode", nullable = false, length = 36, updatable = false)
private String privateCode;
/**
* 所属供应商
*/
@Basic
@Getter
@Setter
@Column(name = "VenID", nullable = false, updatable = false)
private Integer venID;
/**
* 类型
*/
@Basic
@Getter
@Setter
@Column(name = "TypeID", updatable = false)
private Integer typeID;
/**
* 状态
*/
@Basic
@Getter
@Setter
@Column(name = "StateID")
private Integer stateID;
/**
* 所属产品
*/
@Basic
@Getter
@Setter
@Column(name = "ProID", nullable = false, updatable = false)
private Integer proID;
}