scripts/test.py

from dotenv import load_dotenv

load_dotenv()
import os

import mysql.connector
from mysql.connector import errorcode

config = {
    "host": os.environ.get("HOSTNAME"),
    "user": os.environ.get("UID"),
    "password": os.environ.get("PASSWORD"),
    "database": os.environ.get("DATABASE"),
}

print(os.environ.get("HOSTNAME"))

try:
    cnx = mysql.connector.connect(**config)
    print("Connection established")
except mysql.connector.Error as err:
    if err.errno == errorcode.ER_ACCESS_DENIED_ERROR:
        print("Something is wrong with username or password")
    elif err.errno == errorcode.ER_BAD_DB_ERROR:
        print("Database does not exist")
    else:
        print(err)
else:
    cursor = cnx.cursor()

    cursor.execute("DROP TABLE IF EXISTS api_key")
    cursor.execute("DROP TABLE IF EXISTS auth")
    cursor.execute(
        "CREATE TABLE IF NOT EXISTS auth(username VARCHAR(15) PRIMARY KEY, password TEXT, email VARCHAR(50))"
    )
    cursor.execute(
        "CREATE TABLE IF NOT EXISTS api_key(username VARCHAR(15),apikey TEXT, FOREIGN KEY (username) REFERENCES auth(username))"
    )
    cursor.execute("ALTER TABLE auth ADD is_verified BOOLEAN NOT NULL DEFAULT(false)")

    # QUERY = ('INSERT INTO {coll_name} '
    #                 '(username, password, email) '
    #                 'VALUES '
    #                 '(%s, %s, %s)').format(coll_name="auth")

    # testlist=[("test2","test2","[email protected]"),("test1","test1","[email protected]")]
    # cursor.executemany(QUERY, testlist)

    # QUERY = ('SELECT {cols} FROM {table_name} WHERE email="[email protected]"').format(cols="*", table_name="auth")
    # cursor.execute(QUERY)
    # for i in cursor.fetchall():
    #     print(i)

    cnx.commit()
    cursor.close()
    cnx.close()
# from jose import jwt
# print(jwt.encode("bruhh"))