Restructure
Pacakge as a CLI application
This commit is contained in:
+24
@@ -0,0 +1,24 @@
|
||||
import asyncclick as click
|
||||
from sagemcom_api.client import SagemcomClient
|
||||
from . import xmo
|
||||
|
||||
@xmo.cli.command()
|
||||
@click.option('-r', '--radios', multiple=True)
|
||||
@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 \
|
||||
radio['Enable']}
|
||||
if not len(radios):
|
||||
radios = click.prompt('Choose radio', type=click.Choice(active_radios), show_choices=True),
|
||||
invalid_radios = set(radios) - active_radios
|
||||
if len(invalid_radios):
|
||||
raise click.BadParameter("Invalid radio(s): {0}".format(", ".join(invalid_radios)))
|
||||
disable_radios = set(radios) & active_radios
|
||||
for alias in disable_radios:
|
||||
await client.set_value_by_xpath(f"Device/WiFi/Radios/Radio[Alias='{alias}']/Enable", False)
|
||||
except Exception as e:
|
||||
click.echo(e, err=True)
|
||||
raise click.Abort()
|
||||
|
||||
Reference in New Issue
Block a user