diff --git a/src/lib/database/admin.py b/src/lib/database/admin.py index 4656c1f..125b6d9 100644 --- a/src/lib/database/admin.py +++ b/src/lib/database/admin.py @@ -32,11 +32,33 @@ def remove_user(user_id: str): def add_movie(movie: Movie): - pass + sql = "INSERT INTO movies (name, imdb_id, actors, release_date, plot, showtime) VALUES (?, ?, ?, ?, ?, ?)" + try: + db_run( + sql, + ( + movie.name, + movie.imdb_id, + movie.actors, + movie.release_date, + movie.plot, + movie.showtime, + ), + ) + except: + return Message(message="Failed to add movie", message_type="error") + + return Message(message="Movie added successfully", message_type="success") -def remove_movie(movie_id: str): - pass +def set_movie_watched(movie_id: str): + sql = f"UPDATE movies SET is_watched = TRUE WHERE id = '{movie_id}'" + try: + db_run(sql) + except: + return Message(message="Failed to set movie as watched", message_type="error") + + return Message(message="Movie set as watched successfully", message_type="success") def add_card(card: Card):