From c83531b0c2be6cb4c5cfffe9ffd35679c3fd2933 Mon Sep 17 00:00:00 2001
From: up-n-atom <adam.jaremko@gmail.com>
Date: Sun, 19 Nov 2023 17:10:51 -0500
Subject: [PATCH] Add mode helpers

---
 xmo/__main__.py |  1 +
 xmo/mode.py     | 16 ++++++++++++++++
 xmo/xmo.py      |  2 +-
 3 files changed, 18 insertions(+), 1 deletion(-)
 create mode 100644 xmo/mode.py

diff --git a/xmo/__main__.py b/xmo/__main__.py
index ac3e160..f224898 100644
--- a/xmo/__main__.py
+++ b/xmo/__main__.py
@@ -1,6 +1,7 @@
 import os
 import xmo.dns
 import xmo.dmz
+import xmo.mode
 import xmo.wifi
 import yaml
 from . import xmo
diff --git a/xmo/mode.py b/xmo/mode.py
new file mode 100644
index 0000000..1856c9c
--- /dev/null
+++ b/xmo/mode.py
@@ -0,0 +1,16 @@
+import asyncclick as click
+import re
+from sagemcom_api.client import SagemcomClient
+from . import xmo
+
+
+@xmo.cli.command()
+@click.pass_context
+async def get_onu_mode(ctx: click.Context) -> None:
+    await ctx.invoke(xmo.get_value, path=['Device/Optical/G988/OnuMode'])
+
+
+@xmo.cli.command()
+@click.pass_context
+async def get_wan_mode(ctx: click.Context) -> None:
+    await ctx.invoke(xmo.get_value, path=['Device/Services/BellNetworkCfg/WanMode'])
diff --git a/xmo/xmo.py b/xmo/xmo.py
index 0916e81..2089d17 100644
--- a/xmo/xmo.py
+++ b/xmo/xmo.py
@@ -48,7 +48,7 @@ async def cli(ctx: click.Context, host: str, username: str, password: str, auth_
 @cli.command()
 @click.option('--path', required=True, multiple=True)
 @click.pass_context
-async def get_value(ctx: click.Context, path: str) -> None:
+async def get_value(ctx: click.Context, path: list[str]) -> None:
     client = ctx.find_object(SagemcomClient)
     if client is None:
         return