http: api: devices: add assign blueprint

Author: rnovatorov

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

@@ -0,0 +1,28 @@
+import argparse
+import json
+
+from enapter import cli, http
+
+
+class DeviceAssignBlueprintCommand(cli.Command):
+
+    @staticmethod
+    def register(parent: cli.Subparsers) -> None:
+        parser = parent.add_parser(
+            "assign-blueprint", formatter_class=argparse.ArgumentDefaultsHelpFormatter
+        )
+        parser.add_argument("id", type=str, help="ID or slug of the device to update")
+        parser.add_argument(
+            "-b",
+            "--blueprint-id",
+            help="Blueprint ID to assign to the device",
+            required=True,
+        )
+
+    @staticmethod
+    async def run(args: argparse.Namespace) -> None:
+        async with http.api.Client(http.api.Config.from_env()) as client:
+            device = await client.devices.assign_blueprint(
+                device_id=args.id, blueprint_id=args.blueprint_id
+            )
+            print(json.dumps(device.to_dto()))

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

@@ -2,6 +2,7 @@
 
 from enapter import cli
 
+from .device_assign_blueprint_command import DeviceAssignBlueprintCommand
 from .device_create_standalone_command import DeviceCreateStandaloneCommand
 from .device_delete_command import DeviceDeleteCommand
 from .device_generate_communication_config_command import (
@@ -23,6 +24,7 @@ def register(parent: cli.Subparsers) -> None:
             dest="http_api_device_command", required=True
         )
         for command in [
+            DeviceAssignBlueprintCommand,
             DeviceCreateStandaloneCommand,
             DeviceDeleteCommand,
             DeviceGenerateCommunicationConfigCommand,
@@ -35,6 +37,8 @@ def register(parent: cli.Subparsers) -> None:
     @staticmethod
     async def run(args: argparse.Namespace) -> None:
         match args.http_api_device_command:
+            case "assign-blueprint":
+                await DeviceAssignBlueprintCommand.run(args)
             case "create-standalone":
                 await DeviceCreateStandaloneCommand.run(args)
             case "delete":

src/enapter/http/api/devices/client.py

@@ -59,6 +59,12 @@ async def delete(self, device_id: str) -> None:
         response = await self._client.delete(url)
         api.check_error(response)
 
+    async def assign_blueprint(self, device_id: str, blueprint_id: str) -> Device:
+        url = f"v3/devices/{device_id}/assign_blueprint"
+        response = await self._client.post(url, json={"blueprint_id": blueprint_id})
+        api.check_error(response)
+        return Device.from_dto(response.json()["device"])
+
     async def generate_communication_config(
         self, device_id: str, mqtt_protocol: MQTTProtocol
     ) -> CommunicationConfig: