diff --git a/controllers/web/product.py b/controllers/web/product.py index 395241a..28c5340 100644 --- a/controllers/web/product.py +++ b/controllers/web/product.py @@ -75,7 +75,7 @@ def display_add_product(): # User needs to be logged in as a seller to view this page if not is_role("Seller"): flash("You must be logged in as a seller to view this page!") - return redirect("/", code=302) + return redirect("/", code=401) return render_template('index.html', content='new_product.html') @@ -90,7 +90,7 @@ def add_product(): # User needs to be logged in as a seller to view this page if not is_role("Seller"): flash("You must be logged in as a seller to view this page!") - return redirect("/", code=302) + return redirect("/", code=401) file = request.files.get('image') @@ -116,17 +116,18 @@ def add_product(): db = ProductController() db.create(product) - return render_template('index.html', content='new_product.html') + return redirect('/products/ownproducts') @blueprint.route('/ownproducts') def display_own_products(): + """ Display products owned by the currently logged in seller """ user_id = session.get('user_id') # User must be logged in as seller to view page if not is_role("Seller"): flash("You must be logged in as a seller to view this page!") - return redirect("/", code=302) + return redirect("/", code=401) db = ProductController() products = db.read_user(user_id)