Continue working on CLI Implementation

cli/test_database_generator.py

@@ -1,21 +1,15 @@
 # Lucas Mathews - Fontys Student ID: 5023572
 # Banking System Test Database Generator
 
-# This program generates a test database for the banking system. The database contains 50 clients, each with 2 accounts. Each account has 40 transactions. 
-# The first client is an administrator. The password for the administrator account is "Happymeal1". The program uses the Faker library to generate fake 
-# data for the clients, accounts, and transactions. The random library is used to generate random data for the accounts and transactions. The program 
-# creates a new SQLite database called test_database.db and writes the test data to the database. The client ID of the  administrator account and the 
-# password for the administrator account.
+# This program generates dumy data for the banking system. 
+# This adds 50 clients, each with 2 accounts. Each account has 40 transactions. 
 
 from faker import Faker
-from connection import login, add_client, logout
+from connection import add_client, add_account, add_transaction
 import random
 import datetime
 import hashlib 
-import uuid
-
-ADMIN_EMAIL = "lmath56@hotmail.com" # Email address of the administrator account
-
+import time
 
 def generate_hash():  # Creates a hash for a password
     seed = str(random.random()).encode('utf-8')
@@ -29,71 +23,50 @@ def generate_test_database(client_id, password):
 
     all_account_ids = []  # List to store all account IDs
 
-    # Log in as the admin using provided username and password
-    client_hash = hashlib.sha512(password.encode()).hexdigest()
-    client = login(client_id, client_hash)
+    for i in range(50):  # Generate 50 clients
+        password = generate_hash()
+        name = fake.name()
+        birthdate = fake.date_of_birth(minimum_age=18, maximum_age=90)
+        address = fake.address()
+        phone_number = fake.phone_number()
+        email =  fake.email()
+        notes = fake.text(max_nb_chars=50)
 
-    if client is not None:  # Check if login was successful
-        print("Admin logged in successfully")
+        client_response = add_client(name, birthdate, address, phone_number, email, password, notes)
+        client_id = client_response['message']
+        for j in range(2):  # Each client has 2 accounts
+            balance = 1000  # Initialize balance to 1000
+            account_type = random.choice(['Spending', 'Savings'])
+            account_notes = fake.text(max_nb_chars=50)
+            
+            account_response = add_account(client_id=client_id, description=fake.text(max_nb_chars=15), account_type=account_type, notes=account_notes)
+            
+            response_dict = account_response
+            account_id = response_dict['message']
 
-        for i in range(50):  # Generate 50 clients
-            is_administrator = 1 if i == 0 else 0  # Set the first client as an administrator
-            # Set the password hash for the first account so that the password is "Happymeal1"
-            password = "Happymeal1" if i == 0 else generate_hash()
-            name = "ADMIN" if i == 0 else fake.name()
-            birthdate = "ADMIN" if i == 0 else fake.date_of_birth(minimum_age=18, maximum_age=90)
-            address = "ADMIN" if i == 0 else fake.address()
-            phone_number = "ADMIN" if i == 0 else fake.phone_number()
-            email = ADMIN_EMAIL if i == 0 else fake.email()
-            notes = fake.text(max_nb_chars=50)
 
-            # Add client using add_client function
-            response = add_client(name, birthdate, address, phone_number, email, password, notes)
-            print(response)  # Print the response message
+            for k in range(40):  # Each account has 40 transactions
+                if not all_account_ids:  # Skip creating a transaction if there are no accounts yet
+                    continue
 
-            for j in range(2):  # Each client has 2 accounts
-                balance = 1000  # Initialize balance to 1000
-                account_type = random.choice(['Spending', 'Savings'])
-                account_notes = fake.text(max_nb_chars=50)
+                transaction_type = random.choice(['Deposit', 'Withdrawal'])
+                amount = random.randint(1, 200)
 
-                # Add account using add_account function
-                account_response = add_account(
-                    client_id=client_id,
-                    description=fake.text(max_nb_chars=200),
-                    account_type=account_type,
-                    notes=account_notes
-                )
-                print(account_response[1])  # Print the response message
+                if transaction_type == 'Withdrawal' and balance - amount < 0:  # Skip withdrawal if it would make balance negative
+                    continue
 
-                for k in range(40):  # Each account has 40 transactions
-                    if not all_account_ids:  # Skip creating a transaction if there are no accounts yet
-                        continue
+                if transaction_type == 'Deposit':  # Update balance based on transaction type
+                    balance += amount
+                elif transaction_type == 'Withdrawal':
+                    balance -= amount
 
-                    transaction_type = random.choice(['Deposit', 'Withdrawal'])
-                    amount = random.randint(1, 200)
+                transaction_description = fake.text(max_nb_chars=50)
+                recipient_account_id = random.choice(all_account_ids)
+                
+                transaction_response = add_transaction(amount=amount, account_id=account_id, recipient_account_id=recipient_account_id, otp_code=123456, description=transaction_description)
+                print(f"Transaction response: {transaction_response}")
+                time.sleep(1)
 
-                    if transaction_type == 'Withdrawal' and balance - amount < 0:  # Skip withdrawal if it would make balance negative
-                        continue
+            all_account_ids.append(account_id)
 
-                    if transaction_type == 'Deposit':  # Update balance based on transaction type
-                        balance += amount
-                    elif transaction_type == 'Withdrawal':
-                        balance -= amount
-
-                    transaction_description = fake.text(max_nb_chars=50)
-                    recipient_account_id = random.choice(all_account_ids)
-
-                    # Add transaction using add_transaction function
-                    transaction_response = add_transaction(
-                        amount=amount,
-                        recipient_account_id=recipient_account_id,
-                        otp_code=123456,  # Replace with actual OTP verification code
-                        description=transaction_description
-                    )
-                    print(transaction_response[1])  # Print the response message
-
-                all_account_ids.append(account_id)
-        logout()  # Log out of the admin account
-        print(f"The client_id of the administrator account of this test database is: {all_account_ids[0]}. The password is: Happymeal1")
-    else:
-        print("Admin login failed.")
+    print("Test database generated successfully.")