def post(self, id): ''' Leave a movie review ''' TokenAuthenticator(request.headers.get('Authorization')).authenticate() is_valid_integer(id) validate_rating(request.json['rating']) validate_review(request.json['review']) session = Session() movie = session.query(Movie).filter(Movie.movieID == id).one_or_none() if not movie: raise NotFound query = session.query(MovieReview).filter( MovieReview.movieID == id, MovieReview.userID == g.userID).one_or_none() if query: # We disallow a user from leaving more than one review for the same movie. raise Forbidden query = session.query(Watchlist).filter( Watchlist.movieID == id, Watchlist.userID == g.userID).one_or_none() if not query: # When a user leaves a review for a film, we add that # film to his Watchlist if it isn't there already. session.add(Watchlist(id, g.userID)) request.json['rating'] = float(request.json['rating']) session.add( MovieReview(id, g.userID, request.json['rating'], request.json['review'])) movie.ratings_sum += request.json['rating'] movie.review_count += 1 session.commit() session.close() return {'message': 'Review saved.'}, 201
def post(self): ''' Add a FilmFinder to your Banned List. ''' TokenAuthenticator(request.headers.get('Authorization')).authenticate() is_valid_integer(request.json['userID']) session = Session() query = session.query(User).filter( User.userID == request.json['userID']).one_or_none() if not query: raise NotFound query = session.query(BannedList).filter( BannedList.userID == g.userID, BannedList.bannedUserID == request.json['userID']).one_or_none() if query or g.userID == request.json['userID']: # If the FilmFinder being banned is already in the user's Banned List, # or if a user is attempting to ban himself, raise an Exception. raise Forbidden # Remove a user's subscription to a FilmFinder when banning that FilmFinder. session.query(Subscription).filter( Subscription.userID == g.userID, Subscription.subscribedUserID == request.json['userID']).delete() session.add(BannedList(g.userID, request.json['userID'])) session.commit() session.close() return {'message': 'Reviewer banned.'}, 201
def post(self): # Get params username = request.json.get('username') username = cleanString(username) yob = request.json.get('yob') email = request.json['email'] email = cleanString(email) isValidEmail(email) password = request.json['password'] session = Session() if session.query(User).filter(User.email == email).first(): raise BadRequest hashed_password = HashGenerator(password).generate() new_user = User(username, email, hashed_password, yob) session.add(new_user) session.commit() session.close() response = {'message': 'Registration successful.'} return response, 201
def post(self, userID): TokenAuthenticator(request.headers.get('Authorization')).authenticate() session = Session() is_valid_integer(userID) # User not Found if not session.query(User).filter(User.userID == userID).first(): raise NotFound cID = g.userID # Can't subscribe to oneself. if cID == userID: raise BadRequest new = set(session.query(Wishlist.movieID).filter(Wishlist.userID == userID)) cur = set(session.query(Wishlist.movieID).filter(Wishlist.userID == cID)) # Add only absent movies toAdd = new.difference(cur) if toAdd: for movie in toAdd: for m in movie: wl = Wishlist(m, cID) session.add(wl) session.commit() session.close() response = {'message': 'Wishlisted Movies imported'} return response, 201 else: response = {'message': 'No Movies were imported'} return response, 200
def post(self): ''' Adds a movie to the users' Watchlist. ''' TokenAuthenticator(request.headers.get('Authorization')).authenticate() movieID = request.json.get('movieID') is_valid_integer(movieID) new_watchlist = Watchlist(movieID, g.userID) session = Session() session.add(new_watchlist) try: session.commit() session.close() except IntegrityError: #If Watchlist already present session.rollback() raise BadRequest response = {'message': 'Movie added to Watchlist.'} return response, 201
def post(self): ''' Subscribe to the user. ''' TokenAuthenticator(request.headers.get('Authorization')).authenticate() subscribedUserID = request.json.get('userID') curUserID = g.userID # Can't subscribe to oneself. if curUserID == int(subscribedUserID): raise BadRequest subscribe = Subscription(curUserID, subscribedUserID) session = Session() session.add(subscribe) try: session.commit() except IntegrityError: #If subscription already present session.rollback() raise BadRequest response = {'message': 'Subscribed to User'} return response, 201