#3 Added ability for images to be uploaded to the site

.gitignore

@@ -4,6 +4,7 @@
 
 ### SQLite 3 Database ###
 data/
+static/assets/img/products/
 
 ### CICD Registration files ###
 cicd/runner-data

controllers/web/product.py

@@ -8,7 +8,10 @@ from controllers.database.category import CategoryController
 from controllers.database.user import UserController
 
 from datetime import datetime
-from werkzeug import secure_filename
+from werkzeug.utils import secure_filename
+import os
+import uuid
+import pathlib
 
 ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg', 'gif'}
 
@@ -101,18 +104,17 @@ def add_product():
     file = request.files.get('image')
     
     # Ensure that the correct file type is uploaded
-    if file == None OR NOT allowed_file(file.filename):
+    if file == None or not allowed_file(file.filename):
         flash("Invalid File Uploaded")
         return redirect("/add")
 
     # Create the product object and push to database
-    filename = secure_filename(file.filename)
-    file.save(os.path.join('static/assets/img/products/', secure_filename))
-    file.save
+    filename = str(uuid.uuid4()) + pathlib.Path(file.filename).suffix
+    file.save(os.path.join('static/assets/img/products/', filename))
 
     product = Product(
         request.form.get('name'),
-        filename
+        filename,
         request.form.get('description'),
         request.form.get('cost'),
         request.form.get('category'),

templates/content.html

@@ -7,7 +7,7 @@
                 <div class="product-title">{{product.name}}</div>
                     <div class="product-information">
                         <div class="product-image">
-                            <img src="{{url_for('static', filename=product.image)}}" alt="Brake Disks"/>
+                            <img src="{{url_for('static', filename='assets/img/products/' + product.image)}}" alt="Brake Disks"/>
                         </div>
                         <div class="product-details">
                             <div class="product-price">£{{product.cost}}</div>

templates/new_product.html

@@ -6,7 +6,7 @@
         <input type="file" id="image" name="image" accept="image/x" required>
         <select name="category" id="category">
             {% for category in categories %}
-                <option value="{{category.name}}">{{category.name}}</option>
+                <option value="{{category.id}}">{{category.name}}</option>
             {% endfor %}
         </select>
         <input type="number" id="cost" name="cost" placeholder=10.99 min=0 step=any required>