Slugy, webp, hromada zmen

src/utils/image.py

@@ -127,11 +127,12 @@ class PhotoEditor:
         return img_io
 
     def write_to_file(
-            self, quality: int = 90, dest_path: Optional[str] = None, dest_filename: Optional[str] = None
+            self, quality: int = 90, dest_path: Optional[str] = None, dest_filename: Optional[str] = None,
+            format_: Optional[str] = "JPEG"
     ) -> str:
         check_directories(dest_path or self.path)
 
         dest = f"{dest_path or self.path}/{dest_filename or self.filename}"
-        self.img.save(dest, 'JPEG', quality=quality)
+        self.img.save(dest, format_, quality=quality)
 
         return dest

src/utils/upload.py

@@ -1,16 +1,27 @@
+import os
+import re
 import uuid
 from strawberry.file_uploads import Upload
 
 from utils.file import check_directories
 
 
-async def handle_file_upload(file: Upload, path: str) -> str:
+async def handle_file_upload(file: Upload, path: str, filename_maxlength: int = 64, uid_prefix: bool = True, overwrite: bool = True) -> str:
     check_directories(path)
 
-    filename = f"{uuid.uuid4()}-{file.filename}"
+    prefix = f"{uuid.uuid4()}-" if uid_prefix else ""
+    filename = f"{prefix}{file.filename}"[-1 * filename_maxlength:]
+
+    # sanitize filename
+    filename = re.sub('[^\w_. -]', '', filename).replace(" ", "-")
+
+    target_path = f"{path}/{filename}"
+
+    if os.path.exists(target_path) and not overwrite:
+        raise FileExistsError(f"File {filename} already exists")
 
     content = await file.read()
-    image = open(path + "/" + filename, "wb")
+    image = open(target_path, "wb")
     image.write(content)
     image.close()