70 lines
1.7 KiB
Python
70 lines
1.7 KiB
Python
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"))
|