sdwire 0.2.2__py3-none-any.whl → 0.2.4__py3-none-any.whl

sdwire/backend/detect.py

@@ -1,13 +1,12 @@
 import logging
 from typing import List
-from adafruit_board_toolkit import circuitpython_serial as cpserial
-from serial.tools.list_ports_common import ListPortInfo
 
 from sdwire import constants
 from .device.sdwire import SDWire, SDWIRE_GENERATION_SDWIRE3
 from .device.sdwirec import SDWireC
 from .device.usb_device import PortInfo
 
+import pyudev
 import usb.core
 import usb.util
 from usb.core import Device
@@ -46,30 +45,31 @@ def get_sdwirec_devices() -> List[SDWireC]:
 
 
 def get_sdwire_devices() -> List[SDWire]:
-
     # Badgerd SDWire3
     # VID = 0bda PID = 0316
     # Badgerd SDWireC
     # VID = 0x04e8 PID = 0x6001
     result = []
-    devices: List[Device] = usb.core.find(find_all=True)
+    devices: List[Device] = pyudev.Context().list_devices(
+        subsystem="usb",
+        ID_VENDOR_ID=f"{constants.SDWIRE3_VID:04x}",
+        ID_MODEL_ID=f"{constants.SDWIRE3_PID:04x}",
+    )
     if not devices:
         log.info("no usb devices found while searching for SDWire..")
         return []
 
-    device_list = []
     for device in devices:
         product = None
         serial = None
-        manufacturer = None
+        bus = None
+        address = None
         try:
-            product = device.idProduct
-            vendor = device.idVendor
-            serial = (
-                usb.util.get_string(device, device.iSerialNumber, None)
-                + f"{device.bus}{device.port_number}"
-            )
-            manufacturer = device.manufacturer
+            product = int(f"0x{device.get('ID_MODEL_ID')}", 16)
+            vendor = int(f"0x{device.get('ID_VENDOR_ID')}", 16)
+            bus = int(device.get("BUSNUM"))
+            address = int(device.get("DEVNUM"))
+            serial = f"{device.get('ID_USB_SERIAL_SHORT')}:{bus}.{address}"
         except Exception as e:
             log.debug(
                 "not able to get usb product, serial_number and manufacturer information, err: %s",
@@ -77,9 +77,12 @@ def get_sdwire_devices() -> List[SDWire]:
             )
 
         if product == constants.SDWIRE3_PID and vendor == constants.SDWIRE3_VID:
+            usb_device: List[Device] = usb.core.find(
+                idVendor=vendor, idProduct=product, bus=bus, address=address
+            )
             result.append(
                 SDWire(
-                    port_info=PortInfo(device, product, vendor, serial, device),
+                    port_info=PortInfo(device, product, vendor, serial, usb_device),
                     generation=SDWIRE_GENERATION_SDWIRE3,
                 )
             )

sdwire/backend/device/sdwire.py

@@ -1,5 +1,4 @@
 import logging
-from serial import Serial
 from .usb_device import USBDevice, PortInfo
 
 log = logging.getLogger(__name__)
@@ -8,10 +7,18 @@ SDWIRE_GENERATION_SDWIRE3 = 2
 
 
 class SDWire(USBDevice):
+    __block_dev = None
 
     def __init__(self, port_info: PortInfo, generation: int):
         super().__init__(port_info)
         self.generation = generation
+        for sibling in self.dev_string.parent.children:
+            if (
+                self.dev_string.device_path != sibling.device_path
+                and sibling.device_type == "disk"
+            ):
+                self.__block_dev = f"/dev/{sibling.device_path.split('/')[-1]}"
+                break
 
     def switch_ts(self):
         try:
@@ -33,8 +40,12 @@ class SDWire(USBDevice):
                 e,
             )
 
+    @property
+    def block_dev(self):
+        return self.__block_dev
+
     def __str__(self):
-        return f"{self.serial_string}\t[{int(self.manufacturer_string):04x}::{int(self.product_string):04x}]"
+        return f"{self.serial_string}\t[{int(self.manufacturer_string):04x}::{int(self.product_string):04x}]\t\t{self.block_dev}"
 
     def __repr__(self):
         return self.__str__()

