Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import sqlite3
- status= ""
- def displayMenu():
- status= input("Are you a registered user? y/n? Press q to quit: ")
- if status.lower()== "y":
- currentUser()
- elif status.lower()=="n":
- newUser()
- def currentUser():
- conn= sqlite3.connect('OS_Employee.db')
- 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 create\n")
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement