Rename wanmode.py to xmo-web.py

Generalize into a common cli
This commit is contained in:
up-n-atom
2023-08-17 00:28:57 -04:00
committed by GitHub
parent 365b381ae5
commit 14baec2cea
+39 -10
View File
@@ -1,14 +1,11 @@
#!/usr/bin/env python3
import asyncclick as click
from enum import Enum
from ipaddress import IPv4Address
from json import dumps
from sagemcom_api.client import SagemcomClient
from sagemcom_api.enums import EncryptionMethod
XPATH_DEVICE_SERVICES_BELLNETWORKCFG_WANMODE = 'Device/Services/BellNetworkCfg/WanMode'
class EnumChoice(click.Choice):
def __init__(self, enum: Enum, case_sensitive=False):
self.__enum = enum
@@ -22,7 +19,7 @@ class EnumChoice(click.Choice):
return self.__enum(converted_str)
@click.command()
@click.group(chain=True)
@click.option('-H', '--host', default='192.168.2.1', help='Hostname or host IP')
@click.option('-u', '--username', default='admin', help='Administrator username')
@click.option('-p', '--password', required=True, help='Administrator password')
@@ -30,17 +27,49 @@ class EnumChoice(click.Choice):
default=EncryptionMethod.SHA512, type=EnumChoice(EncryptionMethod),
help='Authentication method')
@click.pass_context
async def main(ctx: click.Context, host: str, username: str, password: str, authentication_method: EncryptionMethod) -> None:
async def cli(ctx, host: str, username: str, password: str, authentication_method: EncryptionMethod) -> None:
ctx.obj = client = await ctx.with_async_resource(
SagemcomClient(host, username, password, authentication_method, ssl=True, keep_keys=True)
)
try:
await client.login()
wan_mode = await client.get_value_by_xpath(XPATH_DEVICE_SERVICES_BELLNETWORKCFG_WANMODE)
except Exception as e:
raise click.Abort(e)
@cli.command()
@click.option('--path', required=True)
@click.pass_context
async def get_value(ctx: click.Context, path: str) -> None:
if not isinstance(ctx.obj, SagemcomClient):
return
try:
value = await ctx.obj.get_value_by_xpath(path)
except Exception as e:
click.echo(e, err=True)
else:
click.echo(f"WAN Mode: {wan_mode}")
click.echo(dumps(value, indent=2))
@cli.command()
@click.option('--path', required=True)
@click.option('--value', required=True)
@click.pass_context
async def set_value(ctx: click.Context, path: str, value: str) -> None:
if not isinstance(ctx.obj, SagemcomClient):
return
try:
value = await ctx.obj.set_value_by_xpath(path, value)
except Exception as e:
click.echo(e, err=True)
@cli.command()
@click.pass_context
async def get_wan_mode(ctx: click.Context) -> None:
await ctx.invoke(get_value, path='Device/Services/BellNetworkCfg/WanMode')
if __name__ == '__main__':
main(_anyio_backend='asyncio')
cli(_anyio_backend='asyncio')