async def remove(request, user, watchlist, movie_id):
logger.debug(movie_id)
with db_session():
watchlist = Watchlist.get(id=watchlist.id)
MovieItem.select(lambda m: m.imdb_id == movie_id and m.watchlist.id ==
watchlist.id).delete()
return response.empty()
async def add(request, user, watchlist, movie_id):
logger.debug(movie_id)
if not is_in_default_watchlist(movie_id, user):
with db_session():
watchlist = Watchlist.get(id=watchlist.id)
item = MovieItem(imdb_id=movie_id,
date=datetime.now(),
watchlist=watchlist)
return response.empty()
async def seen(request, user, watchlist, movie_id):
logger.debug(movie_id)
with db_session():
# TODO: Catche error if does not exist
watchlist = Watchlist.get(id=watchlist.id)
user = User.get(id=user.id)
MovieItem.select(lambda m: m.imdb_id == movie_id and m.watchlist.id ==
watchlist.id).delete()
date_string = request.args.get("date", None)
if date_string:
date = datetime.strptime(date_string, "%Y-%m-%d")
else:
date = datetime.now()
movie = MovieItem(imdb_id=movie_id, date=date, user=user)
return response.empty()