From 6730f60f34435ddf119af49d1ad0f4a66773647b Mon Sep 17 00:00:00 2001
From: Matt Theissen <mtheissen@Matts-MacBook-Pro.local>
Date: Sat, 23 Apr 2022 01:28:04 -0500
Subject: [PATCH] Add user created

---
 ddbank.py | 32 +++++++++++++++++++++-----------
 1 file changed, 21 insertions(+), 11 deletions(-)

diff --git a/ddbank.py b/ddbank.py
index e147db8..f32be57 100755
--- a/ddbank.py
+++ b/ddbank.py
@@ -2,18 +2,28 @@
 import sqlite3
 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):
-    print("""
+    while True:
+        print("""
         1. Add a user.
         2. Delete a user.
         3. List users.
         4. Banking Menu.
         Q to quit.
         """)
-    while True:
-        menuOption = input("Choose from the menu: ")
+        menuOption = input("Bank: Choose from the menu: ")
         if menuOption == '1':
-            print('Add a user')
+            addUser(a)
         elif menuOption == '2':
             print('delete a user')
         elif menuOption == '3':
@@ -50,16 +60,16 @@ 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,
+    connDb.execute('''CREATE TABLE account
+         (ID INTEGER PRIMARY KEY AUTOINCREMENT,
          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,
+    connDb.execute('''CREATE TABLE balance
+         (ID INTEGER PRIMARY KEY AUTOINCREMENT,
+         BALANCE           REAL    NOT NULL,
          TIMESTAMP  DATETIME DEFAULT CURRENT_TIMESTAMP);''')
-    connDb.execute('''CREATE TABLE BANKTRANS
-         (ID INT PRIMARY KEY     NOT NULL   UNIQUE,
+    connDb.execute('''CREATE TABLE banktrans
+         (ID INTEGER PRIMARY KEY AUTOINCREMENT,
          USERNAME           TEXT    NOT NULL,
          TIMESTAMP  DATETIME DEFAULT CURRENT_TIMESTAMP);''')
     connDb.close()