diff --git a/wanmode.py b/xmo-web.py similarity index 53% rename from wanmode.py rename to xmo-web.py index 376daa1..3cf2579 100755 --- a/wanmode.py +++ b/xmo-web.py @@ -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') +