gen_example_data.py

from utilities import util
from utilities import test_utils


def reset(table: str):
    connection = util.make_connection()

    if table != "no-drop":
        util.exec_query(connection, "DROP TABLE :table", table=table)

    # SQLite optimizations
    util.exec_query(connection, "PRAGMA synchronous = NORMAL")
    util.exec_query(connection, "PRAGMA mmap_size = 1000000000")

    # ! This operation may not be supported, disable if you run into issues
    util.exec_query(connection, "PRAGMA journal_mode = WAL")

    # Projects Data
    util.exec_query(
        connection,
        """CREATE TABLE \"projects\"(
            type TEXT NOT NULL,
            author INT NOT NULL,
            title TEXT NOT NULL,
            description TEXT NOT NULL,
            body TEXT NOT NULL,
            icon TEXT,
            url TEXT NOT NULL UNIQUE,
            status TEXT NOT NULL DEFAULT \"draft\",
            category TEXT NOT NULL,
            uploaded INT NOT NULL,
            updated INT NOT NULL,
            mod_message TEXT,
            downloads INT NOT NULL DEFAULT 0, 
            featured_until INT, 
            licence TEXT, 
            dependencies TEXT)
    """,
    )

    # Versions Data
    util.exec_query(
        connection,
        """CREATE TABLE IF NOT EXISTS versions(
        name TEXT NOT NULL,
        description TEXT NOT NULL,
        primary_download TEXT NOT NULL,
        resource_pack_download TEXT,
        minecraft_versions TEXT NOT NULL,
        version_code TEXT NOT NULL,
        project INT NOT NULL
    );
    """,
    )

    # User data
    util.exec_query(
        connection,
        """CREATE TABLE IF NOT EXISTS users (
        username TEXT NOT NULL UNIQUE, 
        token TEXT NOT NULL UNIQUE, 
        role TEXT NOT NULL, 
        bio TEXT, 
        github_id int UNIQUE,
        discord_id int UNIQUE,
        badges TEXT,
        profile_icon TEXT NOT NULL,
        join_date INTEGER NOT NULL
    )""",
    )

    # Banned User Data
    util.exec_query(
        connection,
        """CREATE TABLE IF NOT EXISTS banned_users (
        id int NOT NULL UNIQUE,
        expires int,
        reason TEXT
    )""",
    )

    # Notification Data
    util.exec_query(
        connection,
        """CREATE TABLE IF NOT EXISTS notifs(
        message TEXT NOT NULL,
        description TEXT NOT NULL,
        read BOOL NOT NULL,
        type TEXT NOT NULL,
        user INT NOT NULL
    );
    """,
    )

    # Report Data
    util.exec_query(
        connection,
        """CREATE TABLE IF NOT EXISTS reports(
        message TEXT NOT NULL,
        reporter INT NOT NULL,
        project INT NOT NULL
    );
    """,
    )

    # Comment data
    util.exec_query(
        connection,
        """CREATE TABLE IF NOT EXISTS comments(
        thread_id INT,
        message TEXT NOT NULL,
        author INT NOT NULL,
        sent INT NOT NULL,
        parent_id INT
    );
    """,
    )

    # Follow data
    util.exec_query(
        connection,
        """CREATE TABLE IF NOT EXISTS follows(
        follower INT,
        followed INT
    );
    """,
    )

    # save and exit
    connection.commit()


if __name__ == "__main__":
    reset("no-drop")

    util.commit_query(
        """INSERT INTO users (username, token, role, bio, github_id, profile_icon, join_date) VALUES ("HoodieRocks", "LOREMIPSUM", "admin", "rock", 123897432978, "https://example.com/", 0)""",
    )

    util.commit_query(
        """INSERT INTO users (username, token, role, bio, github_id, profile_icon, join_date) VALUES ("sacrifice", "POTATOES", "default", "paper", 238746238746, "https://example.com/1", 0)""",
    )

    test_utils.commit_fake_project(10)