Refactor start.py to remove dotenv and db dependencies; implement new_table function for table creation and streamline admin function
This commit is contained in:
@@ -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()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user