import DBcm


def create_movie_database(db_name="movie_db.db"):

    create_my_list = """
    create table if not exists movie_list (
         id integer not null primary key autoincrement,
        titel varchar(64) not null,
        genre_id integer not null,
        regie_id integer not null
    )
    """

    create_genre = """
    create table if not exists genre (
         id integer not null primary key autoincrement,
        name varchar(64) not null
    )
    """

    create_regie = """
    create table if not exists regie (
         id integer not null primary key autoincrement,
        surname varchar(64) not null,
        lastname varchr(64) not null
    )
    """
    
    create_medium = """
    create table if not exists medium (
         id integer not null primary key autoincrement,
         medium varchar(64) not null
    )
    """

    with DBcm.UseDatabase(db_name) as db:
        db.execute(create_my_list)
        db.execute(create_genre)
        db.execute(create_regie)
        db.execute(create_medium)


def all_genres(db_name="movie_db.db"):
    ALL_GENRE = "SELECT * from genre"
    with DBcm.UseDatabase(db_name) as db:
        db.execute(ALL_GENRE)
        all_genre = [i[1] for i in db.fetchall()]
    return all_genre


def search_genre_id(db_name="movie_db.db", genre_name=str):
    GENRE_QUERY = """
            select id from genre
            where name = ?
            """
    try:
        with DBcm.UseDatabase(db_name) as db:
            db.execute(GENRE_QUERY, (genre_name,))
            genre_id = db.fetchone()[0]
        return int(genre_id)
    except:
        return int(0)


if __name__ == "__main__":
    create_movie_database()
    print(all_genres())
    print(search_genre_id(genre_name="war"))