Rearranging code: new class LedNameBadge for the methods header() and write() (and the USB decision and initialization code), which are the ones, that other content creation sw might want to use.

1 year ago · d79a15f119
parent 6b06508ef9
commit d79a15f119
1 changed files with 143 additions and 114 deletions
--- a/led-badge-11x44.py
+++ b/led-badge-11x44.py
@ -49,39 +49,22 @@
 # v0.11, 2019-09-29, jw New option --brightness added.
 # v0.12, 2019-12-27, jw hint at pip3 -- as discussed in https://github.com/jnweiger/led-name-badge-ls32/issues/19

-import sys, os, re, time, argparse
-from datetime import datetime
+import argparse
+import os
+import re
+import sys
+import time
 from array import array
+from datetime import datetime

 try:
-    if sys.version_info[0] < 3: raise Exception(
-        "prefer usb.core with python-2.x because of https://github.com/jnweiger/led-badge-ls32/issues/9")
    import pyhidapi
-
-    pyhidapi.hid_init()
-    have_pyhidapi = True
 except:
-    have_pyhidapi = False
-    try:
+    pass
+try:
    import usb.core
-    except:
-        print("ERROR: Need the pyhidapi or usb.core module.")
-        if sys.platform == "darwin":
-            print("""Please try
-  pip3 install pyhidapi
-  pip install pyhidapi
-  brew install hidapi""")
-        elif sys.platform == "linux":
-            print("""Please try
-  sudo pip3 install pyhidapi
-  sudo pip install pyhidapi
-  sudo apt-get install libhidapi-hidraw0
-  sudo ln -s /usr/lib/x86_64-linux-gnu/libhidapi-hidraw.so.0  /usr/local/lib/
-or
-  sudo apt-get install python3-usb""")
-        else:  # windows?
-            print("""Please with Linux or MacOS or help us implement support for """ + sys.platform)
-        sys.exit(1)
+except:
+    pass

 __version = "0.12"

@ -397,30 +380,72 @@ def bitmap(arg):
    return bitmap_text(arg)


-def expand_tuple(l):
+class LedNameBadge:
+    _protocol_header_template = (
+        0x77, 0x61, 0x6e, 0x67, 0x00, 0x00, 0x00, 0x00, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40,
+        0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
+        0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
+        0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00
+    )
+    _have_pyhidapi = False
+
+    @staticmethod
+    def init_class():
+        try:
+            if sys.version_info[0] < 3:
+                raise Exception("prefer usb.core with python-2.x because of https://github.com/jnweiger/led-badge-ls32/issues/9")
+            if pyhidapi:
+                pyhidapi.hid_init()
+                LedNameBadge._have_pyhidapi = True
+                print("Pyhidapi detected")
+            else:
+                raise Exception()
+        except Exception as e:
+            try:
+                if usb.core:
+                    print("Pyusb detected")
+                else:
+                    raise Exception()
+            except:
+                print("ERROR: Need the pyhidapi or usb.core module.")
+                if sys.platform == "darwin":
+                    print("""Please try
+  pip3 install pyhidapi
+  pip install pyhidapi
+  brew install hidapi
+""")
+                elif sys.platform == "linux":
+                    print("""Please try
+  sudo pip3 install pyhidapi
+  sudo pip install pyhidapi
+  sudo apt-get install libhidapi-hidraw0
+  sudo ln -s /usr/lib/x86_64-linux-gnu/libhidapi-hidraw.so.0  /usr/local/lib/
+or
+  sudo apt-get install python3-usb
+""")
+                else:  # windows?
+                    print("""Please with Linux or MacOS or help us implement support for """ + sys.platform)
+                sys.exit(1)
+
+    @staticmethod
+    def _expand_tuple(l):
        l = l + (l[-1],) * (8 - len(l))  # repeat last element
        return l

-def header(lengths, speeds, modes, blinks, ants, brightness=100, date=datetime.now()):
+    @staticmethod
+    def header(lengths, speeds, modes, blinks, ants, brightness=100, date=datetime.now()):
        """ lengths[0] is the number of chars of the first text

          Speeds come in as 1..8, but are needed 0..7 here.
        """
-    protocol_header_template = (
-        0x77, 0x61, 0x6e, 0x67, 0x00, 0x00, 0x00, 0x00, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40,
-        0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
-        0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
-        0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00
-    )
-
-    ants = expand_tuple(ants)
-    blinks = expand_tuple(blinks)
-    speeds = expand_tuple(speeds)
-    modes = expand_tuple(modes)
+        ants = LedNameBadge._expand_tuple(ants)
+        blinks = LedNameBadge._expand_tuple(blinks)
+        speeds = LedNameBadge._expand_tuple(speeds)
+        modes = LedNameBadge._expand_tuple(modes)

        speeds = [x - 1 for x in speeds]

-    h = list(protocol_header_template)
+        h = list(LedNameBadge._protocol_header_template)

        if brightness <= 25:
            h[5] = 0x40
