Update views.py

Ekaterina-Vititneva · Aug 13, 2024 · bc535f0 · bc535f0
1 parent 432b15c
commit bc535f0
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/project_2/commerce/auctions/views.py b/project_2/commerce/auctions/views.py
@@ -103,6 +103,12 @@ def listing(request, title):
             return handle_comment(request, listing)
         elif 'close_listing' in request.POST:
             return handle_close_listing(request, listing)
+        elif 'add_to_watchlist' in request.POST:
+            user.watchlist.add(listing)
+            messages.success(request, "Listing added to your watchlist.")
+        elif 'remove_from_watchlist' in request.POST:
+            user.watchlist.remove(listing)
+            messages.success(request, "Listing removed from your watchlist.")
 
     comments = listing.comments.all()
 
@@ -113,6 +119,7 @@ def listing(request, title):
         "comments": comments,
     })
 
+
 
 def handle_bid(request, listing):
     form = BidForm(request.POST)