def wishlist(request): current_user = userassert(request) try: wishlist = Wishlist.objects.get(customer__name=current_user) except ObjectDoesNotExist: try: c = Customer.objects.get(name=current_user) except ObjectDoesNotExist: c = Customer(name=current_user) c.save() wishlist = Wishlist(customer=c) wishlist.save() context = {'wishlist': wishlist} return render(request, 'osp/wishlist.html', context)
def add_to_wishlist(request, prod_name): current_user = userassert(request) item = Item.objects.get(name=prod_name, listed=True) try: c = Customer.objects.get(name=current_user) except: c = Customer(name=current_user) c.save() try: wishlist = Wishlist.objects.get(customer__name=current_user) except ObjectDoesNotExist: wishlist = Wishlist(customer=c) wishlist.save() wishlist.item_set.add(item) return HttpResponseRedirect(reverse('osp:wishlist'))