current

movie-db/.ipynb_checkpoints/moviedb_func-checkpoint.py

@@ -0,0 +1,69 @@
+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"))