Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import sqlite3
- def currentUser():
- conn = sqlite3.connect('OS_Employee.db')
- print(conn.fetchall())
- userEmail = input("Enter login name: ")
- password = input("Enter password: ")
- with conn:
- cur = conn.cursor()
- try:
- cur.execute("SELECT*FROM employee WHERE(email='" + userEmail + "'AND password= '" + password + "')")
- results = cur.fetchall()
- rowCounter = 0
- for row in results:
- rowCounter += 1
- if results:
- for i in results:
- print("Login successful")
- loginSuccess = True
- else:
- print("Login unsucessful")
- except:
- print("Connection failed")
- def newUser():
- try:
- createID = int(input("Enter employee ID: "))
- except ValueError:
- print("\nenter a valid employee ID\n")
- with sqlite3.connect('OS_Employee.db') as db:
- user = db.cursor()
- if createID in user:
- print("ID already exists")
- else:
- createLogin = input("Enter your email: ")
- if createLogin in user:
- print("email already exist")
- else:
- createPass = input("Enter new password: ")
- user[createLogin] = createPass
- print("\nNew user created\n")
- status = input("Are you a registered user? y/n? Press q to quit: ")
- if status.lower() == "y":
- currentUser()
- elif status.lower() == "n":
- newUser()
- elif status.lower() == "q":
- quit()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement