学步园

#coding=utf8
#登录注册功能齐了
import wx
import sqlite3
class MyFrame(wx.Frame):

    def __init__(self):
        wx.Frame.__init__(self, None, -1, 'DB EXAMPLE',pos=wx.DefaultPosition,size=(300, 150))
        panel = wx.Panel(self, -1) 
        
        usernameLabel = wx.StaticText(panel, -1, "用户名：")#设置用户名Label
        self.usernameText = wx.TextCtrl(panel, -1, "",size=(175, -1))#设置输入用户名的文本框
        self.usernameText.SetInsertionPoint(0)
        
        
        pwdLabel = wx.StaticText(panel, -1, "密码：")#设置密码的Label
        self.pwdText = wx.TextCtrl(panel, -1, "", size=(175, -1),style=wx.TE_PASSWORD)#设置密码的文本框
        
        
        loginButton=wx.Button(panel,-1,"登录")#登录按钮
        exitButton=wx.Button(panel,-1,"退出")#退出按钮
        registerButton=wx.Button(panel,-1,"注册")
        
        sizer = wx.FlexGridSizer(cols=2, hgap=6, vgap=6)#sizer设置
        sizer.AddMany([usernameLabel, self.usernameText, pwdLabel, self.pwdText,loginButton,exitButton,registerButton])#把它们都安在sizer里
        panel.SetSizer(sizer)
        
        self.Bind(wx.EVT_BUTTON, self.OnLogIn, loginButton)#登录按钮绑定事件
        self.Bind(wx.EVT_BUTTON, self.OnCloseWindow, exitButton)#退出按钮绑定事件
        self.Bind(wx.EVT_BUTTON, self.OnRegister, registerButton)#注册按钮绑定事件
        
        
        
        
#         self.buildingDB()#创建数据库和表，此语句只运行第一次，之后将其注释掉
        
    def OnLogIn(self,event):#登录方法
        self.username=self.usernameText.GetValue()
        self.password=self.pwdText.GetValue()
        username=str(self.username.strip())
        conn=sqlite3.connect('db01')
        cur=conn.cursor()
        cur.execute("SELECT password FROM table01 WHERE username='%s'"% username)
        t=cur.fetchone()[0]
        print t
        if str(self.password)==str(t):
            print 'Password is correct!'
            self.Maximize(True)#窗口最大化，意思意思主界面
        else:
            print 'failed'
    def OnCloseWindow(self,event):#关闭窗口
        self.Close()
        
#     def loginmethod(self):
#         
#         pass
    
    def buildingDB(self):#建立数据库
        conn=sqlite3.connect("db01")
        cur=conn.cursor()
        cur.execute("""
        CREATE TABLE table01(username text,password text, realname text,account text,workingdept text,phonenumber text)
        """)
        cur.execute("""INSERT INTO table01 values('zhangsan','123','zhangsan','','','')""")
        cur.execute("""INSERT INTO table01 values('lisi','123','zhangsan','','','')""")
        cur.execute("""INSERT INTO table01 values('wangwu','123','zhangsan','','','')""")
        conn.commit()
        cur.execute("""SELECT username FROM table01 WHERE username='zhangsan'""")
#         p=cur.fetchone()
#         print p
        cur.close()
         
    def OnRegister(self,event):#注册方法
        self.username=self.usernameText.GetValue()
        self.password=self.pwdText.GetValue()
        conn=sqlite3.connect("db01")
        cur=conn.cursor()
        cur.execute("INSERT INTO table01 VALUES('%s','%s','','','','')"%(self.username,self.password))
        conn.commit()
        print "Registered successfully!"
        cur.close()
        
if __name__ == '__main__':
    app = wx.PySimpleApp()
    frame = MyFrame()
    frame.Show()
    app.MainLoop()