Add user created

master
Matt Theissen 3 years ago
parent 10de6e04ab
commit 6730f60f34

@ -2,18 +2,28 @@
import sqlite3 import sqlite3
import os import os
def addUser(a):
connDb = sqlite3.connect(a)
cur = connDb.cursor()
userIn = input('What user would you like to add? ')
cur.execute("INSERT INTO account (username) VALUES(?)", (userIn,))
cur.execute("INSERT INTO balance (balance) VALUES(0.00)")
cur.execute("INSERT INTO banktrans (username) VALUES(?)", (userIn,))
connDb.commit()
connDb.close()
def manageBankmenu(a): def manageBankmenu(a):
print(""" while True:
print("""
1. Add a user. 1. Add a user.
2. Delete a user. 2. Delete a user.
3. List users. 3. List users.
4. Banking Menu. 4. Banking Menu.
Q to quit. Q to quit.
""") """)
while True: menuOption = input("Bank: Choose from the menu: ")
menuOption = input("Choose from the menu: ")
if menuOption == '1': if menuOption == '1':
print('Add a user') addUser(a)
elif menuOption == '2': elif menuOption == '2':
print('delete a user') print('delete a user')
elif menuOption == '3': elif menuOption == '3':
@ -50,16 +60,16 @@ def addBank():
bankName = input("What would you like to call this bank? ") bankName = input("What would you like to call this bank? ")
bankName = bankName+".db.sqlite" bankName = bankName+".db.sqlite"
connDb = sqlite3.connect(bankName) connDb = sqlite3.connect(bankName)
connDb.execute('''CREATE TABLE ACCOUNT connDb.execute('''CREATE TABLE account
(ID INT PRIMARY KEY NOT NULL UNIQUE, (ID INTEGER PRIMARY KEY AUTOINCREMENT,
USERNAME TEXT NOT NULL, USERNAME TEXT NOT NULL,
TIMESTAMP DATETIME DEFAULT CURRENT_TIMESTAMP);''') TIMESTAMP DATETIME DEFAULT CURRENT_TIMESTAMP);''')
connDb.execute('''CREATE TABLE BALANCE connDb.execute('''CREATE TABLE balance
(ID INT PRIMARY KEY NOT NULL UNIQUE, (ID INTEGER PRIMARY KEY AUTOINCREMENT,
BALANCE INT NOT NULL, BALANCE REAL NOT NULL,
TIMESTAMP DATETIME DEFAULT CURRENT_TIMESTAMP);''') TIMESTAMP DATETIME DEFAULT CURRENT_TIMESTAMP);''')
connDb.execute('''CREATE TABLE BANKTRANS connDb.execute('''CREATE TABLE banktrans
(ID INT PRIMARY KEY NOT NULL UNIQUE, (ID INTEGER PRIMARY KEY AUTOINCREMENT,
USERNAME TEXT NOT NULL, USERNAME TEXT NOT NULL,
TIMESTAMP DATETIME DEFAULT CURRENT_TIMESTAMP);''') TIMESTAMP DATETIME DEFAULT CURRENT_TIMESTAMP);''')
connDb.close() connDb.close()