def initializeAndFillDatabase():
"""
Main where we implement most methods above (create connection, create table, insert data, close connection.)
"""
# Database already exists; do nothing
if len(glob.glob(PATH_TO_DATABASE)) == 1:
return False
database = app.classes.database_container.DatabaseContainer.get_instance()
app.classes.database_container.DatabaseContainer.commit_lock = True
print("- Filling database -")
# initialized variable with query that creates book table with columns/attributes
table_creation_dict = {
"book_table":
"""CREATE TABLE IF NOT EXISTS book (
id INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,
author TEXT NOT NULL,
title TEXT NOT NULL,
format TEXT NOT NULL,
pages INTEGER NOT NULL,
publisher TEXT NOT NULL,
year_of_publication INTEGER NOT NULL,
language TEXT NOT NULL,
isbn_10 TEXT NOT NULL,
isbn_13 TEXT NOT NULL,
total_quantity INTEGER NOT NULL,
quantity_available INTEGER NOT NULL
);""",
# initialized variable with query that creates magazine table with columns/attributes
"magazine_table":
"""CREATE TABLE IF NOT EXISTS magazine (
id INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,
title TEXT NOT NULL,
publisher TEXT NOT NULL,
year_of_publication INTEGER NOT NULL,
language TEXT NOT NULL,
isbn_10 TEXT NOT NULL,
isbn_13 TEXT NOT NULL,
total_quantity INTEGER NOT NULL,
quantity_available INTEGER NOT NULL
);""",
# initialized variable with query that creates movie table with columns/attributes
"movie_table":
"""CREATE TABLE IF NOT EXISTS movie (
id INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,
title TEXT NOT NULL,
director TEXT NOT NULL,
producers TEXT NOT NULL,
actors TEXT NOT NULL,
language TEXT NOT NULL,
subtitles TEXT NOT NULL,
dubbed TEXT NOT NULL,
release_date INTEGER NOT NULL,
run_time INTEGER NOT NULL,
total_quantity INTEGER NOT NULL,
quantity_available INTEGER NOT NULL
);""",
# initialized variable with query that creates album table with columns/attributes
"album_table":
"""CREATE TABLE IF NOT EXISTS album (
id INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,
type TEXT NOT NULL,
title TEXT NOT NULL,
artist TEXT NOT NULL,
label TEXT NOT NULL,
release_date INTEGER NOT NULL,
asin TEXT NOT NULL,
total_quantity INTEGER NOT NULL,
quantity_available INTEGER NOT NULL
);""",
# initialized variable with query that creates client table with columns/attributes
"user_table":
"""CREATE TABLE IF NOT EXISTS client (
id INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,
firstName TEXT NOT NULL,
lastName TEXT NOT NULL,
physicalAddress TEXT NOT NULL,
email TEXT NOT NULL,
phoneNumber TEXT NOT NULL,
username TEXT NOT NULL,
password TEXT NOT NULL,
isAdmin INTEGER NOT NULL,
isLogged INTEGER NOT NULL,
lastLogged INTEGER NOT NULL
);""",
# initialized variable with query that creates book_copy table with columns/attributes
#FOREIGN KEY(book_id) REFERENCES book(id),
"book_copy_table":
"""CREATE TABLE IF NOT EXISTS book_copy (
id INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,
book_id INTEGER NOT NULL,
isLoaned INTEGER NOT NULL,
FOREIGN KEY(book_id) REFERENCES book(id)
);""",
# initialized variable with query that creates magazine_copy table with columns/attributes
"magazine_copy_table":
"""CREATE TABLE IF NOT EXISTS magazine_copy (
id INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,
magazine_id INTEGER NOT NULL,
FOREIGN KEY(magazine_id) REFERENCES magazine(id)
);""",
# initialized variable with query that creates movie_copy table with columns/attributes
"movie_copy_table":
"""CREATE TABLE IF NOT EXISTS movie_copy (
id INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,
movie_id INTEGER NOT NULL,
isLoaned INTEGER NOT NULL,
FOREIGN KEY(movie_id) REFERENCES movie(id)
);""",
# initialized variable with query that creates album_copy table with columns/attributes
"album_copy_table":
"""CREATE TABLE IF NOT EXISTS album_copy (
id INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,
album_id INTEGER NOT NULL,
isLoaned INTEGER NOT NULL,
FOREIGN KEY(album_id) REFERENCES album(id)
);""",
"loan_table":
"""CREATE TABLE IF NOT EXISTS loan (
id INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,
user_id INTEGER NOT NULL,
record_id INTEGER NOT NULL,
table_name TEXT NOT NULL,
loan_time INTEGER NOT NULL,
due_time INTEGER NOT NULL,
return_time INTEGER NOT NULL,
is_returned INTEGER NOT NULL,
FOREIGN KEY(user_id) REFERENCES client(id)
);"""
}
# Create all tables
for table_name, table_sql in table_creation_dict.items():
database.execute_query(table_sql)
app.classes.database_container.DatabaseContainer.commit_lock = False
database.commit_db()
COPIES = 3
NUM_BOOKS = 50
MAX_BOOK_PAGES = 1500
NUM_MAGAZINES = 50
NUM_MOVIES = 50
NUM_ALBUMS = 50
NUM_USERS = 50
book_types = [
'Paperback', 'Hardcover', 'Graphic', 'Coffee Table Book', 'Textbook'
]
languages = [
'English', 'French', 'Italian', 'Spanish', 'Greek', 'Russian', 'German'
]
album_types = ["Vinyl", "CD", "Cassette"]
MAX_QUANTITY = 10
MAX_TOTAL = 4
# Get all catalogs in order to use them to fill the database
book_catalog = BookCatalog.get_instance()
magazine_catalog = MagazineCatalog.get_instance()
album_catalog = AlbumCatalog.get_instance()
movie_catalog = MovieCatalog.get_instance()
loan_catalog = LoanCatalog.get_instance()
# Create controllers in order to create users
admin_controller = AdminController.get_instance()
client_controller = ClientController.get_instance()
movie_name = lambda: "The " + f.job() if f.random_int(
) % 2 == 0 else " ".join(f.words()).capitalize()
album_name = movie_name
names = lambda: ", ".join(
[f.name() for x in range(1 + f.random_int() % 9)])
date = lambda: int(time.time() - f.random_int() * f.random_int())
asin = lambda: "".join([
f.random_letter().upper()
if f.random_int() % 2 == 0 else str(f.random_digit())
for x in range(10)
])
phone_number = lambda: "".join([str(f.random_digit()) for x in range(
3)]) + "-" + "".join([str(f.random_digit()) for x in range(
3)]) + "-" + "".join([str(f.random_digit()) for x in range(4)])
# Don't commit until the end
app.classes.database_container.DatabaseContainer.commit_lock = True
# Fake data generator
f = Faker()
for b in range(NUM_BOOKS):
book_attributes = {
'author': f.name(),
'title': f.catch_phrase(),
'format': book_types[f.random_int() % len(book_types)],
'pages': f.random_int() % MAX_BOOK_PAGES,
'publisher': f.last_name(),
'year_of_publication': (f.random_int() % 100) + 1910,
'language': languages[f.random_int() % len(languages)],
'isbn_10': f.isbn10(),
'isbn_13': f.isbn13()
}
new_book = Book(book_attributes)
# Create copies of the same book - also done for every record type below
for c in range(COPIES):
book_catalog.add(new_book, True)
for m in range(NUM_MAGAZINES):
magazine_attributes = {
'title': f.word().upper(),
'publisher': f.last_name(),
'year_of_publication': f.random_int() % 100 + 1910,
'language': languages[f.random_int() % len(languages)],
'isbn_10': f.isbn10(),
'isbn_13': f.isbn13()
}
new_magazine = Magazine(magazine_attributes)
for c in range(COPIES):
magazine_catalog.add(new_magazine, True)
for m in range(NUM_MOVIES):
movie_attributes = {
'title': movie_name(),
'director': f.name(),
'producers': names(),
'actors': names(),
'language': languages[f.random_int() % len(languages)],
'subtitles': languages[f.random_int() % len(languages)],
'dubbed': languages[f.random_int() % len(languages)],
'release_date': date(),
'run_time': 60 + f.random_int() % (2 * 60)
}
new_movie = Movie(movie_attributes)
for c in range(COPIES):
movie_catalog.add(new_movie, True)
for a in range(NUM_ALBUMS):
album_attributes = {
'type': album_types[f.random_int() % len(album_types)],
'title': album_name(),
'artist': f.name(),
'label': f.word().upper(),
'release_date': date(),
'asin': asin()
}
new_album = Album(album_attributes)
for cop in range(COPIES):
album_catalog.add(new_album, True)
for u in range(NUM_USERS):
user_attributes = {
'firstName': f.first_name(),
'lastName': f.last_name(),
'physicalAddress': f.address().replace("\n", ", "),
'email': f.email(),
'phoneNumber': phone_number(),
'username': f.user_name(),
'password': f.password(),
'isAdmin': f.random_int() % 2,
'isLogged': f.random_int() % 2,
'lastLogged': int(time.time() - f.random_int() * f.random_int())
}
# Add user based on admin status
if user_attributes["isAdmin"] == 1:
admin_controller._admin_catalog.add(Admin(user_attributes), True)
elif user_attributes["isAdmin"] == 0:
client_controller._client_catalog.add(Client(user_attributes),
True)
client1 = dict(
(('firstName', "Aaron"), ('lastName', 'Doe'),
('physicalAddress',
'1451 De Maisonneuve Blvd. W. Montreal, QC H3G 1M8 Canada'),
('email', '*****@*****.**'), ('phoneNumber', '514-555-0001'),
('username', 'antman'), ('password', 'password1'), ('isAdmin', 0),
('isLogged', 1), ('lastLogged', 1537207100)))
admin1 = dict(
(('firstName', "Chloe"), ('lastName', 'Doe'),
('physicalAddress',
'1452 De Maisonneuve Blvd. W. Montreal, QC H3G 1M8 Canada'),
('email', '*****@*****.**'), ('phoneNumber', '514-555-0002'),
('username', 'catwoman'), ('password', 'password3'), ('isAdmin', 1),
('isLogged', 1), ('lastLogged', 1537207100)))
# create a new clients inside client table
client_controller._client_catalog.add(Client(client1), True)
admin_controller._admin_catalog.add(Admin(admin1), True)
print("- Finished filling database -")
# Turn off commit lock
app.classes.database_container.DatabaseContainer.commit_lock = False
database.commit_db()
return True
from app.database import sqlite_script
from app.controllers.client_controller import ClientController
from app.controllers.admin_controller import AdminController
from app.controllers.catalog_controller import CatalogController
from app.common_definitions import helper_functions
# Create and fill database with values - closes connection to
# Database was filled, objects are in memory if function call returns True
objects_in_memory = sqlite_script.initializeAndFillDatabase()
# Connect to the database through an abstracted object - this object must be imported into route files for use
databaseObject = DatabaseContainer.get_instance()
# Send the database object to all Controllers
catalog_controller = CatalogController.get_instance()
client_controller = ClientController.get_instance()
admin_controller = AdminController.get_instance()
if not objects_in_memory:
catalog_controller.load_database_into_memory()
client_controller.load_database_into_memory()
admin_controller.load_database_into_memory()
client_controller.load_loans_db_to_memory()
# Helper function to be used in front end
app.jinja_env.globals.update(
convert_epoch_to_datetime=helper_functions.convert_epoch_to_datetime)
from app import routes