def run(self):
#===========================================================================
# Initialize db object and connect to database
#===========================================================================
# Init the db object
db = DB(self.dbHost, self.dbDatabase, self.dbTable, self.dbUser, self.dbPass)
# Connect to the remote database server
loginStatus = db.connect()
self.postLoginSignal.emit(loginStatus, db)
def run(self):
#===========================================================================
# Initialize db object and connect to database
#===========================================================================
# Init the db object
db = DB(self.dbHost, self.dbDatabase, self.dbTable, self.dbUser,
self.dbPass)
# Connect to the remote database server
loginStatus = db.connect()
self.postLoginSignal.emit(loginStatus, db)
def start(self):
#===========================================================================
# Main function - start connection to db then open main menu
#===========================================================================
try:
while 1:
# Get DB info
self.getDbInfo()
# Create the DB object
self.db = DB(self.dbHost, self.dbName, self.dbUsersTable, self.dbVisitsTable, self.dbUser, self.dbPass)
# Connect to the database
connectStatus = self.connectToDatabase()
# If we failed to connect to the database offer to re-enter db info
if connectStatus != c.SUCCESS:
reenter = input("Failed to connect to database. Re-enter database info? (Y,n) ")
if reenter.lower() == "n":
print("Bye.")
sys.exit(0)
else:
break
# Start the main menu loop
self.displayMenu()
except KeyboardInterrupt:
pass
finally:
print("Cleaning up and exiting...")
if self.db is not None:
self.db.close()
def start(self):
#===========================================================================
# Main function - start connection to db then open main menu
#===========================================================================
try:
while 1:
# Get DB info
self.getDbInfo()
# Create the DB object
self.db = DB(self.dbHost, self.dbName, self.dbUsersTable,
self.dbVisitsTable, self.dbUser, self.dbPass)
# Connect to the database
connectStatus = self.connectToDatabase()
# If we failed to connect to the database offer to re-enter db info
if connectStatus != c.SUCCESS:
reenter = input(
"Failed to connect to database. Re-enter database info? (Y,n) "
)
if reenter.lower() == "n":
print("Bye.")
sys.exit(0)
else:
break
# Start the main menu loop
self.displayMenu()
except KeyboardInterrupt:
pass
finally:
print("Cleaning up and exiting...")
if self.db is not None:
self.db.close()
class TextUI:
def __init__(self):
self.db = None
self.tools = Utils()
def start(self):
#===========================================================================
# Main function - start connection to db then open main menu
#===========================================================================
try:
while 1:
# Get DB info
self.getDbInfo()
# Create the DB object
self.db = DB(self.dbHost, self.dbName, self.dbUsersTable,
self.dbVisitsTable, self.dbUser, self.dbPass)
# Connect to the database
connectStatus = self.connectToDatabase()
# If we failed to connect to the database offer to re-enter db info
if connectStatus != c.SUCCESS:
reenter = input(
"Failed to connect to database. Re-enter database info? (Y,n) "
)
if reenter.lower() == "n":
print("Bye.")
sys.exit(0)
else:
break
# Start the main menu loop
self.displayMenu()
except KeyboardInterrupt:
pass
finally:
print("Cleaning up and exiting...")
if self.db is not None:
self.db.close()
def displayMenu(self):
#===========================================================================
# Display main cli and take appropriate actions
#===========================================================================
#print("\nType \"back\" at any time to go up a menu level.")
while 1:
# Display main menu
print("\n\t1.) Check-in\n\t2.) Show Visits\n\t3.) Exit")
try:
option = input("\n>> ")
if option == "1":
self.checkIn()
elif option == "2":
self.showVisits()
elif option == "3":
sys.exit(0)
#elif option == "back" or option == "exit":
# exit = input("Exit? (y,N) ")
# if exit.lower() == "y":
# sys.exit(0)
else:
self.invalidInput()
except ValueError:
self.invalidInput()
def connectToDatabase(self):
#===========================================================================
# Open connection to database and return status
#===========================================================================
# Use stdout.write to prevent newline
sys.stdout.write("Connecting to database...")
# Connect to the DB!
status = self.db.connect()
if status == c.SUCCESS:
print("done.")
return status
elif status == c.BAD_PASSWD:
print("\nError connecting to database: Bad username or password.")
return status
else:
print("\nUnknown Error connecting to database.")
return c.FAILURE
def checkIn(self):
#===========================================================================
# Log card data to db
#===========================================================================
# Get and validate the visit value for this check-in
# Limited to 500 visits to prevent bad typos
"""while 1:
visitValue = self.tools.sanitizeInput(input("\nVisit Value (" + str(c.DEFAULT_VISITS) + "): "))
# Validate visit input
if visitValue == "":
visitValue = str(c.DEFAULT_VISITS)
break
elif (visitValue.isdigit() and int(visitValue) <= 500) or visitValue == "back":
break
else:
print("Invalid input. Try again.")"""
while 1:
CUID = self.tools.getCardSwipe()
# If the user requested to exit the loop, break
if CUID == c.BACK:
break
elif CUID == c.ERROR_READING_CARD:
print("Error reading card. Swipe card again.")
continue
# Sanitize CUID
CUID = self.tools.sanitizeInput(CUID)
# CUID will be empty if it failed sanitization. Skip checkIn if that is the case
if CUID == "":
continue
# Do the checkIn
checkInResult = self.db.checkIn(CUID)
if checkInResult["checkInStatus"] == c.SQL_ERROR:
self.showDatabaseError(checkInResult["sqlError"])
elif checkInResult["checkInStatus"] == c.BAD_CHECKIN_TIME:
print("Error: You may only check-in once per hour.")
elif checkInResult["checkInStatus"] == c.FUTURE_CHECKIN_TIME:
print(
"Error: Previous check-in time was in the future. Check your local system time."
)
elif checkInResult["checkInStatus"] == c.CUID_NOT_IN_DB:
# Ask if user wants to add the card
addCard = input(
"Error: Card not found in database. Add it now? (Y,n) ")
if addCard == "n":
continue
# Get the userID for the new card
firstName = self.tools.sanitizeInput(input("First Name: "))
lastName = self.tools.sanitizeInput(input("Last Name: "))
email = self.tools.sanitizeInput(input("Clemson Username: "******"addCardStatus"] == c.SUCCESS:
self.showCheckinConfirmation(email)
elif addCardResult["addCardStatus"] == c.SQL_ERROR:
self.showDatabaseError(addCardResult["sqlError"])
elif checkInResult["checkInStatus"] == c.SUCCESS:
self.showCheckinConfirmation(checkInResult["userID"])
else:
print("Unknown error checking in.")
def showVisits(self):
#===========================================================================
# Poll db for visit data based on userID
#===========================================================================
userID = self.tools.sanitizeInput(input("\nUser ID (blank for all): "))
showVisitsResult = self.db.showVisits(userID)
if showVisitsResult["showVisitsStatus"] == c.SQL_ERROR:
self.showDatabaseError(showVisitsResult["sqlError"])
elif showVisitsResult["showVisitsStatus"] == c.NO_RESULTS:
print("\nThere were no results to that query.")
elif showVisitsResult["showVisitsStatus"] == c.SUCCESS:
# If showing all users, display a pretty table
if userID == "":
print(
"\n+--------------------+\n| User ID | Visits |\n+--------------------+"
)
for i in range(len(showVisitsResult["visitsTuple"])):
print("|%10s | %6s |" %
(showVisitsResult["visitsTuple"][i][0],
showVisitsResult["visitsTuple"][i][1]))
print("+--------------------+")
# Show a single user's visits
else:
print("\n%s has %s visits." %
(userID, str(showVisitsResult["visitsTuple"][0][0])))
def getDbInfo(self):
#===========================================================================
# Request dbInfo from user - suggest default info from constants
#===========================================================================
self.dbName = input("Database name: (" + c.DEFAULT_DATABASE + ") ")
if self.dbName == "":
self.dbName = c.DEFAULT_DATABASE
self.dbHost = input("Database host: (" + c.DEFAULT_HOST + ") ")
if self.dbHost == "":
self.dbHost = c.DEFAULT_HOST
self.dbUsersTable = input("Database User table: (" + c.TABLE_USERS +
") ")
if self.dbUsersTable == "":
self.dbUsersTable = c.TABLE_USERS
self.dbVisitsTable = input("Database Visits table: (" +
c.TABLE_VISITS + ") ")
if self.dbVisitsTable == "":
self.dbVisitsTable = c.TABLE_VISITS
self.dbUser = input("Database Username: (" + c.DEFAULT_USER + ") ")
if self.dbUser == "":
self.dbUser = c.DEFAULT_USER
while 1:
self.dbPass = getpass.getpass("Database Password: "******"":
print("Database password cannot be blank.")
else:
break
def showCheckinConfirmation(self, userID):
#===========================================================================
# show confirmation of successful check in
#===========================================================================
print("\n%s is checked in" % (userID))
def showDatabaseError(self, error):
#===========================================================================
# Warn of database error - and provide information
#===========================================================================
print("\nWARNING! Database error:\n%s" % error.pgerror)
def invalidInput(self):
#===========================================================================
# Complain about invalid input
#===========================================================================
print("Invalid option. Try again.")
class TextUI:
def __init__(self):
self.db = None
self.tools = Utils()
def start(self):
#===========================================================================
# Main function - start connection to db then open main menu
#===========================================================================
try:
while 1:
# Get DB info
self.getDbInfo()
# Create the DB object
self.db = DB(self.dbHost, self.dbName, self.dbUsersTable, self.dbVisitsTable, self.dbUser, self.dbPass)
# Connect to the database
connectStatus = self.connectToDatabase()
# If we failed to connect to the database offer to re-enter db info
if connectStatus != c.SUCCESS:
reenter = input("Failed to connect to database. Re-enter database info? (Y,n) ")
if reenter.lower() == "n":
print("Bye.")
sys.exit(0)
else:
break
# Start the main menu loop
self.displayMenu()
except KeyboardInterrupt:
pass
finally:
print("Cleaning up and exiting...")
if self.db is not None:
self.db.close()
def displayMenu(self):
#===========================================================================
# Display main cli and take appropriate actions
#===========================================================================
#print("\nType \"back\" at any time to go up a menu level.")
while 1:
# Display main menu
print("\n\t1.) Check-in\n\t2.) Show Visits\n\t3.) Exit")
try:
option = input("\n>> ")
if option == "1":
self.checkIn()
elif option == "2":
self.showVisits()
elif option == "3":
sys.exit(0)
#elif option == "back" or option == "exit":
# exit = input("Exit? (y,N) ")
# if exit.lower() == "y":
# sys.exit(0)
else:
self.invalidInput()
except ValueError:
self.invalidInput()
def connectToDatabase(self):
#===========================================================================
# Open connection to database and return status
#===========================================================================
# Use stdout.write to prevent newline
sys.stdout.write("Connecting to database...")
# Connect to the DB!
status = self.db.connect()
if status == c.SUCCESS:
print("done.")
return status
elif status == c.BAD_PASSWD:
print("\nError connecting to database: Bad username or password.")
return status
else:
print("\nUnknown Error connecting to database.")
return c.FAILURE
def checkIn(self):
#===========================================================================
# Log card data to db
#===========================================================================
# Get and validate the visit value for this check-in
# Limited to 500 visits to prevent bad typos
"""while 1:
visitValue = self.tools.sanitizeInput(input("\nVisit Value (" + str(c.DEFAULT_VISITS) + "): "))
# Validate visit input
if visitValue == "":
visitValue = str(c.DEFAULT_VISITS)
break
elif (visitValue.isdigit() and int(visitValue) <= 500) or visitValue == "back":
break
else:
print("Invalid input. Try again.")"""
while 1:
CUID = self.tools.getCardSwipe()
# If the user requested to exit the loop, break
if CUID == c.BACK:
break
elif CUID == c.ERROR_READING_CARD:
print("Error reading card. Swipe card again.")
continue
# Sanitize CUID
CUID = self.tools.sanitizeInput(CUID)
# CUID will be empty if it failed sanitization. Skip checkIn if that is the case
if CUID == "":
continue
# Do the checkIn
checkInResult = self.db.checkIn(CUID)
if checkInResult["checkInStatus"] == c.SQL_ERROR:
self.showDatabaseError(checkInResult["sqlError"])
elif checkInResult["checkInStatus"] == c.BAD_CHECKIN_TIME:
print("Error: You may only check-in once per hour.")
elif checkInResult["checkInStatus"] == c.FUTURE_CHECKIN_TIME:
print("Error: Previous check-in time was in the future. Check your local system time.")
elif checkInResult["checkInStatus"] == c.CUID_NOT_IN_DB:
# Ask if user wants to add the card
addCard = input("Error: Card not found in database. Add it now? (Y,n) ")
if addCard == "n":
continue
# Get the userID for the new card
firstName = self.tools.sanitizeInput(input("First Name: "))
lastName = self.tools.sanitizeInput(input("Last Name: "))
email = self.tools.sanitizeInput(input("Clemson Username: "******"addCardStatus"] == c.SUCCESS:
self.showCheckinConfirmation(email)
elif addCardResult["addCardStatus"] == c.SQL_ERROR:
self.showDatabaseError(addCardResult["sqlError"])
elif checkInResult["checkInStatus"] == c.SUCCESS:
self.showCheckinConfirmation(checkInResult["userID"])
else:
print("Unknown error checking in.")
def showVisits(self):
#===========================================================================
# Poll db for visit data based on userID
#===========================================================================
userID = self.tools.sanitizeInput(input("\nUser ID (blank for all): "))
showVisitsResult = self.db.showVisits(userID)
if showVisitsResult["showVisitsStatus"] == c.SQL_ERROR:
self.showDatabaseError(showVisitsResult["sqlError"])
elif showVisitsResult["showVisitsStatus"] == c.NO_RESULTS:
print("\nThere were no results to that query.")
elif showVisitsResult["showVisitsStatus"] == c.SUCCESS:
# If showing all users, display a pretty table
if userID == "":
print("\n+--------------------+\n| User ID | Visits |\n+--------------------+")
for i in range(len(showVisitsResult["visitsTuple"])):
print("|%10s | %6s |" % (showVisitsResult["visitsTuple"][i][0], showVisitsResult["visitsTuple"][i][1]))
print("+--------------------+")
# Show a single user's visits
else:
print("\n%s has %s visits." % (userID, str(showVisitsResult["visitsTuple"][0][0])))
def getDbInfo(self):
#===========================================================================
# Request dbInfo from user - suggest default info from constants
#===========================================================================
self.dbName = input("Database name: (" + c.DEFAULT_DATABASE + ") ")
if self.dbName == "":
self.dbName = c.DEFAULT_DATABASE
self.dbHost = input("Database host: (" + c.DEFAULT_HOST + ") ")
if self.dbHost == "":
self.dbHost = c.DEFAULT_HOST
self.dbUsersTable = input("Database User table: (" + c.TABLE_USERS + ") ")
if self.dbUsersTable == "":
self.dbUsersTable = c.TABLE_USERS
self.dbVisitsTable = input("Database Visits table: (" + c.TABLE_VISITS + ") ")
if self.dbVisitsTable == "":
self.dbVisitsTable = c.TABLE_VISITS
self.dbUser = input("Database Username: (" + c.DEFAULT_USER + ") ")
if self.dbUser == "":
self.dbUser = c.DEFAULT_USER
while 1:
self.dbPass = getpass.getpass("Database Password: "******"":
print("Database password cannot be blank.")
else:
break
def showCheckinConfirmation(self, userID):
#===========================================================================
# show confirmation of successful check in
#===========================================================================
print("\n%s is checked in" % (userID))
def showDatabaseError(self, error):
#===========================================================================
# Warn of database error - and provide information
#===========================================================================
print("\nWARNING! Database error:\n%s" % error.pgerror)
def invalidInput(self):
#===========================================================================
# Complain about invalid input
#===========================================================================
print("Invalid option. Try again.")