@ -449,10 +474,49 @@ def header(lengths, speeds, modes, blinks, ants, brightness=100, date=datetime.n

        return h

+    @staticmethod
+    def write(buf):
+        if len(buf) > 8192:
+            print("Writing more than 8192 bytes damages the display!")
+            sys.exit(1)
+
+        if LedNameBadge._have_pyhidapi:
+            dev_info = pyhidapi.hid_enumerate(0x0416, 0x5020)
+            # dev = pyhidapi.hid_open(0x0416, 0x5020)
+            if dev_info:
+                dev = pyhidapi.hid_open_path(dev_info[0].path)
+                print("using [%s %s] int=%d page=%s via pyHIDAPI" % (
+                    dev_info[0].manufacturer_string, dev_info[0].product_string, dev_info[0].interface_number, dev_info[0].usage_page))
+            else:
+                print("No led tag with vendorID 0x0416 and productID 0x5020 found.")
+                print("Connect the led tag and run this tool as root.")
+                sys.exit(1)
+            pyhidapi.hid_write(dev, buf)
+            pyhidapi.hid_close(dev)
+        else:
+            dev = usb.core.find(idVendor=0x0416, idProduct=0x5020)
+            if dev is None:
+                print("No led tag with vendorID 0x0416 and productID 0x5020 found.")
+                print("Connect the led tag and run this tool as root.")
+                sys.exit(1)
+            try:
+                # win32: NotImplementedError: is_kernel_driver_active
+                if dev.is_kernel_driver_active(0):
+                    dev.detach_kernel_driver(0)
+            except:
+                pass
+            dev.set_configuration()
+            print("using [%s %s] bus=%d dev=%d" % (dev.manufacturer, dev.product, dev.bus, dev.address))
+            for i in range(int(len(buf) / 64)):
+                time.sleep(0.1)
+                dev.write(1, buf[i * 64:i * 64 + 64])

-def split_to_ints(list_str):
-    return [int(x) for x in re.split(r'[\s,]+', list_str)]

+LedNameBadge.init_class()
+
+
+def split_to_ints(list_str):
+    return tuple([int(x) for x in re.split(r'[\s,]+', list_str)])

 if __name__ == '__main__':
    parser = argparse.ArgumentParser(formatter_class=argparse.RawDescriptionHelpFormatter,
@ -519,15 +583,15 @@ if __name__ == '__main__':
    else:
        print("Type: 11x44")

-    lengths = [b[1] for b in msg_bitmaps]
-    speeds = split_to_ints(args.speeds)
+    lengths = tuple([b[1] for b in msg_bitmaps])
+    speeds = split_to_ints(args.speed)
    modes = split_to_ints(args.mode)
    blinks = split_to_ints(args.blink)
    ants = split_to_ints(args.ants)
    brightness = int(args.brightness)

    buf = array('B')
-    buf.extend(header(lengths, speeds, modes, blinks, ants, brightness))
+    buf.extend(LedNameBadge.header(lengths, speeds, modes, blinks, ants, brightness))

    for msg_bitmap in msg_bitmaps:
        buf.extend(msg_bitmap[0])
@ -536,39 +600,4 @@ if __name__ == '__main__':
    if need_padding:
        buf.extend((0,) * (64 - need_padding))

-    # print(buf)      # array('B', [119, 97, 110, 103, 0, 0, 0, 0, 48, 48, 48, 48, 48, 48, 48, 48, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 60, 126, 255, 255, 255, 255, 126, 60, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0])
-
-    if len(buf) > 8192:
-        print("Writing more than 8192 bytes damages the display!")
-        sys.exit(1)
-
-    if have_pyhidapi:
-        devinfo = pyhidapi.hid_enumerate(0x0416, 0x5020)
-        # dev = pyhidapi.hid_open(0x0416, 0x5020)
-        if devinfo:
-            dev = pyhidapi.hid_open_path(devinfo[0].path)
-            print("using [%s %s] int=%d page=%s via pyHIDAPI" % (
-            devinfo[0].manufacturer_string, devinfo[0].product_string, devinfo[0].interface_number, devinfo[0].usage_page))
-        else:
-            print("No led tag with vendorID 0x0416 and productID 0x5020 found.")
-            print("Connect the led tag and run this tool as root.")
-            sys.exit(1)
-        pyhidapi.hid_write(dev, buf)
-        pyhidapi.hid_close(dev)
-    else:
-        dev = usb.core.find(idVendor=0x0416, idProduct=0x5020)
-        if dev is None:
-            print("No led tag with vendorID 0x0416 and productID 0x5020 found.")
-            print("Connect the led tag and run this tool as root.")
-            sys.exit(1)
-        try:
-            # win32: NotImplementedError: is_kernel_driver_active
-            if dev.is_kernel_driver_active(0):
-                dev.detach_kernel_driver(0)
-        except:
-            pass
-        dev.set_configuration()
-        print("using [%s %s] bus=%d dev=%d" % (dev.manufacturer, dev.product, dev.bus, dev.address))
-        for i in range(int(len(buf) / 64)):
-            time.sleep(0.1)
-            dev.write(1, buf[i * 64:i * 64 + 64])
+    LedNameBadge.write(buf)