def delete(self, wishlist_user): """ Removes all wishlists of a user""" app.logger.info('Request to delete all wishlists of a user') wishlists = Wishlist.find_by_user(wishlist_user) for wishlist in wishlists: wishlist.delete_wishlist() return '', status.HTTP_204_NO_CONTENT
def get(self): """ Retrieves all the wishlists """ app.logger.info('Request to list wishlists') wishlists = [] wishlist_user = request.args.get('wishlist_user') wishlist_name = request.args.get('wishlist_name') app.logger.info('Request to list wishlists of user %s with name: %s', wishlist_user, wishlist_name) if wishlist_user: a = Wishlist.find_by_user(wishlist_user) ay = [w.serialize() for w in a] if wishlist_name: b = Wishlist.find_by_name(wishlist_name) bee = [w.serialize() for w in b] wishlists = [x for x in ay if x in bee] else: wishlists = ay else: wishlists = [w.serialize() for w in Wishlist.all()] app.logger.info('[%s] Wishlists returned', len(wishlists)) return wishlists, status.HTTP_200_OK
def test_find_by_user(self): """ Find Wishlist by Name """ Wishlist("mike's wishlist", "mike").save() Wishlist("joan's wishlist", "joan").save() wishlists = Wishlist.find_by_user("mike") self.assertNotEqual(len(wishlists), 0) self.assertEqual(wishlists[0].user, "mike") self.assertEqual(wishlists[0].name, "mike's wishlist")