6Ms7kpR3

1# -*- coding: utf-8 -*-
2
3import sqlite3
4from server import *
5import tkMessageBox
6
7
8class DB:
9    def insert_user(self, user_name, user_pass):
10        try:
11           c = self.conn.cursor()
12           c.execute("INSERT INTO users VALUES (?, ?)", (user_name, user_pass))
13           c.commit()
14
15        except Exception, e:
16            print (e)
17
18    def checkLogin(self, user_name, user_pass):
19        c = self.conn.cursor()
20        w = (user_name,)
21        c.execute('''SELECT * FROM users WHERE user_name=?''', w)
22        results = c.fetchone()
23
24        print w[0]
25        print results
26        if results is None:
27            tkMessageBox.showinfo("Title", "The username doesn't exist.")
28            print (" the username doesn't exist.")
29        else:
30            correctpass = results[1]
31            if correctpass == user_pass:
32                tkMessageBox.showinfo("Title", "logged in successfully")
33                print "logged in successfully "
34            else:
35                tkMessageBox.showinfo("Title", "The password doesn't match the username.")
36                print " the password doesn't match the username."
37
38    def checkSignon(self, user_name, user_pass):
39        c = self.conn.cursor()
40        w = (user_name,)
41        c.execute('''SELECT * FROM users WHERE user_name=?''', w)
42        results = c.fetchone()
43        print results
44        if results is None:
45            print (" the username isn't taken.")
46            c.execute("INSERT INTO users VALUES (?, ?)", (user_name, user_pass))
47            self.conn.commit()
48            print "user added successfully"
49            tkMessageBox.showinfo("Title", "signup successfully")
50        else:
51            tkMessageBox.showinfo("Title", "the username already exists")
52            print "the username already exists"
53
54    def __init__(self):
55        self.conn = sqlite3.connect('D:\PycharmProjects\cyberApp/example.db')
56        c = self.conn.cursor()
57        # Create table
58        c.execute('''CREATE TABLE IF NOT EXISTS users (user_name text PRIMARY KEY, password text)''')