sdwire/backend/device/sdwirec.py

@@ -1,24 +1,37 @@
 import logging
 from pyftdi.ftdi import Ftdi
-from serial import Serial
 from .usb_device import USBDevice, PortInfo
 
 log = logging.getLogger(__name__)
 
 
 class SDWireC(USBDevice):
+    __block_dev = None
 
     def __init__(self, port_info: PortInfo):
         super().__init__(port_info)
+        for d in self._pyudev_context.list_devices(ID_MODEL="sd-wire"):
+            d_serial = d.get("ID_USB_SERIAL_SHORT", None)
+            if d_serial is not None and d_serial == self.serial_string:
+                for sibling in d.parent.children:
+                    if (
+                        d.device_path != sibling.device_path
+                        and sibling.device_type == "disk"
+                    ):
+                        self.__block_dev = f"/dev/{sibling.device_path.split('/')[-1]}"
+                        break
+                break
 
     def __str__(self):
-        return (
-            f"{self.serial_string}\t[{self.product_string}::{self.manufacturer_string}]"
-        )
+        return f"{self.serial_string}\t[{self.product_string}::{self.manufacturer_string}]\t{self.block_dev}"
 
     def __repr__(self):
         return self.__str__()
 
+    @property
+    def block_dev(self):
+        return self.__block_dev
+
     def switch_ts(self):
         self._set_sdwire(1)
 

sdwire/backend/device/usb_device.py

@@ -1,5 +1,5 @@
 from collections import namedtuple
