#3 Added basic create product page

controllers/web/endpoints.py

@@ -24,7 +24,6 @@ def get_user() -> dict[User|None]:
     if user_id != None:
         db = UserController()
         user = db.read_id(user_id)
-        print(user)
         
     return dict(user=user)
 

controllers/web/product.py

@@ -1,8 +1,9 @@
 from flask import Blueprint
 
-from flask import render_template, session, flash, request
+from flask import render_template, session, flash, request, redirect
 from controllers.database.product import ProductController
 from controllers.database.category import CategoryController
+from controllers.database.user import UserController
 
 blueprint = Blueprint("products", __name__, url_prefix="/products")
 
@@ -50,3 +51,21 @@ def category(category: str):
 def id(id: int):
     return "ID: " + str(id)
 
+
+# Launches the page to add a new product to the site
+@blueprint.route('/add')
+def add_product():
+    user_id = session.get('user_id')
+    
+    # User must be logged in to view this page
+    if user_id == None:
+        flash("Please Login to view this page")
+        return redirect('/login')
+    
+    db = UserController()
+    user = db.read_id(user_id)
+    if user == None or user.role != "Seller":
+        flash("You must be logged in as a Seller to view this page")
+        return redirect('/')
+
+    return render_template('index.html', content='new_product.html')

controllers/web/user.py

@@ -86,5 +86,6 @@ def signup():
 # Function responsible for handling logouts from the site
 @blueprint.route('/logout')
 def logout():
-    session.pop('user_id')
+    # Clear the current user from the session if they are logged in
+    session.pop('user_id', None)
     return redirect("/")