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