From c8884ebeecdca2dd8ef4f43e58d68a54cb4bc3c7 Mon Sep 17 00:00:00 2001
From: Esa Kataja <ekataja@gmail.com>
Date: Sat, 9 Nov 2024 19:36:35 +0200
Subject: [PATCH] Add movie admin endpoints

---
 src/lib/database/admin.py | 28 +++++++++++++++++++++++++---
 1 file changed, 25 insertions(+), 3 deletions(-)

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):