103 lines
4.0 KiB
Python
103 lines
4.0 KiB
Python
# Lucas Mathews - Fontys Student ID: 5023572
|
|
# Banking System CLI Utility
|
|
|
|
import json
|
|
import argparse
|
|
import sys
|
|
from config import CONFIG
|
|
from getpass import getpass
|
|
from connection import login, logout, get_client
|
|
from test_database_generator import generate_test_database
|
|
|
|
|
|
def show_menu():
|
|
print("\nAvailable options:")
|
|
print("1. Logout")
|
|
print("2. New user")
|
|
print("3. Add test users")
|
|
print("4. Exit")
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description='Banking System CLI Utility')
|
|
parser.add_argument('-u', '--username', type=str, default=CONFIG["client"]["default_id"], help='Username for login')
|
|
parser.add_argument('-p', '--password', type=str, default=CONFIG["client"]["default_password"], help='Password for login')
|
|
|
|
subparsers = parser.add_subparsers(dest='command')
|
|
|
|
login_parser = subparsers.add_parser('login', help='Login to the system')
|
|
logout_parser = subparsers.add_parser('logout', help='Logout from the system')
|
|
|
|
args = parser.parse_args()
|
|
|
|
if not args.command:
|
|
while True:
|
|
if not args.username:
|
|
args.username = input("Enter username: ")
|
|
if not args.password:
|
|
args.password = getpass("Enter password: ")
|
|
|
|
response = login(args.username, args.password)
|
|
print(response)
|
|
if response['success']:
|
|
print(f"Login successful: {response['message']}")
|
|
try:
|
|
with open('session_data.json', 'r') as f:
|
|
session_data = json.load(f)
|
|
client_id = session_data['client_id']
|
|
client_info = get_client(client_id)
|
|
if client_info.get('success'):
|
|
client = client_info['data']
|
|
name = client['name']
|
|
print(f"Welcome, {name}!")
|
|
except Exception as e:
|
|
print(f"Error loading client data: {str(e)}")
|
|
|
|
while True:
|
|
show_menu()
|
|
option = input("Choose an option: ")
|
|
if option == "1":
|
|
response = logout()
|
|
json_response = response.json()
|
|
if json_response['success']:
|
|
print(f"Logout successful: {json_response['message']}")
|
|
else:
|
|
print(f"Logout failed: {json_response['message']}")
|
|
args.username = None
|
|
args.password = None
|
|
elif option == "2":
|
|
print("New user option selected.")
|
|
# Implement new user functionality here
|
|
elif option == "3":
|
|
print("Add test users option selected.")
|
|
generate_test_database(args.username, args.password)
|
|
|
|
elif option == "4":
|
|
print("Exiting...")
|
|
break
|
|
else:
|
|
print("Invalid option. Please try again.")
|
|
break
|
|
else:
|
|
print(f"Login failed: {response['message']}. Please try again.")
|
|
args.username = None
|
|
args.password = None
|
|
elif args.command == 'login':
|
|
if not args.username or not args.password:
|
|
print("Username and password are required for login.")
|
|
sys.exit(1)
|
|
response = login(args.username, args.password)
|
|
if response['success']:
|
|
print(f"Login successful: {response['message']}")
|
|
else:
|
|
print(f"Login failed: {response['message']}. Please try again.")
|
|
elif args.command == 'logout':
|
|
response = logout()
|
|
if response['success']:
|
|
print(f"Logout successful: {response['message']}")
|
|
else:
|
|
print(f"Logout failed: {response['message']}")
|
|
else:
|
|
print("Invalid command. Use 'login' or 'logout'.")
|
|
|
|
if __name__ == "__main__":
|
|
main() |