From ea894f917253b7f708e1ef7525a88d174497a08e Mon Sep 17 00:00:00 2001
From: Infinidoge <infinidoge@inx.moe>
Date: Sat, 7 Dec 2024 11:27:30 -0500
Subject: [PATCH] add printer library

---
 printer.py | 92 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 92 insertions(+)
 create mode 100644 printer.py

diff --git a/printer.py b/printer.py
new file mode 100644
index 0000000..10c4274
--- /dev/null
+++ b/printer.py
@@ -0,0 +1,92 @@
+from escpos.printer import CupsPrinter
+from itertools import zip_longest
+
+
+class CallbackContext:
+    def __init__(self, enter=None, exit=None):
+        self.enter = enter
+        self.exit = exit
+
+    def __enter__(self):
+        if self.enter:
+            return self.enter()
+
+    def __exit__(self, *_):
+        if self.exit:
+            self.exit()
+
+
+class Split2Context:
+    def __init__(
+        self, printer, num_cols=2, widths=None, align="<", aligns=None, columns=48
+    ):
+        self.printer = printer
+
+        self.num_cols = num_cols or len(widths)
+        self.widths = widths or [columns // self.num_cols] * self.num_cols
+
+        self.aligns = aligns or [align] * self.num_cols
+
+        self.col_length = columns // 2
+
+        self.columns = [[] for i in range(self.num_cols)]
+
+    def col(self, num, text):
+        self.columns[num].append(f"{text:{self.aligns[num]}{self.widths[num]}}")
+
+    def left(self, text):
+        self.col(0, text)
+
+    def right(self, text):
+        self.col(-1, text)
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, *_):
+        lines = map("".join, zip_longest(*self.columns, fillvalue=" " * self.num_cols))
+        for line in lines:
+            self.printer.textln(line)
+
+
+class ReceiptPrinter(CupsPrinter):
+    def __init__(self, *args, autocut=True, **kwargs):
+        self.autocut = autocut
+        super().__init__(*args, **kwargs)
+
+    def __enter__(self):
+        self.open()
+        return self
+
+    def __exit__(self, _type, _value, _traceback):
+        if self.autocut:
+            self.cut()
+        self.close()
+
+    def set(self, *args, **kwargs):
+        super().set(*args, **kwargs)
+        return CallbackContext(exit=super().set_with_default)
+
+    def set_with_default(self, *args, **kwargs):
+        super().set_with_default(*args, **kwargs)
+        return CallbackContext(exit=super().set_with_default)
+
+    def split(self, left, right, cols=48):
+        self.textln(f"{left:<{cols//2}}{right:<{cols//2}}")
+
+    def vsplit(self, *args, **kwargs):
+        return Split2Context(self, *args, **kwargs)
+
+    def invert(self, text):
+        with self.set(invert=True):
+            self.text(text)
+
+    def bold(self, text):
+        with self.set(bold=True):
+            self.text(text)
+
+    def boldln(self, text):
+        self.bold(text)
+        self.ln()
+
+p = ReceiptPrinter("EPSON-TM-m30", profile="default")