-from serial.tools.list_ports_common import ListPortInfo
+import pyudev
 
 PortInfo = namedtuple(
     "PortInfo", ("device", "product", "manufacturer", "serial", "usb_device")
@@ -8,9 +8,11 @@ PortInfo = namedtuple(
 
 class USBDevice:
     __port_info = None
+    _pyudev_context = None
 
     def __init__(self, port_info: PortInfo):
         self.__port_info = port_info
+        self._pyudev_context = pyudev.Context()
 
     @property
     def usb_device(self):

sdwire/backend/utils.py

@@ -20,9 +20,11 @@ def handle_switch_target_command(ctx):
 
 def handle_switch_off_command(ctx):
     device = ctx.obj["device"]
-    if isinstance(device, SDWireC):
-        log.info("SDWireC or legacy sdwire devices dont have off functionality")
-        print("SDWireC dont have off functionality")
+    if isinstance(device, SDWireC) or isinstance(device, SDWire):
+        log.info(
+            "SDWire3, SDWireC or legacy sdwire devices dont have off functionality"
+        )
+        print("SDWireC and SDWire3 dont have off functionality implemented")
         sys.exit(1)
 
 

sdwire/main.py

@@ -1,19 +1,20 @@
 #!/usr/bin/env python
+import logging
 import click
 from .backend import utils
 from .backend import detect
-from .backend.device.sdwire import SDWire
 
 
 @click.group()
-def main():
-    pass
+@click.option("--debug", required=False, is_flag=True, help="Enable debug output")
+def main(debug=None):
+    if debug:
+        logging.basicConfig(level=logging.DEBUG)
 
 
 @main.command()
 def list():
-
-    print(f"Serial\t\t\tProduct Info")
+    print("Serial\t\t\tProduct Info\t\tBlock Dev")
     for sdwire in detect.get_sdwire_devices():
         print(sdwire)
 
@@ -88,10 +89,5 @@ def off(ctx: click.Context):
     utils.handle_switch_off_command(ctx)
 
 
-# investigate further to use click in circuitpython
-# def invoke():
-#     main.main("switch ts".split(), "sdwire", "_SDWIRE_COMPLETE")
-
-
 if __name__ == "__main__":
     main()

{sdwire-0.2.2.dist-info → sdwire-0.2.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sdwire
-Version: 0.2.2
+Version: 0.2.4
 Summary: CLI application to interact with Badgerd SDWire devices
 License: GPL-3
 Author: Talha Can Havadar
@@ -9,11 +9,10 @@ Requires-Python: >=3.12,<4.0
 Classifier: License :: Other/Proprietary License
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.12
-Requires-Dist: adafruit-board-toolkit (>=1.1.1,<2.0.0)
 Requires-Dist: click (>=8.1.7,<9.0.0)
-Requires-Dist: pyftdi (>=0.55.4,<0.56.0)
+Requires-Dist: pyftdi (>=0.56.0,<0.57.0)
+Requires-Dist: pyudev (>=0.24.3,<0.25.0)
 Requires-Dist: pyusb (>=1.2.1,<2.0.0)
-Requires-Dist: semver (>=3.0.2,<4.0.0)
 Description-Content-Type: text/markdown
 
 # CLI for Badgerd SDWire Devices

sdwire-0.2.4.dist-info/RECORD

@@ -0,0 +1,14 @@
+sdwire/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+sdwire/backend/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+sdwire/backend/detect.py,sha256=jiQ2JS9sMGUuymwyUO4FaGzIkXK3J7Ci45V1X8c94b4,2931
+sdwire/backend/device/sdwire.py,sha256=5Glk4M20sY7EJA5XDewT1F5F1-AUW1fG7mI9_wrABoM,1518
+sdwire/backend/device/sdwirec.py,sha256=imkj2ObrgowiM7DP9TnVbCES-8bju0T69Qzdw5OURgE,1657
+sdwire/backend/device/usb_device.py,sha256=5sdyOyPikIpFydKywzDkvvgLHFdMcsBC8tKosHRhfvQ,808
+sdwire/backend/utils.py,sha256=kMEyBbRO91REgvgICuBONGiImO6jTr6EpqkZaxQH4aU,1507
+sdwire/constants.py,sha256=s5mCx5NlJUM3aBbCPq2y3oOFoPPJ5wA3BR1snyK-cMg,119
+sdwire/main.py,sha256=G1WaVd4kBR3bTfbmHekjx1yObiz4f1NLGqznFepd2NY,2132
+sdwire-0.2.4.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+sdwire-0.2.4.dist-info/METADATA,sha256=ZtSg0QMJGsQF52Aa_GlRkwbjn6mxK0EAiv15r_ITW-E,3104
+sdwire-0.2.4.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+sdwire-0.2.4.dist-info/entry_points.txt,sha256=pxy0zJKVcNWXPk5PtNjTLExOBpqFNth37wtdYdYRXCE,43
+sdwire-0.2.4.dist-info/RECORD,,

sdwire-0.2.2.dist-info/RECORD

@@ -1,14 +0,0 @@
-sdwire/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-sdwire/backend/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-sdwire/backend/detect.py,sha256=Ze_nmepeuMIiXRmiDBdbAkH4qmnyNd4pylurlWFcSZM,2752
-sdwire/backend/device/sdwire.py,sha256=8chgdYm5D83GqJy-v1iHomE_ZTXk6-N9o0Yyw-X9yIY,1122
-sdwire/backend/device/sdwirec.py,sha256=BZXvCq3bmJzzA5seCVOYIdmr6FBeAreE95_n-2kiuMQ,1035
-sdwire/backend/device/usb_device.py,sha256=6Mt24B22yTD4RdY24LgnRaVtbX60EPE2SAK-pN0ZESc,775
-sdwire/backend/utils.py,sha256=D_eO9LiMCWM9UPPumXfmqY30UxsHn-PvpURKJYFdmEY,1422
-sdwire/constants.py,sha256=s5mCx5NlJUM3aBbCPq2y3oOFoPPJ5wA3BR1snyK-cMg,119
-sdwire/main.py,sha256=ndbRoCkSckItLDH0q3zacC0mqfa0ZL-TJGDkIGJKrBU,2138
-sdwire-0.2.2.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-sdwire-0.2.2.dist-info/METADATA,sha256=POgYay_dxKiDXKHo984s_Xi9u4DdRsOaJK9Igk0PzOo,3157
-sdwire-0.2.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-sdwire-0.2.2.dist-info/entry_points.txt,sha256=pxy0zJKVcNWXPk5PtNjTLExOBpqFNth37wtdYdYRXCE,43
-sdwire-0.2.2.dist-info/RECORD,,