pstruebi/bumble_fork
1
0
Fork 0
forked from auracaster/bumble_mirror
Code Pull Requests Activity

pyusb: power_cycle if '!' is present at the start of the transport

Browse Source
This commit is contained in:
Charlie Boutier
2024-04-17 17:37:10 +00:00
committed by Slvr
parent 049f9021e9
commit 71df062e07
1 changed files with 10 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
bumble/transport/pyusb.py
View File

@@ -227,6 +227,10 @@ async def open_pyusb_transport(spec: str) -> Transport:
usb_find = libusb_package.find usb_find = libusb_package.find
# Find the device according to the spec moniker # Find the device according to the spec moniker
power_cycle = False
if spec.startswith('!'):
power_cycle = True
spec = spec[1:]
if ':' in spec: if ':' in spec:
vendor_id, product_id = spec.split(':') vendor_id, product_id = spec.split(':')
device = usb_find(idVendor=int(vendor_id, 16), idProduct=int(product_id, 16)) device = usb_find(idVendor=int(vendor_id, 16), idProduct=int(product_id, 16))
@@ -259,11 +263,12 @@ async def open_pyusb_transport(spec: str) -> Transport:
logger.debug(f'USB Device: {device}') logger.debug(f'USB Device: {device}')
# Power Cycle the device # Power Cycle the device
try: if power_cycle:
device = await _power_cycle(device) # type: ignore try:
except Exception as e: device = await _power_cycle(device) # type: ignore
logging.debug(e) except Exception as e:
logging.info(f"Unable to power cycle {hex(device.idVendor)} {hex(device.idProduct)}") # type: ignore logging.debug(e)
logging.info(f"Unable to power cycle {hex(device.idVendor)} {hex(device.idProduct)}") # type: ignore
# Collect the metadata # Collect the metadata
device_metadata = {'vendor_id': device.idVendor, 'product_id': device.idProduct} device_metadata = {'vendor_id': device.idVendor, 'product_id': device.idProduct}