81 lines
1.9 KiB
Python
81 lines
1.9 KiB
Python
import re
|
|
|
|
import click
|
|
from PIL import Image
|
|
|
|
from niimprint import BluetoothTransport, PrinterClient, SerialTransport
|
|
|
|
|
|
@click.command("print")
|
|
@click.option(
|
|
"-m",
|
|
"--model",
|
|
type=click.Choice(["b1", "b21", "d11"], False),
|
|
default="b21",
|
|
show_default=True,
|
|
help="Niimbot printer model",
|
|
)
|
|
@click.option(
|
|
"-c",
|
|
"--conn",
|
|
type=click.Choice(["usb", "bluetooth"]),
|
|
default="usb",
|
|
show_default=True,
|
|
help="Connection type",
|
|
)
|
|
@click.option(
|
|
"-a",
|
|
"--addr",
|
|
help="Bluetooth MAC address OR serial device path",
|
|
)
|
|
@click.option(
|
|
"-d",
|
|
"--density",
|
|
type=click.IntRange(1, 5),
|
|
default=5,
|
|
show_default=True,
|
|
help="Print density",
|
|
)
|
|
@click.option(
|
|
"-r",
|
|
"--rotate",
|
|
type=click.Choice(["0", "90", "180", "270"]),
|
|
default="0",
|
|
show_default=True,
|
|
help="Image rotation (clockwise)",
|
|
)
|
|
@click.option(
|
|
"-i",
|
|
"--image",
|
|
type=click.Path(exists=True),
|
|
required=True,
|
|
help="Image path",
|
|
)
|
|
def print_cmd(model, conn, addr, density, rotate, image):
|
|
if conn == "bluetooth":
|
|
assert conn is not None, "--addr argument required for bluetooth connection"
|
|
addr = addr.upper()
|
|
assert re.fullmatch(r"([0-9A-F]{2}:){5}([0-9A-F]{2})", addr), "Bad MAC address"
|
|
transport = BluetoothTransport(addr)
|
|
if conn == "usb":
|
|
port = addr if addr is not None else "auto"
|
|
transport = SerialTransport(port=port)
|
|
|
|
if model in ("b1", "b21"):
|
|
max_width_px = 384
|
|
if model == "d11":
|
|
max_width_px = 96
|
|
|
|
image = Image.open(image)
|
|
if rotate != "0":
|
|
# PIL library rotates counter clockwise, so we need to multiply by -1
|
|
image = image.rotate(-int(rotate), expand=True)
|
|
assert image.width <= max_width_px, f"Image width too big for {model.upper()}"
|
|
|
|
printer = PrinterClient(transport)
|
|
printer.print_image(image, density=density)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
print_cmd()
|