# -*- coding: utf-8 -*-
"""
Created on Sat Dec 11 16:20:05 2021

@author: 86150
"""

import pymysql
import wx
class MyFrame(wx.Frame):
    def __init__(self,parent,id):
        wx.Frame.__init__(self,parent,id,'班级信息收集',size=(400,300))
        #创建面板
        panel = wx.Panel(self)

    #创建“保存”和“查询”按钮，并绑定事件
        self.bt_storage = wx.Button(panel,label="保存")
        self.bt_storage.Bind(wx.EVT_BUTTON,self.OnclickStorage)
        self.bt_inquire = wx.Button(panel,label ='查询')
        self.bt_inquire.Bind(wx.EVT_BUTTON,self.OnclickInquire)
        #创建文本，左对齐
        self.title =wx.StaticText(panel,label ="保存信息请输入用户的学号班级和姓名\n\t查询请输入姓名或学号")
        self.label_class =wx.StaticText(panel,label ="班级：")
        self.text_class =wx.TextCtrl(panel,style =wx.TE_LEFT)
        self.label_user =wx.StaticText(panel,label ="姓名：")
        self.text_user =wx.TextCtrl(panel,style =wx.TE_LEFT)
        self.label_number = wx.StaticText(panel,label ="学号：")
        self.text_number = wx.TextCtrl(panel,style =wx.TE_LEFT)
        #添加容器，容器中控件横向排列
        hsizer_class =wx.BoxSizer(wx.HORIZONTAL)
        hsizer_class.Add(self.label_class,proportion=0,flag=wx.ALL,border=5)
        hsizer_class.Add(self.text_class,proportion=1,flag=wx.ALL,border=5)
        hsizer_user = wx.BoxSizer(wx.HORIZONTAL)
        hsizer_user.Add(self.label_user,proportion=0,flag=wx.ALL,border=5)
        hsizer_user.Add(self.text_user,proportion=1,flag =wx.ALL,border=5)
        hsizer_number =wx.BoxSizer(wx.HORIZONTAL)
        hsizer_number.Add(self.label_number,proportion=0,flag=wx.ALL,border=5)
        hsizer_number.Add(self.text_number,proportion=1,flag=wx.ALL,border=5)
        hsizer_button =wx.BoxSizer(wx.HORIZONTAL)
        hsizer_button.Add(self.bt_storage,proportion=0,flag=wx.ALIGN_CENTER,border=5)
        hsizer_button.Add(self.bt_inquire,proportion=0,flag=wx.ALIGN_CENTER,border=5)
        #添加容器，容器中的控件纵向排列
        vsizer_all = wx.BoxSizer(wx.VERTICAL)
        vsizer_all.Add(self.title,proportion=0,flag=wx.BOTTOM |wx.TOP |wx.ALIGN_CENTER,border=15)
        vsizer_all.Add(hsizer_class,proportion=0,flag=wx.EXPAND |wx.LEFT |wx.RIGHT,border=45)
        vsizer_all.Add(hsizer_user,proportion=0,flag=wx.EXPAND |wx.LEFT |wx.RIGHT,border=45)
        vsizer_all.Add(hsizer_number,proportion=0,flag=wx.EXPAND |wx.LEFT |wx.RIGHT,border=45)
        vsizer_all.Add(hsizer_button,proportion=0,flag=wx.ALIGN_CENTER |wx.TOP,border=15)
        panel.SetSizer(vsizer_all)

    def OnclickStorage(self,event):
        "单击保存按钮"
        #连接数据库
        db =pymysql.connect(host='localhost',user='root',password='15019599752',database="huan",charset='utf8')
        message =""
        classname =self.text_class.GetValue()                   #获取输入的班级
        username =self.text_user.GetValue()                     #获取输入的用户名
        number =self.text_number.GetValue()                     #获取输入的学号
        #使用cursor()方法创建一个游标对象
        cursor = db.cursor()
        #数据列表
        data = [(classname,username,number)]
        try:
            #执行sql语句，插入数据
            sql="insert into xinji(class, name, number) values(%s,%s,%s)"
            cursor.executemany(sql,data)
            #提交数据
            db.commit()
        except:
            #发生错误时回滚
            db.rollback()
        #关闭连接
        db.close()
        #判断是否为空
        if classname =="" or username =="" or number =="":
            message ='班级或名字或密码不能为空'
        else:
            message ='保存成功'                                   #用户名或密码错误
        wx.MessageBox(message)                                   #弹出提示框

    def OnclickInquire(self,event):
        "单击查询按钮"
        username =self.text_user.GetValue()                     #获取输入的用户名
        number =self.text_number.GetValue()                     #获取输入的学号
        db =pymysql.connect(host='localhost',user='root',password='15019599752',database="huan",charset='utf8')
        #使用cursor()方法创建一个游标对象
        cursor = db.cursor()
        sql =""
        message =""
        if username:
            sql ="SELECT * FROM xinji \
                  WHERE NAME = %s"                               #找到该名字
            try:
                #执行sql语句
                cursor.execute(sql,username)
                #获取记录
                results =cursor.fetchall()
                if results:                                      #若找到不为空，输出对应的信息
                    for row in results:
                        classname = row[0]
                        username =row[1]
                        number =row[2]
                    message ="班级：{}\n姓名：{}\n学号：{}\n".format(classname,username,number)
                else:
                    message ="没有这个名字"
            except:
                print("Error: unable to fecth data")              #数据错误--
        elif number:
            sql ="SELECT * FROM ACCOUNTPASSWORD \
                  WHERE NUMBER = %s"                              #查找学号，同上
            try:
                #执行sql语句
                cursor.execute(sql,number)
                #获取记录
                results =cursor.fetchall()
                if results:
                    for row in results:
                        classname = row[0]
                        username =row[1]
                        number =row[2]
                    message ="班级：{}\n姓名：{}\n学号：{}\n".format(classname,username,number)
                else:
                    message ="没有这个学号"
            except:
                print("Error: unable to fecth data")
        else:
            message ="班级或名字或密码不能为空"
        #关闭数据库
        db.close()
        wx.MessageBox(message)
if __name__ =='__main__':
    app = wx.App()                                               #初始化应用
    frame = MyFrame(parent=None, id=-1)                          #实例MyFrame类，并传递参数
    frame.Show()                                                 #显示窗口
    app.MainLoop()                                               #调用主循环方法