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