wxPythonJavaWeb逆向工程实现

import wx
from pymysql import connect as pyconnect
from os.path import exists as oexist
from os import makedirs as omkdirs

class myFrame(wx.Frame):
    def __init__(self):
        wx.Frame.__init__(self,None,-1,'逆向工程',size=(400,500))
        self.panel = wx.Panel(self)
        self.connect_view()
        self.Center()
        # self.Maximize(True)

    def to_file(self,content, path):# 写入文件
        f = open(path, 'w', encoding='utf-8')
        f.write(content)
        f.close()

    def get_field_list(self, table_list):  # 从数据库中读取表结构信息
        reflect_dict = {'tinyint': 'int', 'smallint': 'int', 'mediumint': 'int',
                        'int': 'int', 'bigint': 'int', 'float': 'float', 'double': 'double',
                        'decimal': 'double', 'date': 'Date', 'time': 'Date',
                        'datetime': 'Date', 'timestamp': 'Date', 'char': 'String',
                        'varchar': 'String', 'tinytext': 'String', 'text': 'String',
                        'mediumtext': 'String', 'longtext': 'String'}
        cursor = self.db_connection.cursor()
        cursor.execute('desc ' + table_list)
        field_list = []
        for i in cursor.fetchall():
            col_name = i[0]
            col_type = i[1].split('(')[0]
            field_name = self.sql_to_camel(col_name)
            field_type = reflect_dict[col_type]
            field_list.append((field_type, field_name))
        return field_list

    def sql_to_camel(self,string):  # 下划线形式命名转驼峰式命名
        words = string.split('_')
        name = words[0]
        if len(words) > 1:
            for i in words[1:]:
                name += self.first_upper(i)
        return name

    def first_upper(self,string):  # 字符串首字母大写
        return string[0].upper() + string[1:]

    def connect_view(self):# 连接数据库视图
        self.label1 = wx.StaticText(self.panel,-1,'用户名:',(50,50))
        self.username = wx.TextCtrl(self.panel,-1,'root',(120,50),(200,25))
        self.label2 = wx.StaticText(self.panel, -1, '密码:',(50,100))
        self.password = wx.TextCtrl(self.panel,-1,'root',(120,100),(200,25))
        self.label3 = wx.StaticText(self.panel, -1, '数据库名:',(50,150))
        self.db = wx.TextCtrl(self.panel, -1,pos=(120, 150),size=(200,25))
        self.label4 = wx.StaticText(self.panel, -1, '实体包:', (50, 200))
        self.pojo = wx.TextCtrl(self.panel, -1, 'com.shy.pojo', (120, 200),(200,25))
        self.label5 = wx.StaticText(self.panel, -1, '接口包:', (50, 250))
        self.mapper = wx.TextCtrl(self.panel, -1, 'com.shy.mapper', (120, 250),(200,25))
        self.label6 = wx.StaticText(self.panel, -1, 'XML包:', (50, 300))
        self.xml = wx.TextCtrl(self.panel, -1, 'com.shy.mapperxml', pos=(120, 300),size=(200,25))
        self.next = wx.Button(self.panel, -1, '下一步', (150, 380))
        self.next.Bind(wx.EVT_BUTTON, self.next_e)

    def table_view(self):# 选取数据表视图
        size = self.panel.GetSize()
        self.panel.Destroy()
        self.scroll = wx.ScrolledWindow(self,-1)
        self.scroll.SetSize(size[0], size[1])

        self.select_all_btn = wx.ToggleButton(self.scroll,-1,'全选',(50,10))

        self.select_all_btn.Bind(wx.EVT_TOGGLEBUTTON,self.select_all_e)

        self.next = wx.Button(self.scroll, -1, '创建', (150, 10))

        self.next.Bind(wx.EVT_BUTTON,self.generate_e)

        self.btn = wx.Button(self.scroll, -1, '清空类名', (250, 10))
        self.btn.Bind(wx.EVT_BUTTON, self.clear_tc_e)

        self.label2 = wx.StaticText(self.scroll, -1, '勾选数据表名', (50, 50))
        self.label3 = wx.StaticText(self.scroll, -1, '输入类名', (250, 50))
        cursor = self.db_connection.cursor()
        cursor.execute('show tables')
        data = cursor.fetchall()
        self.tables = []
        for i,j in enumerate(data):
            class_name = self.first_upper(self.sql_to_camel(j[0]))
            self.tables.append((wx.CheckBox(self.scroll , -1 , j[0] , (50,27*i+80)),
                                    wx.TextCtrl(self.scroll, -1, class_name, (200, 27 * i + 75),(150,25))))
        self.scroll.SetScrollbars(0, 1, 0, len(data)*27+120)

    def clear_tc_e(self,e):
        for i in self.tables:
            i[1].SetValue('')

    def create_dirs(self):# 建立文件夹
        path_list = []
        for i in self.packages:
            path = i.replace('.', '\\')
            if not oexist(path):
                omkdirs(path)
            path_list.append(path)
        return path_list

    def create_pojo(self,fields, class_name):  # 创建java实体类
        flag = False
        for i in fields:
            if i[0] == 'Date':
                flag = True
                break
        class_str = 'package ' + self.packages[0] + ';\n\n'
        if flag:
            class_str = 'import java.util.Date;\n\n'
        class_str += 'import org.springframework.stereotype.Component;\n\n@Component\npublic class ' + class_name + '{\n'  # 类名
        for i in fields:
            class_str += '\tprivate ' + i[0] + ' ' + i[1] + ';\n'  # 私有属性
        class_str += '\n\tpublic ' + class_name + '('
        for i in fields:
            class_str += i[0] + ' ' + i[1] + ','  # 构造函数参数
        class_str = class_str[:-1] + '){\n'
        for i in fields:
            class_str += '\t\tthis.' + i[1] + ' = ' + i[1] + ';\n'  # 构造函数传参
        class_str += '\t}\n\n\tpublic ' + class_name + '(){}\n\n'  # 无参构造
        for i in fields:  # gettr和setter方法
            class_str += '\tpublic ' + i[0] + ' get' + self.first_upper(i[1]) + '(){\n\t\treturn ' \
                         + i[1] + ';\n\t}\n\n\tpublic void set' + self.first_upper(i[1]) + '(' + i[0] + \
                         ' ' + i[1] + '){\n\t\tthis.' + i[1] + ' = ' + i[1] + ';\n\t}\n\n'
        class_str += '\tpublic String toString(){\n\t\treturn \"' + class_name + \
                     '{\" +\n'
        to_string = ''
        for i in fields:  # toString方法
            if i[0] == 'int':
                to_string += '\t\t\t\",' + str(i[1]) + '=\" + ' + str(i[1]) + ' +\n'
            elif i[0] == 'String':
                to_string += '\t\t\t\", ' + str(i[1]) + '=\'\" + ' + str(i[1]) + ' + \'\\\'\' +\n'
        class_str += '\t\t\t\"' + to_string[5:] + '\t\t\t\'}\';\n\t}\n}'
        file_name = self.packages[0].replace('.', '\\') + '\\' + class_name + '.java'
        self.to_file(class_str, file_name)

    def create_interface(self,class_name):# 创建mapper接口
        class_str = 'package ' + self.packages[1] + ';\n\npublic interface ' + class_name + '{\n\n}'
        file_name = self.packages[1].replace('.', '\\') + '\\' + class_name + '.java'
        self.to_file(class_str, file_name)

    def create_xml(self,class_name):# 创建SQL映射文件
        xml_str = '<?xml version=\"1.0\" encoding=\"UTF-8\" ?>' \
                  '\n<!DOCTYPE mapper ' \
                  'PUBLIC \"-//mybatis.org//DTD Mapper 3.0//EN\" ' \
                  '\n\"http://mybatis.org/dtd/mybatis-3-mapper.dtd\" >' \
                  '\n\n<mapper namespace=\"' + self.packages[1] + '.' +class_name+'Mapper' + '\">\n\n</mapper>'
        file_name = self.packages[2].replace('.', '\\') + '\\' + class_name + 'XML.xml'
        self.to_file(xml_str, file_name)

    def generate_e(self,e):# 生成相应文件事件
        self.create_dirs()  # 建立文件夹
        for i in self.tables:
            if i[0].GetValue() == True:
                class_name = i[1].GetValue() # 类名
                field_list = self.get_field_list(i[0].GetLabel()) # 所有的属性值和类型
                self.create_pojo(field_list,class_name)# 创建实体类
                self.create_interface(class_name)# 创建接口
                self.create_xml(class_name) # 创建xml文件
        self.db_connection.close() # 关闭数据库连接
        wx.MessageBox('成功!')# 弹出提示信息
        self.Destroy()

    def select_all_e(self,e):# 全选和全不选功能切换事件
        value = self.select_all_btn.GetValue()
        if(value):
            self.select_all_btn.SetLabel('全不选')
            for i in self.tables:
                i[0].SetValue(True)
        else:
            self.select_all_btn.SetLabel('全选')
            for i in self.tables:
                i[0].SetValue(False)

    def next_e(self,e):# 读取表名并获取连接事件
        self.db_connection = pyconnect('localhost',
                                             self.username.GetValue(),
                                             self.password.GetValue(),
                                             self.db.GetValue())# 获取数据库连接对象
        # 包名表
        self.packages = [self.pojo.GetValue(), self.mapper.GetValue(), self.xml.GetValue()]
        # 销毁界面不需要用到的一些组件
        self.label1.Destroy()
        self.label2.Destroy()
        self.label3.Destroy()
        self.label4.Destroy()
        self.label5.Destroy()
        self.label6.Destroy()
        self.username.Destroy()
        self.password.Destroy()
        self.db.Destroy()
        self.pojo.Destroy()
        self.mapper.Destroy()
        self.xml.Destroy()
        self.next.Destroy()
        self.table_view()

if __name__=='__main__':
    app = wx.App()
    frame = myFrame()
    frame.Show()
    app.MainLoop()
发布了23 篇原创文章 · 获赞 1 · 访问量 303
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 深蓝海洋 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览