diff --git a/ddbank.py b/ddbank.py index bf66ffa..28b54ed 100755 --- a/ddbank.py +++ b/ddbank.py @@ -31,12 +31,23 @@ def checkBank(a): dirOutput = os.listdir() matchs = [match for match in dirOutput if 'db.sqlite' in match] for i in matchs: + print(i) checkBank.bnktst = 'fail' if a == i.split(".")[0]: checkBank.bnktst = 'pass' if checkBank.bnktst != "pass": return_37(a) +def userBalance(a): + connDb = sqlite3.connect(a) + cur = connDb.cursor() + balUser = input('User? ').lower() + userCheck(balUser,a) + cur.execute("SELECT id FROM account where username=?", (balUser, )) + userId = cur.fetchone()[0] + bal = cur.execute("SELECT balance from balance1 where id=?", (userId, )).fetchall() + print(bal) + def transfer(a): clear() print() @@ -117,7 +128,7 @@ def manBankin(a): elif userIn == "3": transfer(a) elif userIn == "4": - print("User balance") + userBalance(a) elif userIn == "9": manageBankmenu(a) elif userIn == "8": @@ -218,10 +229,28 @@ def selectBank(): manageBankmenu(manBank) def addBank(): - clear() print('Add a bank: ') print() bankName = input("What would you like to call this bank? ").lower() + if not bankName: + print() + print('Not a valid name. ') + mainFunc() + elif bankName.isspace() == True: + print() + mainFunc() + dirOutput = os.listdir() + matchs = [match for match in dirOutput if 'db.sqlite' in match] + for i in matchs: + checkBank.nametst = 'pass' + if bankName == i.split(".")[0]: + addBank.nametst = 'fail' + if addBank.nametst != "pass": + print() + print('~Name already exists! ') + print() + mainFunc() + bankName = bankName+".db.sqlite" connDb = sqlite3.connect(bankName) connDb.execute('''CREATE TABLE account @@ -242,7 +271,7 @@ def addBank(): AMOUNT INTEGER NOT NULL, TIMESTAMP DATETIME DEFAULT CURRENT_TIMESTAMP);''') connDb.close() - clear() +# clear() print() print(f'*** Added bank {bankName}') print() @@ -284,7 +313,7 @@ def mainFunc(): 2. Add new bank. 3. List banks. 4. Delete a bank. - Q to quit. + 9 to quit. """) mainOption = int(input("Choose an option: ")) if mainOption == 1: @@ -295,7 +324,7 @@ def mainFunc(): listBank() elif mainOption == 4: delBank() - elif mainOption == 'Q': + elif mainOption == 9: exit() else: