populate_database.py

# This file is part of the Silence framework.
# Silence was developed by the IISSI1-TI team
# (Agustín Borrego, Daniel Ayala, Carlos Ortiz, Inma Hernández & David Ruiz)
# and it is distributed as open source software under the GNU-GPL 3.0 License.

from dal.database.db_connection import get_conn

# Method for destroying and creating the database tables
# This shouldn't be modified, it uses the create_database.sql file
def create_database(verbose=False):
    conn = get_conn()
    cursor = conn.cursor()
    with open("create_database.sql", "r", encoding="utf-8") as f:
        for stmt in f.read().split(";"):
            if not stmt or stmt.strip() == "": continue  # Skip blank lines
            if verbose: print(stmt + ";")
            cursor.execute(stmt)
    conn.commit()
    cursor.close()

#######################################
if __name__ == "__main__":
    print("Creating and populating the database...")
    create_database(verbose=True)
    print("Success!")