zman27/tcg-print-server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

f

Browse Source
This commit is contained in:
zman 2025-04-08 13:25:30 -04:00
parent 72d19bd8b0
commit d00ceef282
1 changed files with 44 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

66
app.py
View File

@ -32,9 +32,15 @@ app.logger.setLevel(logging.DEBUG)
# Create a queue to hold the files for processing # Create a queue to hold the files for processing
file_queue = queue.Queue() file_queue = queue.Queue()
image_queue = queue.Queue()
lock = threading.Lock()
# Directory to save uploaded files (you can modify this path) # Directory to save uploaded files (you can modify this path)
UPLOAD_FOLDER = './uploads' UPLOAD_FOLDER = './uploads'
IMAGE_FOLDER = './images'
os.makedirs(UPLOAD_FOLDER, exist_ok=True) os.makedirs(UPLOAD_FOLDER, exist_ok=True)
os.makedirs(IMAGE_FOLDER, exist_ok=True)
# Printer configuration # Printer configuration
printer_model = "QL-1100" printer_model = "QL-1100"
@ -45,7 +51,6 @@ printer = 'usb://0x04f9:0x20a7'
def convert_pdf_to_images(pdf_path): def convert_pdf_to_images(pdf_path):
"""Converts a PDF to a PIL Image""" """Converts a PDF to a PIL Image"""
try: try:
# Convert PDF to image (get the first page)
images = pdf2image.convert_from_path(pdf_path) images = pdf2image.convert_from_path(pdf_path)
if images: if images:
return images return images
@ -55,23 +60,19 @@ def convert_pdf_to_images(pdf_path):
return None return None
# Function to print address labels # Function to print address labels
def print_address_label(images, pdf_path): def print_address_label(image, pdf_path):
try: try:
processed_images = [] if 'address' in pdf_path.lower():
for image in images: target_width = 1660
if 'address' in pdf_path.lower(): target_height = 1164
target_width = 1660 image = image.convert("RGB")
target_height = 1164 image = image.resize((target_width, target_height), Image.LANCZOS)
image = image.convert("RGB") image = image.rotate(90, expand=True)
image = image.resize((target_width, target_height), Image.LANCZOS) else:
image = image.rotate(90, expand=True) target_width = 1164
processed_images.append(image) target_height = 1660
else: image = image.convert("RGB")
target_width = 1164 image = image.resize((target_width, target_height), Image.LANCZOS)
target_height = 1660
image = image.convert("RGB")
image = image.resize((target_width, target_height), Image.LANCZOS)
processed_images.append(image)
qlr = BrotherQLRaster(printer_model) qlr = BrotherQLRaster(printer_model)
qlr.exception_on_warning = True qlr.exception_on_warning = True
@ -79,7 +80,7 @@ def print_address_label(images, pdf_path):
app.logger.info("Converting image to printer instructions...") app.logger.info("Converting image to printer instructions...")
instructions = convert( instructions = convert(
qlr=qlr, qlr=qlr,
images=processed_images, images=[image],
label='102x152', label='102x152',
threshold=70.0, threshold=70.0,
dither=False, dither=False,
@ -103,8 +104,19 @@ def print_address_label(images, pdf_path):
except Exception as e: except Exception as e:
app.logger.error(f"Error during printing: {str(e)}") app.logger.error(f"Error during printing: {str(e)}")
def save_images(images, pdf_path):
uuid = str(int(time.time()))
for i,image in enumerate(images):
if 'address' in pdf_path.lower():
image.save(os.path.join(IMAGE_FOLDER, f"address_label_{uuid}_{i}.png"), "PNG")
elif 'packing' in pdf_path.lower():
image.save(os.path.join(IMAGE_FOLDER, f"shipping_label_{uuid}_{i}.png"), "PNG")
else:
image.save(os.path.join(IMAGE_FOLDER, f"unknown_{uuid}_{i}.png"), "PNG")
image_queue.put((image, pdf_path))
# Worker thread that processes files from the queue # Worker thread that processes files from the queue
def process_queue(): def file_worker():
while True: while True:
# Wait until a file is available in the queue # Wait until a file is available in the queue
file_path = file_queue.get() file_path = file_queue.get()
@ -114,15 +126,25 @@ def process_queue():
app.logger.info(f"Processing file: {file_path}") app.logger.info(f"Processing file: {file_path}")
images = convert_pdf_to_images(file_path) images = convert_pdf_to_images(file_path)
if images: if images:
print_address_label(images, file_path) save_images(images, file_path)
else: else:
app.logger.error(f"Failed to convert PDF to images: {file_path}") app.logger.error(f"Failed to convert PDF to images: {file_path}")
# Signal that the task is done # Signal that the task is done
file_queue.task_done() file_queue.task_done()
def printer_worker():
while True:
image, path = image_queue.get()
if image is None:
break
with lock:
print_address_label(image, path)
image_queue.task_done()
# Start the worker thread # Start the worker thread
thread = threading.Thread(target=process_queue, daemon=True) threading.Thread(target=file_worker, daemon=True).start()
thread.start() # Start the printer worker thread
threading.Thread(target=printer_worker, daemon=True).start()
@app.route('/upload', methods=['POST']) @app.route('/upload', methods=['POST'])
def upload_file(): def upload_file():