feat(cli): Implement Rule Engine management commands

Author: rnovatorov

src/enapter/cli/http/api/command.py

@@ -5,6 +5,7 @@
 from .blueprint_command import BlueprintCommand
 from .command_command import CommandCommand
 from .device_command import DeviceCommand
+from .rule_engine_command import RuleEngineCommand
 from .site_command import SiteCommand
 from .telemetry_command import TelemetryCommand
 
@@ -21,6 +22,7 @@ def register(parent: cli.Subparsers) -> None:
             BlueprintCommand,
             CommandCommand,
             DeviceCommand,
+            RuleEngineCommand,
             SiteCommand,
             TelemetryCommand,
         ]:
@@ -35,9 +37,11 @@ async def run(args: argparse.Namespace) -> None:
                 await CommandCommand.run(args)
             case "device":
                 await DeviceCommand.run(args)
+            case "rule-engine":
+                await RuleEngineCommand.run(args)
             case "site":
                 await SiteCommand.run(args)
             case "telemetry":
                 await TelemetryCommand.run(args)
             case _:
-                raise NotImplementedError(args.device_command)
+                raise NotImplementedError(args.api_command)

src/enapter/cli/http/api/rule_engine_command.py

@@ -0,0 +1,40 @@
+"""Rule Engine command group."""
+
+import argparse
+
+from enapter import cli
+
+from .rule_engine_get_command import RuleEngineGetCommand
+from .rule_engine_resume_command import RuleEngineResumeCommand
+from .rule_engine_suspend_command import RuleEngineSuspendCommand
+
+
+class RuleEngineCommand(cli.Command):
+    """Command group for Rule Engine management."""
+
+    @staticmethod
+    def register(parent: cli.Subparsers) -> None:
+        """Register the command group in the subparsers."""
+        parser = parent.add_parser(
+            "rule-engine", formatter_class=argparse.ArgumentDefaultsHelpFormatter
+        )
+        subparsers = parser.add_subparsers(dest="rule_engine_command", required=True)
+        for command in [
+            RuleEngineGetCommand,
+            RuleEngineSuspendCommand,
+            RuleEngineResumeCommand,
+        ]:
+            command.register(subparsers)
+
+    @staticmethod
+    async def run(args: argparse.Namespace) -> None:
+        """Run the sub-command."""
+        match args.rule_engine_command:
+            case "get":
+                await RuleEngineGetCommand.run(args)
+            case "suspend":
+                await RuleEngineSuspendCommand.run(args)
+            case "resume":
+                await RuleEngineResumeCommand.run(args)
+            case _:
+                raise NotImplementedError(args.rule_engine_command)

src/enapter/cli/http/api/rule_engine_get_command.py

@@ -0,0 +1,25 @@
+"""Rule Engine get command."""
+
+import argparse
+import json
+
+from enapter import cli, http
+
+
+class RuleEngineGetCommand(cli.Command):
+    """Command to get the rule engine state."""
+
+    @staticmethod
+    def register(parent: cli.Subparsers) -> None:
+        """Register the command in the subparsers."""
+        parser = parent.add_parser(
+            "get", formatter_class=argparse.ArgumentDefaultsHelpFormatter
+        )
+        parser.add_argument("-s", "--site-id", help="ID of the site to retrieve")
+
+    @staticmethod
+    async def run(args: argparse.Namespace) -> None:
+        """Run the command."""
+        async with http.api.Client(http.api.Config.from_env()) as client:
+            engine = await client.rule_engine.get(site_id=args.site_id)
+            print(json.dumps(engine.to_dto()))

src/enapter/cli/http/api/rule_engine_resume_command.py

@@ -0,0 +1,25 @@
+"""Rule Engine resume command."""
+
+import argparse
+import json
+
+from enapter import cli, http
+
+
+class RuleEngineResumeCommand(cli.Command):
+    """Command to resume the rule engine."""
+
+    @staticmethod
+    def register(parent: cli.Subparsers) -> None:
+        """Register the command in the subparsers."""
+        parser = parent.add_parser(
+            "resume", formatter_class=argparse.ArgumentDefaultsHelpFormatter
+        )
+        parser.add_argument("-s", "--site-id", help="ID of the site to resume")
+
+    @staticmethod
+    async def run(args: argparse.Namespace) -> None:
+        """Run the command."""
+        async with http.api.Client(http.api.Config.from_env()) as client:
+            engine = await client.rule_engine.resume(site_id=args.site_id)
+            print(json.dumps(engine.to_dto()))

src/enapter/cli/http/api/rule_engine_suspend_command.py

@@ -0,0 +1,25 @@
+"""Rule Engine suspend command."""
+
+import argparse
+import json
+
+from enapter import cli, http
+
+
+class RuleEngineSuspendCommand(cli.Command):
+    """Command to suspend the rule engine."""
+
+    @staticmethod
+    def register(parent: cli.Subparsers) -> None:
+        """Register the command in the subparsers."""
+        parser = parent.add_parser(
+            "suspend", formatter_class=argparse.ArgumentDefaultsHelpFormatter
+        )
+        parser.add_argument("-s", "--site-id", help="ID of the site to suspend")
+
+    @staticmethod
+    async def run(args: argparse.Namespace) -> None:
+        """Run the command."""
+        async with http.api.Client(http.api.Config.from_env()) as client:
+            engine = await client.rule_engine.suspend(site_id=args.site_id)
+            print(json.dumps(engine.to_dto()))