app.py

@@ -4,7 +4,7 @@ import threading
 import queue
 import pdf2image
 from brother_ql.conversion import convert
-from brother_ql.backends.helpers import send, status
+from brother_ql.backends.helpers import send
 from brother_ql.raster import BrotherQLRaster
 from PIL import Image
 import mimetypes
@@ -16,19 +16,12 @@ app = Flask(__name__)
 
 # Set up logging
 log_file = 'log.txt'
-
-# Create a rotating file handler for logging
 log_handler = RotatingFileHandler(log_file, maxBytes=10*1024*1024, backupCount=5)  # 10MB per log file, keep 5 backups
-log_handler.setLevel(logging.DEBUG)
+log_handler.setLevel(logging.INFO)
 log_formatter = logging.Formatter('%(asctime)s - %(levelname)s - %(message)s')
 log_handler.setFormatter(log_formatter)
-
-# Add the log handler to the Flask app's logger
 app.logger.addHandler(log_handler)
 
-# Ensure Flask app uses our logger, not its default one
-app.logger.setLevel(logging.DEBUG)
-
 # Create a queue to hold the files for processing
 file_queue = queue.Queue()
 
@@ -41,6 +34,9 @@ printer_model = "QL-1100"
 backend = 'pyusb'
 printer = 'usb://0x04f9:0x20a7'
 
+# Create a lock to manage printer access
+printer_lock = threading.Lock()
+
 # Convert PDF to image
 def convert_pdf_to_image(pdf_path):
     """Converts a PDF to a PIL Image"""
@@ -54,6 +50,19 @@ def convert_pdf_to_image(pdf_path):
         app.logger.error(f"Error converting PDF: {str(e)}")
         return None
 
+# Reinitialize the printer by disconnecting and reconnecting
+def reinitialize_printer():
+    try:
+        app.logger.info("Reinitializing printer...")
+        usb_device = usb.core.find(idVendor=0x04f9, idProduct=0x20a7)
+        if usb_device:
+            usb_device.reset()
+            app.logger.info("Printer reinitialized successfully.")
+        else:
+            app.logger.error("Failed to find printer device.")
+    except Exception as e:
+        app.logger.error(f"Error reinitializing printer: {str(e)}")
+
 # Function to print address labels
 def print_address_label(pdf_path):
     try:
@@ -62,13 +71,6 @@ def print_address_label(pdf_path):
         if not image:
             raise Exception("Failed to create label images")
     
-        if 'address' in pdf_path.lower():
-            target_width = 1660
-            target_height = 1164
-            image = image.convert("RGB")
-            image = image.resize((target_width, target_height), Image.LANCZOS)
-            image = image.rotate(90, expand=True)
-        else:
         target_width = 1164
         target_height = 1660
         image = image.convert("RGB")
@@ -91,18 +93,40 @@ def print_address_label(pdf_path):
             cut=True
         )
 
-        app.logger.info("Sending to printer...")
-        status = send(
+        # Try to acquire the printer lock with a longer timeout
+        max_retries = 5  # Retry up to 5 times
+        retry_delay = 5  # Retry every 5 seconds
+        for attempt in range(max_retries):
+            if printer_lock.acquire(timeout=30):  # 30 seconds timeout to allow more time for the printer
+                try:
+                    app.logger.info(f"Attempting to send print job (Attempt {attempt + 1})...")
+                    send(
                         instructions=instructions, 
                         printer_identifier=printer, 
                         backend_identifier=backend, 
-            blocking=True
+                        blocking=True  # Ensure blocking, so we wait for the print to complete
                     )
-        app.logger.info(f"Printer status: {status}")
                     app.logger.info("Print job sent successfully")
-        
+                    break
                 except Exception as e:
                     app.logger.error(f"Error during printing: {str(e)}")
+                    if attempt < max_retries - 1:
+                        app.logger.info(f"Retrying after {retry_delay} seconds...")
+                        time.sleep(retry_delay)
+                    else:
+                        app.logger.error("Max retries reached. Failed to send print job.")
+                        reinitialize_printer()  # Reinitialize printer after retries
+            else:
+                app.logger.error(f"Failed to acquire printer lock (Attempt {attempt + 1}). Printer is busy.")
+                if attempt < max_retries - 1:
+                    app.logger.info(f"Retrying after {retry_delay} seconds...")
+                    time.sleep(retry_delay)
+                else:
+                    app.logger.error("Printer is still busy after retries.")
+                    raise Exception("Printer is busy. Try again later.")
+    
+    except Exception as e:
+        app.logger.error(f"Error during label printing: {str(e)}")
 
 # Worker thread that processes files from the queue
 def process_queue():
@@ -114,6 +138,7 @@ def process_queue():
         # Process the file (convert and print)
         app.logger.info(f"Processing file: {file_path}")
         print_address_label(file_path)
+        time.sleep(10)
         # Signal that the task is done
         file_queue.task_done()