devices/cli: add --json parameter

This commit is contained in:
InsanePrawn 2023-01-02 03:54:31 +01:00
parent 63156776a2
commit 932e739255
2 changed files with 42 additions and 15 deletions

View file

@ -1,4 +1,5 @@
import click
from json import dumps as json_dump
import logging
from config.state import config
@ -7,7 +8,8 @@ from .device import get_devices, get_profile_device
@click.command(name='devices')
def cmd_devices():
@click.option('--json', is_flag=True, help='output machine-parsable JSON format')
def cmd_devices(json: bool = False):
'list the available devices and descriptions'
devices = get_devices()
if not devices:
@ -20,6 +22,7 @@ def cmd_devices():
except Exception as ex:
logging.debug(f"Failed to get profile device for visual highlighting, not a problem: {ex}")
output = ['']
json_output = {}
for name in sorted(devices.keys()):
prefix = ''
suffix = ''
@ -29,6 +32,9 @@ def cmd_devices():
device.parse_deviceinfo(try_download=False)
except Exception as ex:
logging.debug(f"Failed to parse deviceinfo for extended description, not a problem: {ex}")
if json:
json_output[name] = device.get_summary().toDict()
continue
if profile_device and profile_device.name == device.name:
prefix = '>>> '
suffix = f'\n\nCurrently selected by profile "{config.file.profiles.current}"'
@ -36,5 +42,7 @@ def cmd_devices():
# prefix each line in the snippet
snippet = '\n'.join([f'{prefix}{line}' for line in snippet.split('\n')])
output.append(f"{snippet}\n")
if json:
output = ['\n' + json_dump(json_output, indent=4)]
for line in output:
print(line)