From b3210843ee6dd540aeaad41379933eac0e09c1dc Mon Sep 17 00:00:00 2001
From: Wilfred Tyler Gee <wtylergee@gmail.com>
Date: Fri, 19 Aug 2022 10:16:01 -1000
Subject: [PATCH] Adding power board control to cli (#1171)

* Adds basic reporting and the ability to change relay states.
---
 setup.cfg                            |   1 +
 src/panoptes/pocs/utils/cli/main.py  |   2 +
 src/panoptes/pocs/utils/cli/power.py | 115 +++++++++++++++++++++++++++
 3 files changed, 118 insertions(+)
 create mode 100644 src/panoptes/pocs/utils/cli/power.py

diff --git a/setup.cfg b/setup.cfg
index dbd5ae120..5ecf67e62 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -45,6 +45,7 @@ install_requires =
     Pillow>=9.2.0
     pyserial
     requests
+    sparklines
     transitions
     typer
     uvicorn[standard]
diff --git a/src/panoptes/pocs/utils/cli/main.py b/src/panoptes/pocs/utils/cli/main.py
index 8acc486e6..0edd95991 100644
--- a/src/panoptes/pocs/utils/cli/main.py
+++ b/src/panoptes/pocs/utils/cli/main.py
@@ -3,12 +3,14 @@
 from panoptes.pocs.utils.cli import config
 from panoptes.pocs.utils.cli import sensor
 from panoptes.pocs.utils.cli import image
+from panoptes.pocs.utils.cli import power
 
 app = typer.Typer()
 state = {'verbose': False}
 
 app.add_typer(config.app, name="config", help='Interact with the config server.')
 app.add_typer(sensor.app, name="sensor", help='Interact with system sensors.')
+app.add_typer(power.app, name="power", help='Interact with power relays.')
 app.add_typer(image.app, name="image", help='Interact with images.')
 
 
diff --git a/src/panoptes/pocs/utils/cli/power.py b/src/panoptes/pocs/utils/cli/power.py
new file mode 100644
index 000000000..e07230110
--- /dev/null
+++ b/src/panoptes/pocs/utils/cli/power.py
@@ -0,0 +1,115 @@
+from contextlib import suppress
+from pprint import pprint
+
+import numpy as np
+import requests
+import typer
+from dataclasses import dataclass
+
+from sparklines import sparklines
+from panoptes.pocs.utils.service.power import RelayCommand
+
+
+@dataclass
+class HostInfo:
+    host: str = 'localhost'
+    port: str = '6564'
+
+    @property
+    def url(self):
+        return f'http://{self.host}:{self.port}'
+
+
+app = typer.Typer()
+
+
+@app.callback()
+def common(context: typer.Context,
+           host: str = typer.Option('localhost', help='Power monitor host address.'),
+           port: str = typer.Option('6564', help='Power monitor port.'),
+           ):
+    context.obj = HostInfo(host=host, port=port)
+
+
+@app.command()
+def status(context: typer.Context):
+    """Get the status of the power monitor."""
+    url = context.obj.url
+    try:
+        res = requests.get(url)
+        if res.ok:
+            relays = res.json()
+            for relay_index, relay_info in relays.items():
+                with suppress(KeyError, TypeError):
+                    relay_label = typer.style(f'{relay_info["label"]:.<20s}',
+                                              fg=typer.colors.BRIGHT_CYAN)
+                    if relay_info['state'] == 'ON':
+                        status_color = typer.colors.BRIGHT_GREEN
+                    else:
+                        status_color = typer.colors.BRIGHT_RED
+                    status_text = typer.style(relay_info['state'], fg=status_color)
+                    typer.echo(f'[{relay_index}] {relay_label} {status_text}')
+        else:
+            typer.secho(res.content.decode(), fg=typer.colors.RED)
+    except requests.exceptions.ConnectionError:
+        typer.secho(f'Cannot connect to {url}', fg=typer.colors.RED)
+
+
+@app.command()
+def readings(context: typer.Context):
+    """Get the readings of the relays."""
+    url = context.obj.url + '/readings'
+    try:
+        res = requests.get(url)
+        if res.ok:
+            relays = res.json()
+            for relay_label, relay_readings in relays.items():
+                relay_text = typer.style(f'{relay_label:.<20s}', fg=typer.colors.CYAN)
+                relay_readings = [int(x) if int(x) >= 0 else 0 for x in relay_readings.values()]
+                for val in sparklines(relay_readings):
+                    typer.echo(f'{relay_text} {val} [{np.array(relay_readings).mean():.0f}]')
+        else:
+            typer.secho(res.content.decode(), fg=typer.colors.RED)
+    except requests.exceptions.ConnectionError:
+        typer.secho(f'Cannot connect to {url}', fg=typer.colors.RED)
+
+
+@app.command()
+def on(
+        context: typer.Context,
+        relay: str = typer.Argument(..., help='The label or index of the relay to turn on.'),
+):
+    """Turns a relay on."""
+    control(context, relay=relay, command='turn_on')
+
+
+@app.command()
+def off(
+        context: typer.Context,
+        relay: str = typer.Argument(..., help='The label or index of the relay to turn off.'),
+):
+    """Turns a relay off."""
+    control(context, relay=relay, command='turn_off')
+
+
+@app.command()
+def control(
+        context: typer.Context,
+        relay: str = typer.Option(..., help='The label or index of the relay to control.'),
+        command: str = typer.Option(..., help='The control action to perform, '
+                                              'either "turn_on" or "turn_off"')
+):
+    """Control a relay by label or relay index."""
+    url = context.obj.url + '/control'
+
+    try:
+        relay_command = RelayCommand(relay=relay, command=command)
+        res = requests.post(url, data=relay_command.json())
+        content = res.json() if res.ok else res.content.decode()
+        typer.secho(pprint(content))
+    except requests.exceptions.ConnectionError:
+        typer.secho(f'Cannot connect to {url}', fg=typer.colors.RED)
+
+
+if __name__ == "__main__":
+    app()