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

Code cleanup

Browse Source
This commit is contained in:
Lucas Mathews
2024-05-25 11:34:04 +02:00
parent 8ea2f1c173
commit 93c3c6f59b
1 changed files with 4 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
database.py
View File

@@ -6,28 +6,21 @@ import os.path
from sqlalchemy import create_engine from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker from sqlalchemy.orm import sessionmaker
# Import Config from config import CONFIG # Import Config
from config import CONFIG
# Check if the database exists if os.path.exists(CONFIG["database"]["name"]): # Check if the database exists
if os.path.exists(CONFIG["database"]["name"]):
print(f"Database {CONFIG["database"]["name"]} already exists.") print(f"Database {CONFIG["database"]["name"]} already exists.")
else: else:
print(f"Database {CONFIG["database"]["name"]} does not exist. Creating it now.") print(f"Database {CONFIG["database"]["name"]} does not exist. Creating it now.")
# Sets the database file to be used from the configuration file db_url : str = "sqlite:///" + CONFIG["database"]["name"] # Sets the database file to be used from the configuration file
db_url : str = "sqlite:///" + CONFIG["database"]["name"]
print(f"Database file set to: {db_url}") print(f"Database file set to: {db_url}")
# Creates the database engine (does not create the database file if it already exists) engine = create_engine(db_url, echo=True) # Creates the database engine (does not create the database file if it already exists)
engine = create_engine(db_url, echo=True)
# Import base class
from class_base import Base # Imports the base class required by SQLAlchemy from class_base import Base # Imports the base class required by SQLAlchemy
# Create the tables in the database
Base.metadata.create_all(bind=engine) # Creates the tables in the database from the classes Base.metadata.create_all(bind=engine) # Creates the tables in the database from the classes
# Creates a session to interact with the database
Session = sessionmaker(bind=engine) # Creates a session to interact with the database Session = sessionmaker(bind=engine) # Creates a session to interact with the database
session = Session() # Creates a session object session = Session() # Creates a session object