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

sdwire/backend/detect.py

@@ -8,6 +8,7 @@ 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
@@ -52,7 +53,11 @@ def get_sdwire_devices() -> List[SDWire]:
     # 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 []
@@ -62,14 +67,15 @@ def get_sdwire_devices() -> List[SDWire]:
         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}"
+            manufacturer = ""
         except Exception as e:
             log.debug(
                 "not able to get usb product, serial_number and manufacturer information, err: %s",
@@ -77,9 +83,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

@@ -8,10 +8,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 +41,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

@@ -7,18 +7,32 @@ 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,4 +1,5 @@
 from collections import namedtuple
+import pyudev
 from serial.tools.list_ports_common import ListPortInfo
 
 PortInfo = namedtuple(
@@ -8,9 +9,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/main.py

@@ -1,4 +1,5 @@
 #!/usr/bin/env python
+import logging
 import click
 from .backend import utils
 from .backend import detect
@@ -6,14 +7,15 @@ 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(f"Serial\t\t\tProduct Info\t\tBlock Dev")
     for sdwire in detect.get_sdwire_devices():
         print(sdwire)
 

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sdwire
-Version: 0.2.2
+Version: 0.2.3
 Summary: CLI application to interact with Badgerd SDWire devices
 License: GPL-3
 Author: Talha Can Havadar
@@ -12,6 +12,7 @@ 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: 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

sdwire-0.2.3.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=5lvoMjgBSAp9egNP7HwsdVAAEoA92_tw7YqcYtNKiSo,3135
+sdwire/backend/device/sdwire.py,sha256=LevVtz-RkvuJLAxSVok9ftpfIw-HrFSckltg6ur6TIU,1544
+sdwire/backend/device/sdwirec.py,sha256=62g8ujRXRj7PEJZPNXAUWiVUOqDcyQ8bwSWvQN2YDDw,1683
+sdwire/backend/device/usb_device.py,sha256=udfCkHZxnLX4m3FwsVuN04e1S4bA3BWBs7t7Kct_k5I,864
+sdwire/backend/utils.py,sha256=D_eO9LiMCWM9UPPumXfmqY30UxsHn-PvpURKJYFdmEY,1422
+sdwire/constants.py,sha256=s5mCx5NlJUM3aBbCPq2y3oOFoPPJ5wA3BR1snyK-cMg,119
+sdwire/main.py,sha256=HVXRAVj2VKkzIeH0ghW9z3VytdaiaB3dQlXFcIFc3xU,2312
+sdwire-0.2.3.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+sdwire-0.2.3.dist-info/METADATA,sha256=czV1BkIDgIjRCp4VcbTvAzRbSfh0FHG7sLcFZ2Y7FWE,3198
+sdwire-0.2.3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+sdwire-0.2.3.dist-info/entry_points.txt,sha256=pxy0zJKVcNWXPk5PtNjTLExOBpqFNth37wtdYdYRXCE,43
+sdwire-0.2.3.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,,