#4 #6 Completed functionlity to update products in place. Form still needs tidying up

utils/file_utils.py

@@ -1,7 +1,11 @@
+import os
+import uuid
+import pathlib
+
 ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg', 'gif'}
 
 
-def allowed_file(filename):
+def allowed_file(filename) -> bool:
     """ Ensures only filenames ending with the correct extension are allowed.
         Note: This does not verify that the content inside of the file
         matches the type specified
@@ -9,3 +13,31 @@ def allowed_file(filename):
     return '.' in filename and \
            filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
 
+def save_image(file) -> str | None:
+    """ Saves a given file to disk with a random UUID4 generated
+        filename. Returns the filename as a string.
+    """
+    # Ensure that the correct file type is uploaded
+    if file is None or not allowed_file(file.filename):
+        return None
+
+    # Create the product object and push to database
+    filename = str(uuid.uuid4()) + pathlib.Path(file.filename).suffix
+
+    path = os.environ.get('FILESTORE')
+    file.save(os.path.join(path, filename))
+    return filename
+
+def create_directory(dir: str):
+    """ Creates the given directory string is not alreay made """
+    try:
+        os.makedirs(dir)
+    except FileExistsError:
+        pass
+
+def remove_file(dir: str):
+    """ Removes a given file if it is present at the given dir """
+    try:
+        os.remove(dir)
+    except FileNotFoundError:
+        pass