From 5a385d5edb508dbe563f45a6b45ca81e6c5f1593 Mon Sep 17 00:00:00 2001 From: up-n-atom Date: Fri, 29 Dec 2023 19:48:48 -0500 Subject: [PATCH] Clean logic --- xmo/wifi.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/xmo/wifi.py b/xmo/wifi.py index 6dd854f..125994f 100644 --- a/xmo/wifi.py +++ b/xmo/wifi.py @@ -8,13 +8,14 @@ from . import xmo @click.pass_obj async def disable_wifi_radios(client: SagemcomClient, radios: tuple[str] | list[str]) -> None: try: - value = await client.get_value_by_xpath('Device/WiFi/Radios') - active_radios = {radio['alias'] for radio in value if radio.keys() >= {'alias', 'enable'} and \ + _radios = await client.get_value_by_xpath('Device/WiFi/Radios') + active_radios = {radio['alias'] for radio in _radios \ + if radio.keys() >= {'alias', 'enable'} and \ radio['enable']} - if not len(active_radios): + if active_radios: click.echo('No active radios') return - if not len(radios): + if radios: radios = click.prompt('Choose radio', type=click.Choice(active_radios), show_choices=True), invalid_radios = set(radios) - active_radios if len(invalid_radios):