研发基于GUI的随机数据生成工具（V4.0完整版：设置新窗口显示数据并写入文本文件）

#V4.0完整版
#导入相关的类库文件
import random
import string
import wx
import csv
class test_tool_GUI_v4():
    #类的初始化方法，创建所有的控件
    def __init__(self):
        self.app=wx.App()
        self.window=wx.Frame(None,title="测试数据生成工具",size=(400,400))
        self.panel=wx.Panel(self.window)
        #定义标签
        self.minbq=wx.StaticText(self.panel,label="最小长度")
        #定义文本框
        self.mintxt=wx.TextCtrl(self.panel)
        #定义标签2
        self.maxbq=wx.StaticText(self.panel,label="最大长度")
        #定义文本框2
        self.maxtxt=wx.TextCtrl(self.panel)
        #定义标签3
        self.wenjianbq=wx.StaticText(self.panel,label="文件名及路径")
        #定义文本框3
        self.wenjiantxt=wx.TextCtrl(self.panel)
        #定义标签4
        self.shujubq=wx.StaticText(self.panel,label="数据总数")
        #定义文本框4
        self.shujutxt=wx.TextCtrl(self.panel)

        #定义复选框
        self.daxue=wx.CheckBox(self.panel,label="包含大写字母")
        self.xiaoxue=wx.CheckBox(self.panel,label="包含小写字母")
        self.shuzi=wx.CheckBox(self.panel,label="包含数字")
        self.fuhao=wx.CheckBox(self.panel,label="包含符号")
        self.xuhao=wx.CheckBox(self.panel,label="包含序号")
        self.youxiang= wx.CheckBox(self.panel,label="包含邮箱后缀")
        self.baocun=wx.CheckBox(self.panel,label="保存到文件")

        #定义按键
        self.queding=wx.Button(self.panel,label="确定")
        self.chongzhi=wx.Button(self.panel,label="重置")

    #设置布局BoxSizer
    def boxsizer(self):
        #设置一个水平的BoxSizer
        boxsizer1=wx.BoxSizer()
        boxsizer1.Add(self.minbq,flag=wx.EXPAND|wx.LEFT|wx.TOP,border=15)
        boxsizer1.Add(self.mintxt,flag=wx.LEFT|wx.TOP,border=15)
        boxsizer1.Add(self.maxbq,flag=wx.EXPAND|wx.LEFT|wx.TOP,border=15)
        boxsizer1.Add(self.maxtxt,flag=wx.LEFT|wx.TOP,border=15)
        #设置第二个水平的BoxSizer
        boxsizer2=wx.BoxSizer()
        boxsizer2.Add(self.daxue,flag=wx.LEFT|wx.TOP,border=15)
        boxsizer2.Add(self.xiaoxue,flag=wx.LEFT|wx.TOP, border=15)
        #设置第三个水平的BoxSizer
        boxsizer3=wx.BoxSizer()
        boxsizer3.Add(self.shuzi,flag=wx.LEFT|wx.TOP,border=15)
        boxsizer3.Add(self.fuhao,flag=wx.LEFT|wx.TOP,border=15)
        # 设置第四个水平的BoxSizer
        boxsizer4=wx.BoxSizer()
        boxsizer4.Add(self.xuhao,flag=wx.LEFT|wx.TOP,border=15)
        boxsizer4.Add(self.youxiang,flag=wx.LEFT|wx.TOP,border=15)
        # 设置第五个水平的BoxSizer
        boxsizer5=wx.BoxSizer()
        boxsizer5.Add(self.baocun,flag=wx.LEFT|wx.TOP,border=15)
        boxsizer5.Add(self.wenjianbq,flag=wx.LEFT|wx.TOP,border=15)
        boxsizer5.Add(self.wenjiantxt,flag=wx.LEFT|wx.TOP,border=15)
        # 设置第六个水平的BoxSizer
        boxsizer6=wx.BoxSizer()
        boxsizer6.Add(self.shujubq,flag=wx.LEFT|wx.TOP,border=15)
        boxsizer6.Add(self.shujutxt,flag=wx.LEFT|wx.TOP,border=15)
        # 设置第七个水平的BoxSizer
        boxsizer7=wx.BoxSizer()
        boxsizer7.Add(self.queding,flag=wx.LEFT|wx.TOP,border=15)
        boxsizer7.Add(self.chongzhi,flag=wx.LEFT|wx.TOP,border=15)

        #设置一个垂直的boxsizer
        boxsizer8=wx.BoxSizer(wx.VERTICAL)
        for i in range(1,8):
            box="boxsizer"+str(i)
            boxsizer8.Add(eval(box))

        #让boxsizer生效
        self.panel.SetSizer(boxsizer8)

    #设置新窗口显示数据
    def show_window(self,list):
        str=""
        self.windownew=wx.Frame(None,title="测试数据",size=(600, 600))
        self.panelnew=wx.Panel(self.windownew)
        for row in list:
            str=str+row+"\n"
            wx.TextCtrl(self.panelnew,value=str,style=wx.TE_MULTILINE,size=(600, 600))
            self.windownew.Show(True)

    #校验输入长度的最小值与最大值
    def check_len(self):
        # 获取控件中输入的内容
        self.min=self.mintxt.GetValue()
        self.max=self.maxtxt.GetValue()
        # 当为空
        if self.min=="":
            #给出提示不能为空
            a=wx.MessageDialog(None,"最小长度不能为空", "错误信息",wx.YES_DEFAULT|wx.ICON_QUESTION)
            # 点击确定
            if a.ShowModal()==wx.ID_YES:
                #对话框关闭
                a.Destroy()
        elif self.max=="":
            #给出提示不能为空
            a=wx.MessageDialog(None,"最大长度不能为空", "错误信息",wx.YES_DEFAULT|wx.ICON_QUESTION)
            #点击确定
            if a.ShowModal()==wx.ID_YES:
                # 对话框关闭
                a.Destroy()
        elif int(self.min)>int(self.max):
            #提示最小长度不能大于最大长度
            a=wx.MessageDialog(None,"最小长度不能大于最大长度","错误信息", wx.YES_DEFAULT|wx.ICON_QUESTION)
            #点击确定
            if a.ShowModal()==wx.ID_YES:
                # 对话框关闭
                a.Destroy()
        else:
            return 1

    #校验复选框
    def xuanze(self):
        self.list=[]
        #判断是否没有选择任何条件
        if self.daxue.GetValue()==True:
            self.list.append("daxue")
        if self.xiaoxue.GetValue()==True:
            self.list.append("xiaoxue")
        if self.shuzi.GetValue()==True:
            self.list.append("shuzi")
        if self.fuhao.GetValue()==True:
            self.list.append("fuhao")
        if self.xuhao.GetValue()==True:
            self.list.append("xuhao")
        if self.youxiang.GetValue()==True:
            self.list.append("youxiang")
        if self.list==[]:
            a=wx.MessageDialog(None,"前四项至少选择一项！","提示信息",wx.YES_DEFAULT|wx.ICON_QUESTION)
            # 点击确定
            if a.ShowModal() == wx.ID_YES:
                # 对话框关闭
                a.Destroy()
        elif self.list==["xuhao"]:
            a=wx.MessageDialog(None, "前四项至少选择一项！", "提示信息", wx.YES_DEFAULT|wx.ICON_QUESTION)
            # 点击确定
            if a.ShowModal() == wx.ID_YES:
                # 对话框关闭
                a.Destroy()
        elif self.list==["youxiang"]:
            a=wx.MessageDialog(None,"前四项至少选择一项！", "提示信息", wx.YES_DEFAULT|wx.ICON_QUESTION)
            # 点击确定
            if a.ShowModal() == wx.ID_YES:
                # 对话框关闭
                a.Destroy()
        elif self.list==["xuhao","youxiang"]:
            a=wx.MessageDialog(None,"前四项至少选择一项！","提示信息", wx.YES_DEFAULT|wx.ICON_QUESTION)
            # 点击确定
            if a.ShowModal() == wx.ID_YES:
                # 对话框关闭
                a.Destroy()
        else:
            return 1

    #校验文件名及路径是否存在并保存数据
    def savefile(self,list):
        #判断路径及文件名是否存在
        filename=self.wenjiantxt.GetValue()
        if filename=="":
            #弹出提示
            a=wx.MessageDialog(None,"请输入文件名及路径","提示信息", wx.YES_DEFAULT|wx.ICON_QUESTION)
            if a.ShowModal()==wx.ID_YES:
                a.Destroy()
                return 0
        elif filename.endswith(".csv"):
            file=open(filename,"w",newline="")
            w=csv.writer(file)
            print("传入文件中的内容",list)
            for row in list:
                w.writerow([row])
            file.close()
        else :
            #弹出提示
            a=wx.MessageDialog(None,"请输入正确的文件名及路径","提示信息", wx.YES_DEFAULT|wx.ICON_QUESTION)
            if a.ShowModal()==wx.ID_YES:
                a.Destroy()
                return 0

    #校验输入的数量
    def check_num(self):
        if self.shujutxt.GetValue()=="" or int(self.shujutxt.GetValue())<1:
            #弹出提示
            a=wx.MessageDialog(None,"请输入大于1的整数","提示信息", wx.YES_DEFAULT|wx.ICON_QUESTION)
            if a.ShowModal()==wx.ID_YES:
                a.Destroy()
                return 0
        else:
            number=int(self.shujutxt.GetValue())
            return number

    #根据选择条件创建测试数据
    def createdata(self,i):
        strmp=""
        xuhao=i
        youxiang=["@qq.com","@163.com","@126.com","@sina.com","@sogou.com"]
        num1=random.randint(int(self.min),int(self.max))
        #判断勾选的条件
        if "daxue" in self.list:
            strmp=strmp+3*string.ascii_uppercase
        if "xiaoxue" in self.list:
            strmp=strmp+3*string.ascii_lowercase
        if "shuzi" in self.list:
            strmp=strmp+5*string.digits
        if "fuhao" in self.list:
            strmp=strmp+3*string.punctuation
        result=''.join(random.sample(strmp,num1))
        #增加一个序号
        if "xuhao" in self.list:
            result=str(xuhao)+result
        if "youxiang" in self.list:
            #任意取一个邮箱后缀
            email=random.choice(youxiang)
            result=result+email
        #print(result2)
        return result

    #调用界面校验结果
    def check_input(self,event):
        list=[]
        #获取长度校验结果
        a=self.check_len()
        if a==None:
            return 0
        else:
            #获取复选框校验结果
            result=self.xuanze()
            if result==None:
                return 0
            else:
                #获取输入数量校验结果
                number=self.check_num()
                if number==None:
                    return 0
                else:
                    for i in range(1,number+1):
                        data=self.createdata(i)
                        list.append(data)
                    #调用新窗口显示数据
                    if self.baocun.GetValue()==False:
                        self.show_window(list)
                    #调用校验文件名及路径是否存在并保存数据
                    else:
                        self.savefile(list)

    #重置文本框
    def reset(self,event):
        # 提示是否清空文本框
        a=wx.MessageDialog(None,"是否清空文本框","提示信息",wx.YES_NO|wx.ICON_QUESTION)
        #点击"是"
        if a.ShowModal()==wx.ID_YES:
            #清空文本框
            self.mintxt.SetValue("")
            self.maxtxt.SetValue("")
            self.wenjiantxt.SetValue("")
            self.shujutxt.SetValue("")

            #清空勾选项
            self.daxue.SetValue(False)
            self.xiaoxue.SetValue(False)
            self.shuzi.SetValue(False)
            self.fuhao.SetValue(False)
            self.xuhao.SetValue(False)
            self.youxiang.SetValue(False)
            self.baocun.SetValue(False)
            # 对话框关闭
            a.Destroy()
        #点击“否”
        else:
            #对话框关闭
            a.Destroy()

    #将按钮和事件绑定
    def event_bind(self):
        self.queding.Bind(wx.EVT_BUTTON,self.check_input)
        self.chongzhi.Bind(wx.EVT_BUTTON,self.reset)

    def run(self):
        #激活窗体
        self.window.Show()
        # 运行APP
        self.app.MainLoop()
if __name__ == '__main__':
    test_tool_GUI_obj=test_tool_GUI_v4()
    test_tool_GUI_obj.boxsizer()
    test_tool_GUI_obj.event_bind()
    test_tool_GUI_obj.run()