add more subcommands

receipt.py

@@ -1,28 +1,29 @@
 #!/usr/bin/env python3
 
-from escpos import *
+from escpos import printer
 from PIL import Image
-import os 
+import os
 import subprocess
-import time
 import random
 import requests
-from urllib.parse import quote
+
 
 # Return T20II printer
 def get_T20II_usb():
-    p = printer.Usb(0x04b8, 0x0202) #, profile="TM-T20II")
+    p = printer.Usb(0x04B8, 0x0202)  # , profile="TM-T20II")
     # p.set_with_default(align='center', font='a', bold=False, width=2, height=2, custom_size=True, smooth=True)
     return p
 
+
 def get_T20II_ethernet():
-    p = printer.Network("192.168.192.168", port=9100)#, profile="TM-T20II")
+    p = printer.Network("192.168.192.168", port=9100)  # , profile="TM-T20II")
     # p.set_with_default(align='center', font='a', bold=False, width=2, height=2, custom_size=True, smooth=True)
     return p
 
+
 def resize_image_to_fullwidth(img_src_name, target_width=576):
     # Make fullwidth dir
-    os.makedirs('img_fullwidth', exist_ok=True)
+    os.makedirs("img_fullwidth", exist_ok=True)
 
     # Open the image
     img = Image.open(img_src_name)
@@ -38,7 +39,7 @@ def resize_image_to_fullwidth(img_src_name, target_width=576):
     resized_img = img.resize((target_width, new_height), Image.LANCZOS)
 
     # Save the resized image
-    name, ext = os.path.splitext(img_src_name)
+    _, ext = os.path.splitext(img_src_name)
     filename_no_ext = os.path.splitext(os.path.basename(img_src_name))[0]
 
     # Build destination filename
@@ -49,62 +50,113 @@ def resize_image_to_fullwidth(img_src_name, target_width=576):
 
 def get_random_bant_image():
     # Create img directory if it doesn't exist
-    os.makedirs('img', exist_ok=True)
+    os.makedirs("img", exist_ok=True)
-    
+
     # Fetch the catalog JSON for /bant/
     url = "https://a.4cdn.org/bant/catalog.json"
     response = requests.get(url)
     response.raise_for_status()
     catalog = response.json()
-    
+
     # Flatten the list of threads from all pages
-    threads = [thread for page in catalog for thread in page['threads']]
+    threads = [thread for page in catalog for thread in page["threads"]]
-    
+
     # Filter threads that have images (tim and ext fields)
     image_threads = [
-        thread for thread in threads
+        thread
-        if 'tim' in thread and 'ext' in thread and thread['ext'].lower() not in ['.gif', '.mp4', '.webm']
+        for thread in threads
+        if "tim" in thread
+        and "ext" in thread
+        and thread["ext"].lower() not in [".gif", ".mp4", ".webm"]
     ]
-    
+
     if not image_threads:
         return None  # No images found
-    
+
     # Choose a random thread with an image
     thread = random.choice(image_threads)
-    
+
     # Construct image URL
     image_url = f"https://i.4cdn.org/bant/{thread['tim']}{thread['ext']}"
     filename = f"img/{thread['tim']}{thread['ext']}"
-    
+
     # Download and save image
     img_response = requests.get(image_url)
     img_response.raise_for_status()
-    
+
-    with open(filename, 'wb') as f:
+    with open(filename, "wb") as f:
         f.write(img_response.content)
-    
+
     return filename
 
+
 def print_random_bant_image(p, sh_file):
     img = get_random_bant_image()
+    if img is None:
+        print("No images found on /bant/.")
+        return
     filename_no_ext = os.path.splitext(os.path.basename(img))[0]
     p.textln(filename_no_ext)
     fimg = resize_image_to_fullwidth(img)
     try:
-        p.image(fimg, impl='bitImageColumn');
+        p.image(fimg, impl="bitImageColumn")
-    except:
+    except Exception as e:
+        print(f"Print failed, reconnecting: {e}")
         subprocess.run(["bash", sh_file], check=True)
         p = get_T20II_ethernet()
     p.cut()
 
+
+def print_note(p, text):
+    p.textln(text)
+    p.cut()
+
+
+def print_local_image(p, sh_file, img_path, message=None):
+    if not os.path.isfile(img_path):
+        print(f"File not found: {img_path}")
+        return
+    if message:
+        p.textln(message)
+    fimg = resize_image_to_fullwidth(img_path)
+    try:
+        p.image(fimg, impl="bitImageColumn")
+    except Exception as e:
+        print(f"Print failed, reconnecting: {e}")
+        subprocess.run(["bash", sh_file], check=True)
+        p = get_T20II_ethernet()
+    p.cut()
+
+
 def main():
+    import argparse
+
+    parser = argparse.ArgumentParser(description="Receipt printer CLI")
+    subparsers = parser.add_subparsers(dest="command", required=True)
+
+    subparsers.add_parser("image", help="Print a random /bant/ image")
+
+    note_p = subparsers.add_parser("note", help="Print a text note")
+    note_p.add_argument("text", help="The note to print")
+
+    print_p = subparsers.add_parser("print", help="Print a local image")
+    print_p.add_argument("path", help="Path to the image file")
+    print_p.add_argument("message", nargs="?", default=None, help="Optional caption")
+
+    args = parser.parse_args()
+
     script_dir = os.path.dirname(os.path.abspath(__file__))
     sh_file = os.path.join(script_dir, "connect_recipt.sh")
     subprocess.run(["bash", sh_file], check=True)
     p = get_T20II_ethernet()
 
-    print_random_bant_image(p, sh_file)
+    if args.command == "image":
-    return
+        print_random_bant_image(p, sh_file)
+    elif args.command == "note":
+        print_note(p, args.text)
+    elif args.command == "print":
+        print_local_image(p, sh_file, args.path, args.message)
+
 
 if __name__ == "__main__":
     main()