lucas/s2p_banking_system
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Updated and tested database_generator.py

Browse Source
This commit is contained in:
Lucas Mathews
2024-05-23 16:03:43 +02:00
parent 41db324c7f
commit 8dfee22dfc
3 changed files with 65 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
bank.ini
View File

@@ -1,5 +1,5 @@
[database]
name=bank.db
name=test_database.db
[api_file]
name=api.yml

BIN
test_database.db
View File

Binary file not shown.

99
test_database_generator.py
View File

@@ -1,8 +1,17 @@
# 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.
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from class_base import Base
import hashlib # hashlib for password hashing
import uuid # uuid for unique identifiers
import hashlib
import uuid
from class_account import Account
from class_client import Client
from class_transaction import Transaction
@@ -16,30 +25,24 @@ def generate_hash(): # Creates a hash for a password
def generate_uuid(): # Generates a unique identifier for transactions
return str(uuid.uuid4())
def generate_uuid_short(): # Generates a short uuid
def generate_uuid_short(): # Generates a short uuid for accounts and clients
return str(uuid.uuid4())[:8]
# Create a new engine for the test database
engine = create_engine('sqlite:///test_database.db')
engine = create_engine('sqlite:///test_database.db')# Create a new engine for the test database
Base.metadata.create_all(engine) # Create all tables in the test database
Session = sessionmaker(bind=engine) # Create a new sessionmaker bound to the test engine
session = Session() # Create a new session
# Create all tables in the test database
Base.metadata.create_all(engine)
fake = Faker() # Create a Faker instance
# Create a new sessionmaker bound to the test engine
Session = sessionmaker(bind=engine)
all_account_ids = [] # List to store all account IDs
# Create a new session
session = Session()
# Create a Faker instance
fake = Faker()
# List to store all account IDs
all_account_ids = []
# Generate 50 clients
for i in range(50):
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_hash = "7835062ec36ed529fe22cc63baf3ec18d347dacb21c9801da8ba0848cc18efdf1e51717dd5b1240f7556aca3947aa0722452858be6002c1d46b1f1c311b0e9d8" if i == 0 else generate_hash()
client_id = generate_uuid_short()
all_account_ids.append(client_id) # Add the client ID to the list of account IDs
client = Client(
client_id=client_id,
name=fake.name(),
@@ -48,44 +51,58 @@ for i in range(50):
address=fake.address(),
phone_number=fake.phone_number(),
email=fake.email(),
administrator=random.choice([0, 1]),
hash=generate_hash(), # Replace with appropriate value
administrator=is_administrator,
hash=password_hash,
notes=fake.text(max_nb_chars=50), # Generate fake notes
enabled=1,
accounts=[]) # Empty list for accounts, you can add accounts later)
session.add(client)
# Each client has 2 accounts
for j in range(2):
for j in range(2):# Each client has 2 accounts
account_id = generate_uuid_short()
balance = 1000 # Initialize balance to 1000
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
transaction_type = random.choice(['Deposit', 'Withdrawal'])
amount = random.randint(1, 200)
if transaction_type == 'Withdrawal' and balance - amount < 0: # Skip withdrawal if it would make balance negative
continue
if transaction_type == 'Deposit': # Update balance based on transaction type
balance += amount
elif transaction_type == 'Withdrawal':
balance -= amount
transaction = Transaction(
transaction_id=generate_uuid(),
account_id=account_id,
recipient_account_id=random.choice(all_account_ids),
transaction_type=transaction_type,
amount=amount,
timestamp=fake.date_this_year(),
description=fake.text(max_nb_chars=50)
)
session.add(transaction)
account = Account(
account_id=account_id,
client_id=client_id,
description=fake.text(max_nb_chars=200),
open_timestamp=fake.date_this_century(),
account_type=random.choice(['Spending', 'Savings']),
balance=random.randint(100, 10000),
balance=balance, # Set balance to calculated balance
enabled=1,
notes=fake.text(max_nb_chars=50),
transactions=[])
session.add(account)
all_account_ids.append(account_id)
# Each account has 40 transactions
for k in range(40):
transaction = Transaction(
transaction_id=generate_uuid(), # Call the function here
account_id=account_id,
recipient_account_id=random.choice(all_account_ids),
transaction_type=random.choice(['Deposit', 'Withdrawal']),
amount=random.randint(1, 20),
timestamp=fake.date_this_year(),
description=fake.text(max_nb_chars=50)
)
session.add(transaction)
session.commit() # Commit the session to write the test data to the database
# Commit the session to write the test data to the database
session.commit()
print(f"The client_id of the administrator account of this test database is: {all_account_ids[0]}. The password is: Happymeal1")
# Close the session
session.close()
session.close() # Close the session