You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
62 lines
1.7 KiB
62 lines
1.7 KiB
#!/usr/bin/env python3
|
|
import sqlite3
|
|
import os
|
|
def selectBank():
|
|
listBank()
|
|
|
|
def addBank():
|
|
bankName = input("What would you like to call this bank? ")
|
|
bankName = bankName+".db.sqlite"
|
|
connDb = sqlite3.connect(bankName)
|
|
connDb.execute('''CREATE TABLE ACCOUNT
|
|
(ID INT PRIMARY KEY NOT NULL UNIQUE,
|
|
USERNAME TEXT NOT NULL,
|
|
TIMESTAMP DATETIME DEFAULT CURRENT_TIMESTAMP);''')
|
|
connDb.execute('''CREATE TABLE BALANCE
|
|
(ID INT PRIMARY KEY NOT NULL UNIQUE,
|
|
BALANCE INT NOT NULL,
|
|
TIMESTAMP DATETIME DEFAULT CURRENT_TIMESTAMP);''')
|
|
connDb.execute('''CREATE TABLE BANKTRANS
|
|
(ID INT PRIMARY KEY NOT NULL UNIQUE,
|
|
USERNAME TEXT NOT NULL,
|
|
TIMESTAMP DATETIME DEFAULT CURRENT_TIMESTAMP);''')
|
|
connDb.close()
|
|
os.system('clear')
|
|
print()
|
|
print(f'*** Added bank {bankName}')
|
|
print()
|
|
|
|
def listBank():
|
|
dirOutput = os.listdir()
|
|
matchs = [match for match in dirOutput if 'db.sqlite' in match]
|
|
print()
|
|
print('Banks found: ')
|
|
for i in matchs:
|
|
print(i.split(".")[0])
|
|
|
|
while True:
|
|
print("""
|
|
1. Select a Bank.
|
|
2. Add new bank.
|
|
3. List banks.
|
|
4. Delete a bank.
|
|
Q to quit.
|
|
""")
|
|
mainOption = input("Choose an option: ")
|
|
if mainOption == '1':
|
|
selectBank()
|
|
elif mainOption == '2':
|
|
addBank()
|
|
elif mainOption == '3':
|
|
listBank()
|
|
elif mainOption == '4':
|
|
print('Place function here.')
|
|
elif mainOption == 'Q' or mainOption == 'q':
|
|
break
|
|
else:
|
|
print()
|
|
print('*** Option not understood. Please try again.')
|
|
|
|
|
|
|