azalea/sagemcom
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Use explicit type hints

Browse Source
This commit is contained in:
up-n-atom
2023-09-14 21:01:10 -04:00
committed by GitHub
parent aa932620a7
commit b91c6e2f8f
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
xmo-remote-client.py
+2 -2
View File
@@ -136,7 +136,7 @@ async def set_dns_servers(client: SagemcomClient, dns_servers: tuple[IPv4Address
@cli.command() @cli.command()
@click.option('-r', '--radio', multiple=True) @click.option('-r', '--radio', multiple=True)
@click.pass_obj @click.pass_obj
async def disable_wifi_radio(client: SagemcomClient, radio: Any) -> None: async def disable_wifi_radio(client: SagemcomClient, radio: tuple | list) -> None:
try: try:
value = await client.get_value_by_xpath('Device/WiFi/Radios') value = await client.get_value_by_xpath('Device/WiFi/Radios')
active_radios = {radio['Alias'] for radio in value if 'Alias' in radio and \ active_radios = {radio['Alias'] for radio in value if 'Alias' in radio and \
@@ -145,7 +145,7 @@ async def disable_wifi_radio(client: SagemcomClient, radio: Any) -> None:
radio = click.prompt('Choose radio', type=click.Choice(active_radios), show_choices=True), radio = click.prompt('Choose radio', type=click.Choice(active_radios), show_choices=True),
radios = set(radio) - active_radios radios = set(radio) - active_radios
if len(radios): if len(radios):
raise click.BadParameter("Invalid radio(s) {0}".format(", ".join(radios))) raise click.BadParameter("Invalid radio(s): {0}".format(", ".join(radios)))
radios = set(radio) & active_radios radios = set(radio) & active_radios
for alias in radios: for alias in radios:
await client.set_value_by_xpath(f"Device/WiFi/Radios/Radio[Alias='{alias}']/Enable", False) await client.set_value_by_xpath(f"Device/WiFi/Radios/Radio[Alias='{alias}']/Enable", False)