database.py

import mysql.connector

def setup():
    # Set up a connection to the MySQL server
    cnx = mysql.connector.connect(
    host='localhost',
    user='root',
    password='AS324S1'
    )

    # Create a cursor object to execute queries
    cursor = cnx.cursor()

    # Check if the database exist
    cursor.execute("SHOW DATABASES LIKE 'mydatabase'")
    exists = cursor.fetchone()

    if exists:
        print("The database already exists")
    else:
        # Create the database if it doesn't exist
        cursor.execute("CREATE DATABASE mydatabase")
        print("The database was created")

    # Connect to the database
    cnx.database = "mydatabase"

    # Create a table if it doesn't exist
    cursor.execute("""
    CREATE TABLE IF NOT EXISTS mytable (
        name VARCHAR(255) PRIMARY KEY,
        activity VARCHAR(255) 
    )
    """)

    # Insert some data into the table
    # data = [
    # ("test1", "1 1"),
    # ]
    # cursor.executemany("INSERT INTO mytable (name, age) VALUES (%s, %s)", data)

    # Commit the changes to the database
    cnx.commit()

    # Close the cursor and connection
    cursor.close()
    cnx.close()