28 lines
643 B
Python
28 lines
643 B
Python
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)
|