print.py

import argparse
import asyncio
import logging
import sys
import os
import numpy as np

from pricetag_printer.ble import run_ble
from pricetag_printer.img import read_img


EPD_WIDTH = 250
EPD_HEIGHT = 128



def parse_args():
    args = argparse.ArgumentParser(
        description='Sends an image to the EPD BLE service running in the ATC_TLSR_Paper Hanshow open source firmware.')
    args.add_argument('filename', type=str)
    args.add_argument('--log-level', type=str,
                      choices=['debug', 'info', 'warn', 'error'], default='info')
    args.add_argument('--img-binarization-algo', type=str,
                      choices=['mean-threshold', 'floyd-steinberg'], default='floyd-steinberg',
                      help='Which image binarization algorithm to use.')
    args.add_argument('--show-preview', action='store_true',
                      help='If set, displays the final image and asks the user for confirmation before printing.')
    args.add_argument('--devicename', type=str, default='GT01',
                      help='Specify the Bluetooth device name to search for. Default value is GT01.')
    args.add_argument('--img-invert', action='store_true',
                      help='Invert resulting pixels.')
    return args.parse_args()


def make_logger(log_level):
    logger = logging.getLogger('pricetag-printer')
    logger.setLevel(log_level)
    h = logging.StreamHandler(sys.stdout)
    h.setLevel(log_level)
    logger.addHandler(h)
    return logger


def main():
    args = parse_args()

    log_level = getattr(logging, args.log_level.upper())
    logger = make_logger(log_level)

    filename = args.filename
    if not os.path.exists(filename):
        logger.info('🛑 File not found. Exiting.')
        return

    bin_img = read_img(args.filename, EPD_HEIGHT,EPD_WIDTH,
                       logger, args.img_binarization_algo, args.show_preview, args.img_invert)
    if bin_img is None:
        logger.info(f'🛑 No image generated. Exiting.')
        return

    logger.info(f'✅ Read image: {bin_img.shape} (h, w) pixels')

    data = np.packbits(bin_img)

    loop = asyncio.get_event_loop()
    loop.run_until_complete(run_ble(data, args.devicename, logger))


if __name__ == '__main__':
    main()