python-omnilogic-local 0.15.2__tar.gz → 0.17.0__tar.gz

{python_omnilogic_local-0.15.2 → python_omnilogic_local-0.17.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: python-omnilogic-local
-Version: 0.15.2
+Version: 0.17.0
 Summary: A library for local control of Hayward OmniHub/OmniLogic pool controllers using their local API
 License: Apache-2.0
 Author: Chris Jowett
@@ -12,6 +12,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
+Requires-Dist: click (>=8.0.0,<8.1.0)
 Requires-Dist: pydantic (>=1.10.17)
 Requires-Dist: xmltodict (>=0.13.0,<0.14.0)
 Description-Content-Type: text/markdown

{python_omnilogic_local-0.15.2 → python_omnilogic_local-0.17.0}/pyomnilogic_local/api.py

@@ -572,3 +572,44 @@ class OmniLogicAPI:
         req_body = ET.tostring(body_element, xml_declaration=True, encoding="unicode")
 
         return await self.async_send_message(MessageType.SET_SPILLOVER, req_body, False)
+
+    async def async_set_group_enable(
+        self,
+        group_id: int,
+        enabled: int | bool,
+        is_countdown_timer: bool = False,
+        start_time_hours: int = 0,
+        start_time_minutes: int = 0,
+        end_time_hours: int = 0,
+        end_time_minutes: int = 0,
+        days_active: int = 0,
+        recurring: bool = False,
+    ) -> None:
+        body_element = ET.Element("Request", {"xmlns": "http://nextgen.hayward.com/api"})
+
+        name_element = ET.SubElement(body_element, "Name")
+        name_element.text = "RunGroupCmd"
+
+        parameters_element = ET.SubElement(body_element, "Parameters")
+        parameter = ET.SubElement(parameters_element, "Parameter", name="GroupID", dataType="int")
+        parameter.text = str(group_id)
+        parameter = ET.SubElement(parameters_element, "Parameter", name="Data", dataType="int")
+        parameter.text = str(int(enabled))
+        parameter = ET.SubElement(parameters_element, "Parameter", name="IsCountDownTimer", dataType="bool")
+        parameter.text = str(int(is_countdown_timer))
+        parameter = ET.SubElement(parameters_element, "Parameter", name="StartTimeHours", dataType="int")
+        parameter.text = str(start_time_hours)
+        parameter = ET.SubElement(parameters_element, "Parameter", name="StartTimeMinutes", dataType="int")
+        parameter.text = str(start_time_minutes)
+        parameter = ET.SubElement(parameters_element, "Parameter", name="EndTimeHours", dataType="int")
+        parameter.text = str(end_time_hours)
+        parameter = ET.SubElement(parameters_element, "Parameter", name="EndTimeMinutes", dataType="int")
+        parameter.text = str(end_time_minutes)
+        parameter = ET.SubElement(parameters_element, "Parameter", name="DaysActive", dataType="int")
+        parameter.text = str(days_active)
+        parameter = ET.SubElement(parameters_element, "Parameter", name="Recurring", dataType="bool")
+        parameter.text = str(int(recurring))
+
+        req_body = ET.tostring(body_element, xml_declaration=True, encoding="unicode")
+
+        return await self.async_send_message(MessageType.RUN_GROUP_CMD, req_body, False)

python_omnilogic_local-0.17.0/pyomnilogic_local/cli/cli.py

@@ -0,0 +1,25 @@
+import asyncio
+
+import click
+
+from pyomnilogic_local.api import OmniLogicAPI
+from pyomnilogic_local.cli.debug import commands as debug
+from pyomnilogic_local.cli.get import commands as get
+
+
+async def get_omni(host: str) -> OmniLogicAPI:
+    return OmniLogicAPI(host, 10444, 5.0)
+
+
+@click.group()
+@click.pass_context
+@click.option("--host", default="127.0.0.1", help="Hostname or IP address of omnilogic system")
+def entrypoint(ctx: click.Context, host: str) -> None:
+    ctx.ensure_object(dict)
+    omni = asyncio.run(get_omni(host))
+
+    ctx.obj["OMNI"] = omni
+
+
+entrypoint.add_command(debug.debug)
+entrypoint.add_command(get.get)

python_omnilogic_local-0.17.0/pyomnilogic_local/cli/debug/commands.py

@@ -0,0 +1,95 @@
+# Need to figure out how to resolve the 'Untyped decorator makes function "..." untyped' errors in mypy when using click decorators
+# mypy: disable-error-code="misc"
+import asyncio
+from typing import Literal, overload
+
+import click
+
+from pyomnilogic_local.api import OmniLogicAPI
+from pyomnilogic_local.cli.utils import async_get_mspconfig, async_get_telemetry
+from pyomnilogic_local.models.filter_diagnostics import FilterDiagnostics
+
+
+@click.group()
+@click.option("--raw/--no-raw", default=False, help="Output the raw XML from the OmniLogic, do not parse the response")
+@click.pass_context
+def debug(ctx: click.Context, raw: bool) -> None:
+    # Container for all get commands
+
+    ctx.ensure_object(dict)
+    ctx.obj["RAW"] = raw
+
+
+@debug.command()
+@click.pass_context
+def get_mspconfig(ctx: click.Context) -> None:
+    mspconfig = asyncio.run(async_get_mspconfig(ctx.obj["OMNI"], ctx.obj["RAW"]))
+    click.echo(mspconfig)
+
+
+@debug.command()
+@click.pass_context
+def get_telemetry(ctx: click.Context) -> None:
+    telemetry = asyncio.run(async_get_telemetry(ctx.obj["OMNI"], ctx.obj["RAW"]))
+    click.echo(telemetry)
+
+
+@debug.command()
+@click.pass_context
+def get_alarm_list(ctx: click.Context) -> None:
+    alarm_list = asyncio.run(async_get_alarm_list(ctx.obj["OMNI"]))
+    click.echo(alarm_list)
+
+
+async def async_get_alarm_list(omni: OmniLogicAPI) -> str:
+    alarm_list = await omni.async_get_alarm_list()
+    return alarm_list
+
+
+@debug.command()
+@click.option("--pool-id", help="System ID of the Body Of Water the filter is associated with")
+@click.option("--filter-id", help="System ID of the filter to request diagnostics for")
+@click.pass_context
+def get_filter_diagnostics(ctx: click.Context, pool_id: int, filter_id: int) -> None:
+    filter_diags = asyncio.run(async_get_filter_diagnostics(ctx.obj["OMNI"], pool_id, filter_id, ctx.obj["RAW"]))
+    if ctx.obj["RAW"]:
+        click.echo(filter_diags)
+    else:
+        drv1 = chr(filter_diags.get_param_by_name("DriveFWRevisionB1"))
+        drv2 = chr(filter_diags.get_param_by_name("DriveFWRevisionB2"))
+        drv3 = chr(filter_diags.get_param_by_name("DriveFWRevisionB3"))
+        drv4 = chr(filter_diags.get_param_by_name("DriveFWRevisionB4"))
+        dfw1 = chr(filter_diags.get_param_by_name("DisplayFWRevisionB1"))
+        dfw2 = chr(filter_diags.get_param_by_name("DisplayFWRevisionB2"))
+        dfw3 = chr(filter_diags.get_param_by_name("DisplayFWRevisionB3"))
+        dfw4 = chr(filter_diags.get_param_by_name("DisplayFWRevisionB4"))
+        pow1 = filter_diags.get_param_by_name("PowerMSB")
+        pow2 = filter_diags.get_param_by_name("PowerLSB")
+        errs = filter_diags.get_param_by_name("ErrorStatus")
+        click.echo(
+            f"DRIVE FW REV: {drv1}{drv2}.{drv3}{drv4}\n"
+            f"DISPLAY FW REV: {dfw1}{dfw2}.{dfw3}.{dfw4}\n"
+            f"POWER: {pow1:x}{pow2:x}W\n"
+            f"ERROR STATUS: {errs}"
+        )
+
+
+@overload
+async def async_get_filter_diagnostics(omni: OmniLogicAPI, pool_id: int, filter_id: int, raw: Literal[True]) -> str: ...
+@overload
+async def async_get_filter_diagnostics(omni: OmniLogicAPI, pool_id: int, filter_id: int, raw: Literal[False]) -> FilterDiagnostics: ...
+async def async_get_filter_diagnostics(omni: OmniLogicAPI, pool_id: int, filter_id: int, raw: bool) -> FilterDiagnostics | str:
+    filter_diags = await omni.async_get_filter_diagnostics(pool_id, filter_id, raw=raw)
+    return filter_diags
+
+
+@debug.command()
+@click.pass_context
+def get_log_config(ctx: click.Context) -> None:
+    log_config = asyncio.run(async_get_log_config(ctx.obj["OMNI"]))
+    click.echo(log_config)
+
+
+async def async_get_log_config(omni: OmniLogicAPI) -> str:
+    log_config = await omni.async_get_log_config()
+    return log_config

python_omnilogic_local-0.17.0/pyomnilogic_local/cli/get/commands.py

@@ -0,0 +1,33 @@
+# Need to figure out how to resolve the 'Untyped decorator makes function "..." untyped' errors in mypy when using click decorators
+# mypy: disable-error-code="misc"
+import asyncio
+
+import click
+
+from pyomnilogic_local.cli.utils import async_get_mspconfig
+
+
+@click.group()
+@click.pass_context
+def get(ctx: click.Context) -> None:
+    # Container for all get commands
+
+    ctx.ensure_object(dict)
+
+
+@get.command()
+@click.pass_context
+def lights(ctx: click.Context) -> None:
+    mspconfig = asyncio.run(async_get_mspconfig(ctx.obj["OMNI"]))
+    # Return data about lights in the backyard
+    if mspconfig.backyard.colorlogic_light:
+        for light in mspconfig.backyard.colorlogic_light:
+            click.echo(light)
+
+    # Return data about lights in the Body of Water
+    if mspconfig.backyard.bow:
+        for bow in mspconfig.backyard.bow:
+            if bow.colorlogic_light:
+                for cl_light in bow.colorlogic_light:
+                    for k, v in cl_light:
+                        click.echo(f"{k:15}\t{str(v)}")

python_omnilogic_local-0.17.0/pyomnilogic_local/cli/utils.py

@@ -0,0 +1,15 @@
+from pyomnilogic_local.api import OmniLogicAPI
+from pyomnilogic_local.models.mspconfig import MSPConfig
+from pyomnilogic_local.models.telemetry import Telemetry
+
+
+async def async_get_mspconfig(omni: OmniLogicAPI, raw: bool = False) -> MSPConfig:
+    mspconfig: MSPConfig
+    mspconfig = await omni.async_get_config(raw=raw)
+    return mspconfig
+
+
+async def async_get_telemetry(omni: OmniLogicAPI, raw: bool = False) -> Telemetry:
+    telemetry: Telemetry
+    telemetry = await omni.async_get_telemetry(raw=raw)
+    return telemetry

{python_omnilogic_local-0.15.2 → python_omnilogic_local-0.17.0}/pyomnilogic_local/omnitypes.py

@@ -23,6 +23,7 @@ class MessageType(Enum):
     GET_ALARM_LIST = 304
     SET_STANDALONE_LIGHT_SHOW = 308
     SET_SPILLOVER = 311
+    RUN_GROUP_CMD = 317
     RESTORE_IDLE_STATE = 340
     GET_FILTER_DIAGNOSTIC_INFO = 386
     HANDSHAKE = 1000
@@ -63,6 +64,9 @@ class OmniType(str, Enum):
     VALVE_ACTUATOR = "ValveActuator"
     VIRT_HEATER = "VirtualHeater"
 
+    def __str__(self) -> str:
+        return OmniType[self.name].value
+
 
 # Backyard/BoW
 class BackyardState(PrettyEnum):
@@ -172,6 +176,9 @@ class ColorLogicShow(PrettyEnum):
     WARM_WHITE = 25
     BRIGHT_YELLOW = 26
 
+    def __str__(self) -> str:
+        return self.name
+
 
 class ColorLogicPowerState(PrettyEnum):
     OFF = 0
@@ -187,6 +194,9 @@ class ColorLogicLightType(str, PrettyEnum):
     FOUR_ZERO = "COLOR_LOGIC_4_0"
     TWO_FIVE = "COLOR_LOGIC_2_5"
 
+    def __str__(self) -> str:
+        return ColorLogicLightType[self.name].value
+
 
 class CSADType(str, PrettyEnum):
     ACID = "ACID"

{python_omnilogic_local-0.15.2 → python_omnilogic_local-0.17.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "python-omnilogic-local"
-version = "0.15.2"
+version = "0.17.0"
 description = "A library for local control of Hayward OmniHub/OmniLogic pool controllers using their local API"
 authors = [
     {name = "Chris Jowett",email = "421501+cryptk@users.noreply.github.com"},
@@ -12,11 +12,12 @@ readme = "README.md"
 requires-python = ">=3.10,<3.14"
 dependencies = [
     "pydantic (>=1.10.17)",
-    "xmltodict (>=0.13.0,<0.14.0)"
+    "xmltodict (>=0.13.0,<0.14.0)",
+    "click (>=8.0.0,<8.1.0)",
 ]
 
 [project.scripts]
-omnilogic = "pyomnilogic_local.cli:main"
+omnilogic = "pyomnilogic_local.cli.cli:entrypoint"
 
 [tool.poetry]
 packages = [{include = "pyomnilogic_local"}]
@@ -41,7 +42,9 @@ profile = "black"
 
 [tool.mypy]
 python_version = "3.13"
-plugins = "pydantic.mypy"
+plugins = [
+    "pydantic.mypy",
+]
 follow_imports = "silent"
 strict = true
 ignore_missing_imports = true
@@ -56,6 +59,9 @@ warn_untyped_fields = true
 
 [tool.pylint.MAIN]
 py-version = "3.13"
+extension-pkg-allow-list = [
+    "pydantic",
+]
 ignore = [
     "tests",
 ]