diff --git a/start.py b/start.py index 77e6e86..5b754b4 100644 --- a/start.py +++ b/start.py @@ -1,8 +1,8 @@ -from dotenv import load_dotenv -import db -import os +from migration.alembic_scipt import auto_lookup +import time import logging import auth.api_security as auth +import os #Logging @@ -18,41 +18,19 @@ try: except Exception as e: logging.exception('An error occurred') -load_dotenv() -db_name = os.getenv("DB_NAME") -db_psswd = os.getenv("DB_PASSWORD") -db_user = os.getenv("DB_USER") -db_port = os.getenv("DB_PORT") or "3306" -db_host = os.getenv("DB_HOST") -if db_host == None: - logging.error("no hostname provided") + +def new_table(): + auto_lookup() + +def admin(): + token = auth.admin_token() + print(token) + logging.info(token) + time.sleep(300) + os.remove("app.log") os._exit(1) -db.connect(db_user, db_psswd, db_host, int(db_port), db_name) -def new_table(name): - cursor = db.connection.cursor() - if name == "tokens": - executed = "CREATE TABLE tokens (tokens VARCHAR(255) PRIMARY KEY, `time` VARCHAR(255), end_date VARCHAR(255))" - cursor.execute(executed) - db.connection.commit() - logging.info("Table 'tokens' created") - - if name == "luggage": - executed = "CREATE TABLE luggage (clothes VARCHAR(50) PRIMARY KEY, weight DECIMAL(20,2), pieces INT(11))" - cursor.execute(executed) - db.connection.commit() - logging.info("Table 'luggage' created") - - if name == "user": - executed = "CREATE TABLE user (username VARCHAR(50) PRIMARY KEY, password VARCHAR(255))" - cursor.execute(executed) - db.connection.commit() - logging.info("Table 'user' created") - - -new_table("tokens") -new_table("luggage") -new_table("user") -print(auth.admin_token()) +new_table() +admin()