Add Poster url and Add pandas

This commit is contained in:
Esa Kataja
2024-11-17 13:04:24 +02:00
parent b32e689d11
commit f8609ac317
5 changed files with 127 additions and 18 deletions
+6 -16
View File
@@ -27,25 +27,15 @@ def get_all_movies() -> list[Movie]:
return
def get_next_movie() -> Movie:
def get_next_movie() -> Movie | None:
sql = "SELECT * FROM movies WHERE is_watched = FALSE AND showtime > CURRENT_TIMESTAMP ORDER BY showtime ASC"
with get_db() as conn:
result = conn.execute(sql).fetchone()
if result is None:
result = conn.execute(sql).fetchdf()
if result.empty:
return None
movie = Movie(
id=str(result[0]),
name=result[1],
imdb_id=result[2],
actors=result[3],
release_date=result[4],
plot=result[5],
showtime=result[6],
is_watched=result[7],
created_at=result[8],
updated_at=result[9],
)
response_data = result.to_dict(orient="records")[0]
response_data["id"] = str(response_data["id"])
movie = Movie(**response_data)
return movie
return