python 根据 数据库创建java 文件

#coding=utf-8

import pymysql

import os

import re

# 包全路径

packagepath=r'E:\idea工程\dc-exam\dc-exam\src\main\java\org\dcexam\cms\module\entity'.decode('utf-8')

# 包名称

packageName="org.dcexam.cms.module.entity"

#数据表前缀

prefix='exam_'

#数据表名称 ,传入一个list 集合  请注意，如果有重名的表 那么 会 出错误

#就算不再同一个数据库也会出错。。。。请谨慎操作

tableNames=["exam_history"]

host="localhost"

user="root"

password="root"

db="exam"

for tableName in tableNames:

        # capitalize 可以将首字母大写

    fileName=tableName.replace(prefix,"").capitalize()+".java"

    print fileName

    conn = pymysql.connect(host=host,user=user,password=password,database=db,charset="utf8")

    with conn.cursor() as cursor:

        cursor.execute('SELECT COLUMN_NAME, COLUMN_TYPE, COLUMN_COMMENT FROM information_schema.columns WHERE TABLE_NAME="'+tableName+'"')

        rs = cursor.fetchall()

        with open(os.path.join(packagepath,fileName),'w+') as file:

            file.writelines('''package '''+packageName+''';

import lombok.Data;

import javax.persistence.GeneratedValue;

import javax.persistence.Id;

import javax.persistence.Table;

@Data

@Table(name = "'''+tableName+'''")

public class '''+fileName[:-5]+'''{\n''')

            for r in rs:

                col=r[0].encode('utf-8')  # 字段名

                types=r[1].encode('utf-8') # 类型

                javaType=''

                comment=r[2].encode('utf-8') # 注视

                if types.find("tinyint(1)")!=-1:

                    javaType="Boolean "

                elif types.find("int")!=-1:

                    javaType="Integer "

                elif types.find("date")!=-1:

                    javaType="String "

                elif types.find("blob")!=-1:

                    javaType='String '

                elif types.find("text")!=-1:

                    javaType='String '

                elif types.find("varchar")!=-1:

                    javaType='String '

                elif types.find("char")!=-1:

                    javaType='String '

                elif types.find("float")!=-1:

                    javaType="Float "

                elif types.find("double")!=-1:

                    javaType="Double "

                if(col=='id' or re.compile(r'[a-zA-Z0-9]id').search(col)):

                    line='\t@Id\n\t@GeneratedValue(generator="JDBC")\n\t'+javaType+col+' ; //'+comment+"\n"

                else:

                    line= '''    '''+javaType+col+' ; //'+comment

                file.writelines(line+'\n')

            file.writelines('}')