def handleLogin():
# ask user for email and password
user_email = input("Email: ")
user_pw = getpass.getpass()
# queries
checkIfUserExists = "SELECT email FROM users WHERE email = '" + user_email + "' AND pass = '******'"
checkIfAgent = "SELECT email FROM airline_agents WHERE email = '" + user_email + "'"
# check if user exists
db = main.getDatabase()
db.execute(checkIfUserExists)
user_results = db.cursor.fetchall()
if len(user_results) > 0: # user exists!
db.execute(checkIfAgent)
agent_results = db.cursor.fetchall()
if len(agent_results) > 0:
menu = agentMenu.AgentMenu(user_email) # agents should lead to agent menu
else:
menu = userMenu.UserMenu(user_email) # normal users go to user menu
menu.showMenu()
else:
print("User does not exist or password is incorrect.")
main.showMainMenu()
def handleRegister():
# ask for email and password
print("Please register your email and password.\n")
user_email = input("Email: ")
user_pw = getpass.getpass()
# queries
checkIfUserExists = "SELECT email FROM users WHERE email = '" + user_email + "'"
addToUsers = "INSERT INTO users VALUES('" + user_email + "', '" + user_pw + "', SYSDATE)"
# check if user exists
db = main.getDatabase()
db.execute(checkIfUserExists)
user_results = db.cursor.fetchall()
if len(user_results) > 0: # user exists!
print("User already exists. Please log in.\n")
main.showMainMenu()
# valid email check from: http://stackoverflow.com/questions/8022530/python-check-for-valid-email-address
if not re.match(r"[^@]+@[^@]+\.[^@]+", user_email):
print("Not a valid email. Please try again.")
handleRegister()
if (len(user_pw) > 4):
print("Password must NOT be more than 4 characters.")
handleRegister()
else: # everything is valid! go ahead and add to user table
db.execute(addToUsers)
db.execute("commit")
print("You've successfully registered! Please log in.\n")
def handleRegister():
# ask for email and password
print("Please register your email and password.\n")
user_email = input("Email: ")
user_pw = getpass.getpass()
# queries
checkIfUserExists = "SELECT email FROM users WHERE email = '" + user_email + "'"
addToUsers = "INSERT INTO users VALUES('" + user_email + "', '" + user_pw + "', SYSDATE)"
# check if user exists
db = main.getDatabase()
db.execute(checkIfUserExists)
user_results = db.cursor.fetchall()
if len(user_results) > 0: # user exists!
print("User already exists. Please log in.\n")
main.showMainMenu()
# valid email check from: http://stackoverflow.com/questions/8022530/python-check-for-valid-email-address
if not re.match(r"[^@]+@[^@]+\.[^@]+", user_email):
print("Not a valid email. Please try again.")
handleRegister()
if (len(user_pw) > 4):
print("Password must NOT be more than 4 characters.")
handleRegister()
else: # everything is valid! go ahead and add to user table
db.execute(addToUsers)
db.execute("commit")
print("You've successfully registered! Please log in.\n")
def showMenu(self):
# user menu plus more
while True:
userInput = input("Search for flights (S) or List existing bookings (B), Logout (L)? ")
if (userInput == "S"):
m = input("Would you like to search for multiple passengers? (y/n): ")
if m == 'y' or m == 'Yes' or m =='yes':
self.promptAndSearchForFlights(True)
else:
self.promptAndSearchForFlights()
elif (userInput == "B"):
self.showExistingBookings()
self.promptForBooking()
elif (userInput == "RD"):
print("Recording flight departure. ")
(fno, date) = self.findFlight()
update = self.promptUpdate()
self.recordDep(fno, date, update)
elif (userInput == "RA"):
print("Recording flight arrival. ")
(fno, date) = self.findFlight()
update = self.promptUpdate()
self.recordArr(fno, date, update)
elif (userInput == "L"):
self.setLastLogin()
main.showMainMenu()
else:
print("Pick a valid option. \n")
def showMenu(self):
"""
This shows the user menu for searching flights, listing existing bookings or logging out.
"""
while True:
userInput = input("Search for flights (S) or List existing bookings (B), Logout (L)? ")
if (userInput == "S"):
m = input("Would you like to search for multiple passengers? (y/n): ")
if m == 'y' or m == 'Yes' or m =='yes':
self.promptAndSearchForFlights(True)
else:
self.promptAndSearchForFlights()
elif (userInput == "B"):
self.showExistingBookings()
self.promptForBooking()
elif (userInput == "L"):
self.setLastLogin()
main.showMainMenu()
else:
print("Pick a valid option. \n")
def showMenu(self):
"""
This shows the user menu for searching flights, listing existing bookings or logging out.
"""
while True:
userInput = input(
"Search for flights (S) or List existing bookings (B), Logout (L)? "
)
if (userInput == "S"):
m = input(
"Would you like to search for multiple passengers? (y/n): "
)
if m == 'y' or m == 'Yes' or m == 'yes':
self.promptAndSearchForFlights(True)
else:
self.promptAndSearchForFlights()
elif (userInput == "B"):
self.showExistingBookings()
self.promptForBooking()
elif (userInput == "L"):
self.setLastLogin()
main.showMainMenu()
else:
print("Pick a valid option. \n")
