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