f

app/services/tcgplayer_api.py

@@ -11,6 +11,7 @@ from jinja2 import Environment, FileSystemLoader
 from weasyprint import HTML
 import json
 import time
+import re
 
 logger = logging.getLogger(__name__)
 
@@ -240,8 +241,11 @@ class TCGPlayerAPIService:
         response = self.requests_util.send_request(url, method='POST', cookies=self.cookies, json=payload)
         if response:
             # get filename from response headers
-            filename = response.headers.get('Content-Disposition').split('filename=')[1].strip('"')
+            header = response.headers.get('Content-Disposition', '')
+            match = re.search(r'filename="?([^";]+)"?', header)
+            filename = match.group(1) if match else f'packingslip{datetime.now().strftime("%Y%m%d_%H%M%S")}.pdf'
             output_filename = f'/app/tmp/{filename}'
+            
             # save file to disk
             with open(output_filename, 'wb') as f:
                 f.write(response.content)
@@ -257,7 +261,9 @@ class TCGPlayerAPIService:
         response = self.requests_util.send_request(url, method='POST', cookies=self.cookies, json=payload)
         if response:
             # get filename from response headers
-            filename = response.headers.get('Content-Disposition').split('filename=')[1].strip('"')
+            header = response.headers.get('Content-Disposition', '')
+            match = re.search(r'filename="?([^";]+)"?', header)
+            filename = match.group(1) if match else f'packingslip{datetime.now().strftime("%Y%m%d_%H%M%S")}.pdf'
             output_filename = f'/app/tmp/{filename}'
             # save file to disk
             with open(output_filename, 'wb') as f: