#!/usr/bin/env python3
import sqlite3
import os

def selectBank():
    print('Make things to select the bank...')

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()

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':
        print('Place function here.')
    elif mainOption == '2':
        addBank()
    elif mainOption == '3':
        print('Place function here.')
    elif mainOption == '4':
        print('Place function here.')
    elif mainOption == 'Q' or mainOption == 'q':
        break
    else:
        print()
        print('*** Option not understood. Please try again.')