Add Poster url and Add pandas
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user