Merge pull request #21 from SST-CTF/Stan

SQL completed

SQL Program Files/SQLite3.py

@@ -0,0 +1,96 @@
+import sqlite3
+import sys
+
+
+conn = sqlite3.connect('bank.db')
+c = conn.cursor()
+
+
+def checkBalance(cardNumber):
+ c.execute("SELECT balance FROM account WHERE Number = ?", (cardNumber,))
+ new_balance = c.fetchall()
+ new_balance1 = new_balance[0] #Turning array into tuple
+ new_balance2 = new_balance1[0] #turnig tuple into number
+ print('your balance is: ' + str(new_balance2))
+
+def deposit(cardNumber):
+ deposit_value = int(input("How much do you want to deposit? "))
+ c.execute("SELECT Balance FROM Account WHERE Number = ?", (cardNumber,))
+ new_balance = list(c.fetchall())
+ cardNumber = cardNumber [0:5]
+ new_balance = ''.join(str(e) for e in new_balance)
+ new_balance = new_balance[1:-2]
+ new_balance = int(new_balance[:])
+ new_balance += deposit_value
+ c.execute('UPDATE Account SET Balance = ? WHERE Number = ?', (new_balance, cardNumber))
+ conn.commit()
+ print("Your new balance is " + str(new_balance))
+
+def withdraw(cardNumber):
+ deposit_value = int(input("How much do you want to withdraw? "))
+ c.execute("SELECT Balance FROM Account WHERE Number = ?", (cardNumber,))
+ new_balance = list(c.fetchall())
+ cardNumber = cardNumber[0:5]
+ print(cardNumber)
+ new_balance = ''.join(str(e) for e in new_balance)
+ new_balance = new_balance[1:-2]
+ new_balance = int(new_balance[:])
+ if new_balance > deposit_value:
+ new_balance -= deposit_value
+ c.execute('UPDATE Account SET Balance = ? WHERE Number = ?', (new_balance, cardNumber))
+ conn.commit()
+ print("Your new balance is " + str(new_balance))
+ else:
+ print("Error: Not enough balance.")
+
+def checkCard(cardNumber):
+ print(cardNumber)
+ c.execute("SELECT Number, PIN FROM account WHERE Number = ?", (cardNumber,))
+ row = c.fetchall()
+ if row != []:
+ raw = row[0]
+ raw1 = int(raw[1])
+ k = 1
+ pin = (int(input("Please enter your four digit pin: ")))
+ while k < 5:
+ if pin == raw1:
+ print ('CORRECT!')
+ return 1
+ else:
+ pin = int(input("Incorrect, please try again: "))
+ k += 1
+ print("Access Denied!")
+ sys.exit()
+ else:
+ print("This card does not exist, please try again.")
+ return 0 # Possible add create new account in the future
+
+
+# Program Begins (main menu and GUI)
+print("**********************************************")
+print("*********** Welcome to the SST ATM ***********")
+print("**********************************************")
+cardExists = 0
+while cardExists == 0:
+ cardNumber = input("Please enter your credit card number or '1' to exit: ")
+ if int(cardNumber) == 1:
+ sys.exit()
+ else:
+ cardExists = checkCard(cardNumber)
+
+# Selection is made here, each selection leads to a function above
+selection = 0 # Define initially to run through loop
+while int(selection) < 4:
+ selection = input(
+ "What would you like to do today?:\n 1. Check balance\n 2. Deposit Money\n 3. Withdraw Money\n 4. Exit\n\n")
+ if int(selection) == 1: # Check Balance
+ checkBalance(cardNumber)
+ elif int(selection) == 2: # Deposit
+ deposit(cardNumber)
+ elif int(selection) == 3: # Withdraw
+ withdraw(cardNumber)
+ elif int(selection) == 4: # Exit program
+ print("Thank you, and have a nice day!")
+ # elif selection is 5: # Change
+ #change_account(cardNumber)
+sys.exit()