#2 Added functionality for user to signup as a Seller

controllers/web/user.py

@@ -3,6 +3,7 @@ from flask import Blueprint
 from flask import render_template, redirect, request, session, flash
 from controllers.database.user import UserController
 from models.users.customer import Customer
+from models.users.seller import Seller
 from hashlib import sha512
 
 # Blueprint to append user endpoints to
@@ -54,14 +55,27 @@ def signup():
         flash(error)
         return redirect("/signup")
 
-    database.create(Customer(
-        request.form['username'],
-        sha512(request.form['password'].encode()).hexdigest(), # Hashed as soon as it is recieved on the backend
-        request.form['firstname'],
-        request.form['lastname'],
-        request.form['email'],
-        "123"
-    ))
+    # Signup as seller or Customer
+    if request.form.get('seller'):
+        user = Seller(
+            request.form['username'],
+            sha512(request.form['password'].encode()).hexdigest(), # Hashed as soon as it is recieved on the backend
+            request.form['firstname'],
+            request.form['lastname'],
+            request.form['email'],
+            "123"
+        )
+    else:
+        user = Customer(
+            request.form['username'],
+            sha512(request.form['password'].encode()).hexdigest(), # Hashed as soon as it is recieved on the backend
+            request.form['firstname'],
+            request.form['lastname'],
+            request.form['email'],
+            "123"
+        )
+        
+    database.create(user)
 
     # Code 307 Preserves the original request (POST)
     return redirect("/login", code=307)