from typing import Optional

from folkugat_web.dal.sql import Connection, get_connection


def create_db(con: Optional[Connection] = None):
    with get_connection(con) as con:
        create_playlists_table(con)


def drop_playlists_table(con: Connection):
    query = "DROP TABLE IF EXISTS playlists"
    cur = con.cursor()
    cur.execute(query)


def create_playlists_table(con: Connection):
    query = """
      CREATE TABLE IF NOT EXISTS playlists (
        id INTEGER PRIMARY KEY,
        session_id INTEGER NOT NULL,
        set_id INTEGER NOT NULL,
        tema_id INTEGER
    )
    """
    cur = con.cursor()
    cur.execute(query)