Add Movie endpoint
This commit is contained in:
@@ -0,0 +1,51 @@
|
||||
from models.movie import Movie
|
||||
from lib.database.database import get_db
|
||||
|
||||
|
||||
def get_all_movies() -> list[Movie]:
|
||||
sql = "SELECT * FROM movies"
|
||||
with get_db() as conn:
|
||||
result = conn.execute(sql).fetchall()
|
||||
movies = []
|
||||
for row in result:
|
||||
movie = Movie(
|
||||
id=str(row[0]),
|
||||
name=row[1],
|
||||
imdb_id=row[2],
|
||||
actors=row[3],
|
||||
release_date=row[4],
|
||||
plot=row[5],
|
||||
showtime=row[6],
|
||||
is_watched=row[7],
|
||||
created_at=row[8],
|
||||
updated_at=row[9],
|
||||
)
|
||||
movies.append(movie)
|
||||
|
||||
return movies
|
||||
|
||||
return
|
||||
|
||||
|
||||
def get_next_movie() -> Movie:
|
||||
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:
|
||||
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],
|
||||
)
|
||||
|
||||
return movie
|
||||
|
||||
return
|
||||
Reference in New Issue
Block a user