sdwire 0.3.0__tar.gz → 0.3.1__tar.gz

{sdwire-0.3.0 → sdwire-0.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sdwire
-Version: 0.3.0
+Version: 0.3.1
 Summary: CLI application to interact with Badgerd SDWire devices
 License: GPL-3
 Author: Talha Can Havadar

{sdwire-0.3.0 → sdwire-0.3.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "sdwire"
-version = "0.3.0"
+version = "0.3.1"
 description = "CLI application to interact with Badgerd SDWire devices"
 authors = ["Talha Can Havadar <havadartalha@gmail.com>"]
 license = "GPL-3"

{sdwire-0.3.0 → sdwire-0.3.1}/sdwire/backend/device/sdwire.py

@@ -17,6 +17,20 @@ class SDWire(USBDevice):
         self.generation = generation
         self._update_block_device()
 
+    def query_device_state(self):
+        if not self.usb_device:
+            log.error("USB device not available")
+            return
+
+        try:
+            state = self.usb_device.is_kernel_driver_active(0)
+            return state
+        except Exception as e:
+            log.debug(
+                "not able to query device state. err: %s",
+                e,
+            )
+
     def switch_ts(self) -> None:
         if not self.usb_device:
             log.error("USB device not available")

{sdwire-0.3.0 → sdwire-0.3.1}/sdwire/backend/utils.py

@@ -82,3 +82,50 @@ def handle_switch_command(ctx, serial):
             raise click.UsageError(
                 f"There is no such sdwire device connected with serial={serial}"
             )
+
+
+def handle_state_command(ctx, serial):
+    """Handle querying device state.
+
+    Args:
+        ctx: Click context containing device information
+        serial: Serial number of the device to query
+    """
+
+    devices = detect.get_sdwire_devices()
+
+    if serial is None:
+        # check the devices
+        if len(devices) == 0:
+            raise click.UsageError("There is no sdwire device connected!")
+        if len(devices) > 1:
+            raise click.UsageError(
+                "There is more then 1 sdwire device connected, please use --serial|-s to specify!"
+            )
+        log.info("1 sdwire/sdwirec device detected")
+        ctx.obj["device"] = devices[0]
+    else:
+        for device in devices:
+            if device.serial_string == serial:
+                ctx.obj["device"] = device
+                break
+        else:
+            raise click.UsageError(
+                f"There is no such sdwire device connected with serial={serial}"
+            )
+
+    device = ctx.obj["device"]
+    if isinstance(device, (SDWireC)):
+        log.info(
+            "SDWireC or legacy sdwire devices don't have state functionality"
+        )
+        print("SDWireC don't have state functionality implemented")
+        sys.exit(1)
+    try:
+        state = device.query_device_state()
+        print(f"{'Serial':<30}\tState")
+        print(f"{device.serial_string:<30}\t{'Host' if state else 'Target'}")
+    except Exception as e:
+        log.error(f"Failed to get device state: {e}")
+        print(f"Error: Failed to get device state: {e}")
+        sys.exit(1)

{sdwire-0.3.0 → sdwire-0.3.1}/sdwire/main.py

@@ -29,6 +29,20 @@ def list() -> None:
         print(sdwire)
 
 
+@main.command()
+@click.pass_context
+@click.option(
+    "-s",
+    "--serial",
+    required=False,
+    help="Serial number of the sdwire device, if there is only one sdwire connected then it will be used by default",
+)
+def state(ctx: click.Context, serial=None):
+    """Query and display the state of all connected SDWire devices."""
+    ctx.ensure_object(dict)
+    utils.handle_state_command(ctx, serial)
+
+
 @main.group()
 @click.pass_context
 @click.option(