order and api and more

app/services/address_label_service.py

@@ -4,6 +4,11 @@ import os
 from pathlib import Path
 from jinja2 import Environment, FileSystemLoader
 from weasyprint import HTML
+import logging
+import asyncio
+from concurrent.futures import ThreadPoolExecutor
+
+logger = logging.getLogger(__name__)
 
 class AddressLabelService:
     def __init__(self):
@@ -16,8 +21,9 @@ class AddressLabelService:
         self.return_address_path = "file://" + os.path.abspath("app/data/assets/images/ccrcardsaddress.png")
         self.output_dir = "app/data/cache/tcgplayer/address_labels/"
         os.makedirs(self.output_dir, exist_ok=True)
+        self.executor = ThreadPoolExecutor()
     
-    def generate_labels_from_csv(self, csv_path: str, label_type: Literal["dk1201", "dk1241"]) -> List[str]:
+    async def generate_labels_from_csv(self, csv_path: str, label_type: Literal["dk1201", "dk1241"]) -> List[str]:
         """Generate address labels from a CSV file and save them as PDFs.
         
         Args:
@@ -29,17 +35,30 @@ class AddressLabelService:
         """
         generated_files = []
 
-        with open(csv_path, 'r') as csvfile:
-            reader = csv.DictReader(csvfile)
-            for row in reader:
-                # Generate label for each row
-                pdf_path = self._generate_single_label(row, label_type)
-                if pdf_path:
-                    generated_files.append(str(pdf_path))
+        # Read CSV file in a thread pool
+        loop = asyncio.get_event_loop()
+        rows = await loop.run_in_executor(self.executor, self._read_csv, csv_path)
+        
+        for row in rows:
+            # if value of Value Of Products is greater than 50, skip
+            if row.get('Value Of Products') and float(row['Value Of Products']) > 50:
+                logger.info(f"Skipping order {row.get('Order #')} because value of products is greater than 50")
+                continue
+
+            # Generate label for each row
+            pdf_path = await self._generate_single_label(row, label_type)
+            if pdf_path:
+                generated_files.append(str(pdf_path))
 
         return generated_files
 
-    def _generate_single_label(self, row: Dict[str, str], label_type: Literal["dk1201", "dk1241"]) -> Optional[str]:
+    def _read_csv(self, csv_path: str) -> List[Dict[str, str]]:
+        """Read CSV file and return list of rows."""
+        with open(csv_path, 'r') as csvfile:
+            reader = csv.DictReader(csvfile)
+            return list(reader)
+
+    async def _generate_single_label(self, row: Dict[str, str], label_type: Literal["dk1201", "dk1241"]) -> Optional[str]:
         """Generate a single address label PDF.
         
         Args:
@@ -67,11 +86,15 @@ class AddressLabelService:
             # Render HTML
             html_content = self.templates[label_type].render(**template_data)
 
-            # Generate PDF
+            # Generate PDF in a thread pool
+            loop = asyncio.get_event_loop()
             pdf_path = self.output_dir + f"{row['Order #']}_{label_type}.pdf"
-            HTML(string=html_content).write_pdf(str(pdf_path))
+            await loop.run_in_executor(
+                self.executor,
+                lambda: HTML(string=html_content).write_pdf(str(pdf_path))
+            )
             return pdf_path
 
         except Exception as e:
-            print(f"Error generating label for order {row.get('Order #', 'unknown')}: {str(e)}")
+            logger.error(f"Error generating label for order {row.get('Order #', 'unknown')}: {str(e)}")
             return None