feat: CRUD for product barcode images

barcodes/images_uploader/__init__.py

@@ -0,0 +1 @@
+from .images_uploader import BarcodeImagesUploader

barcodes/images_uploader/base.py

@@ -0,0 +1,18 @@
+from abc import abstractmethod
+
+from fastapi import UploadFile
+
+
+class BaseImagesUploader:
+
+    @abstractmethod
+    def get_url(self, filename: str) -> bytes:
+        pass
+
+    @abstractmethod
+    def delete(self, filename: str):
+        pass
+
+    @abstractmethod
+    async def upload(self, upload_file: UploadFile) -> str:
+        pass

barcodes/images_uploader/images_uploader.py

@@ -0,0 +1,46 @@
+from pathlib import Path
+from uuid import uuid4
+
+from aioshutil import copyfileobj
+from fastapi import UploadFile
+
+from barcodes.images_uploader.base import BaseImagesUploader
+from barcodes.pdf import PDFGenerator
+from constants import APP_PATH, API_ROOT
+
+
+class BarcodeImagesUploader(BaseImagesUploader):
+    def __init__(self):
+        self.relative_path = Path("static/images/product_barcodes")
+        self.storage_path = APP_PATH / self.relative_path
+        if not Path.exists(self.storage_path):
+            Path.mkdir(self.storage_path)
+
+    def get_url(self, filename: str) -> str:
+        file_location = self.relative_path / filename
+        return f"{API_ROOT}/{file_location}"
+
+    def delete(self, filename: str):
+        file_location = self.storage_path / filename
+        if file_location.exists():
+            file_location.unlink()
+
+    async def upload(self, upload_file: UploadFile) -> str:
+        # Create temp file in filesystem
+        temp_filename = str(uuid4()) + '.' + upload_file.filename.split('.')[-1]
+        temp_file_location = self.storage_path / temp_filename
+        with open(temp_file_location, 'wb') as buffer:
+            await copyfileobj(upload_file.file, buffer)
+
+        # Generate PDF file and save it
+        res_filename = str(uuid4()) + '.pdf'
+        res_file_location = f"{self.storage_path}/{res_filename}"
+        temp_file_url = f"{self.relative_path}/{temp_filename}"
+
+        pdf_gen = PDFGenerator()
+        pdf_gen.generate_barcode_image(temp_file_url, res_file_location)
+
+        # Remove temp file
+        self.delete(temp_filename)
+
+        return res_filename