Mirrors/fediplug
1
0
Fork 0
mirror of https://github.com/nova-r/fediplug.git synced 2025-03-10 07:39:00 +01:00
Code Issues Projects Releases Packages Wiki Activity

prevent some buttplugio related crashes

Browse source 
Co-authored-by. M\303\246ve Rey <42996147M-rey@users.noreply.github.com>
This commit is contained in:
Nova 2023-03-14 01:09:42 +01:00
parent 09045cdc82
commit cda5265d60
1 changed files with 31 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

37
fediplug/buttplugio.py
View file

@ -20,16 +20,26 @@ async def connect_plug_client(websocket_url: str) -> Client:
await plug_client.connect(connector)
except Exception as e:
logging.error(f"Could not connect to server, exiting: {e}")
return
exit(1)
print("plug client connected")
return plug_client
async def scan_devices(plug_client: Client) -> Client:
MAX_RETRIES = 5
for i in range(MAX_RETRIES):
# scan for devices for 5 seconds
if options["debug"]:
print(f"scanning for devices ({i+1}/{MAX_RETRIES}) tries")
await plug_client.start_scanning()
await asyncio.sleep(5)
await asyncio.sleep(3)
await plug_client.stop_scanning()
await asyncio.sleep(0.5)
if plug_client.devices:
break
else:
logging.error(f"No devices found, exiting...")
exit(1)
plug_client.logger.info(f"Devices: {plug_client.devices}")
# If we have any device we can print the list of devices
@ -48,19 +58,22 @@ async def trigger_actuators(
duration = clamp(actuator_command[0], 0, MAX_DURATION)
power = clamp(actuator_command[1], 0, MAX_POWER)
# If we have any device we can access it by its ID:
device = plug_client.devices[0]
if len(device.actuators) != 0:
print(len(device.actuators), "actuators found")
# cycle through all actuators in device
print(device.actuators)
print(f"{duration=} {power=}")
for actuator in device.actuators:
print(plug_client.devices)
for key in plug_client.devices:
device = plug_client.devices[key]
try:
if options["debug"]:
print(f"using device {device}")
print(len(device.actuators), f"actuators found:{device.actuators}")
# print(device.actuators)
print(f"sending {duration=} {power=}")
for actuator in device.actuators: # cycle through all actuators in device
await actuator.command(power)
print("generic actuator")
await asyncio.sleep(duration)
# stops all actuators
for actuator in device.actuators:
for actuator in device.actuators: # stops all actuators
await actuator.command(0)
except Exception as e:
logging.error(f"Could not find devices/actuators: {e}")
"""