From 14baec2cea28c99b462d3166a351979f73ac39dc Mon Sep 17 00:00:00 2001
From: up-n-atom <adam.jaremko@gmail.com>
Date: Thu, 17 Aug 2023 00:28:57 -0400
Subject: [PATCH] Rename wanmode.py to xmo-web.py

Generalize into a common cli
---
 wanmode.py => xmo-web.py | 49 ++++++++++++++++++++++++++++++++--------
 1 file changed, 39 insertions(+), 10 deletions(-)
 rename wanmode.py => xmo-web.py (53%)

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')
+