feat(http-api): add device profile download support

Author: rnovatorov

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

@@ -4,6 +4,7 @@
 
 from .blueprint_download_command import BlueprintDownloadCommand
 from .blueprint_get_command import BlueprintGetCommand
+from .blueprint_profiles_command import BlueprintProfilesCommand
 from .blueprint_upload_command import BlueprintUploadCommand
 from .blueprint_validate_command import BlueprintValidateCommand
 
@@ -19,6 +20,7 @@ def register(parent: cli.Subparsers) -> None:
         for command in [
             BlueprintDownloadCommand,
             BlueprintGetCommand,
+            BlueprintProfilesCommand,
             BlueprintUploadCommand,
             BlueprintValidateCommand,
         ]:
@@ -31,9 +33,11 @@ async def run(args: argparse.Namespace) -> None:
                 await BlueprintDownloadCommand.run(args)
             case "get":
                 await BlueprintGetCommand.run(args)
+            case "profiles":
+                await BlueprintProfilesCommand.run(args)
             case "upload":
                 await BlueprintUploadCommand.run(args)
             case "validate":
                 await BlueprintValidateCommand.run(args)
             case _:
-                raise NotImplementedError(args.command_command)
+                raise NotImplementedError(args.blueprint_command)

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

@@ -0,0 +1,29 @@
+import argparse
+
+from enapter import cli
+
+from .blueprint_profiles_download_command import BlueprintProfilesDownloadCommand
+
+
+class BlueprintProfilesCommand(cli.Command):
+
+    @staticmethod
+    def register(parent: cli.Subparsers) -> None:
+        parser = parent.add_parser(
+            "profiles", formatter_class=argparse.ArgumentDefaultsHelpFormatter
+        )
+        subparsers = parser.add_subparsers(
+            dest="blueprint_profiles_command", required=True
+        )
+        for command in [
+            BlueprintProfilesDownloadCommand,
+        ]:
+            command.register(subparsers)
+
+    @staticmethod
+    async def run(args: argparse.Namespace) -> None:
+        match args.blueprint_profiles_command:
+            case "download":
+                await BlueprintProfilesDownloadCommand.run(args)
+            case _:
+                raise NotImplementedError(args.blueprint_profiles_command)

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

@@ -0,0 +1,31 @@
+import argparse
+import logging
+import pathlib
+
+from enapter import cli, http
+
+LOGGER = logging.getLogger(__name__)
+
+
+class BlueprintProfilesDownloadCommand(cli.Command):
+
+    @staticmethod
+    def register(parent: cli.Subparsers) -> None:
+        parser = parent.add_parser(
+            "download", formatter_class=argparse.ArgumentDefaultsHelpFormatter
+        )
+        parser.add_argument(
+            "-o",
+            "--output",
+            type=pathlib.Path,
+            help="Output directory path",
+            default=pathlib.Path.cwd(),
+        )
+
+    @staticmethod
+    async def run(args: argparse.Namespace) -> None:
+        async with http.api.Client(http.api.Config.from_env()) as client:
+            content = await client.blueprints.download_device_profiles()
+            output_path = args.output / "device_profiles.zip"
+            with output_path.open("wb") as f:
+                f.write(content)

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

@@ -43,6 +43,12 @@ async def download(
         await api.check_error(response)
         return response.content
 
+    async def download_device_profiles(self) -> bytes:
+        url = "v3/blueprints/device_profiles/download"
+        response = await self._client.get(url)
+        await api.check_error(response)
+        return response.content
+
     async def validate_file(self, path: pathlib.Path) -> None:
         with path.open("rb") as file:
             data = file.read()

tests/unit/test_http/test_api/test_blueprints/test_client.py

@@ -149,6 +149,20 @@ async def test_download_blueprint_compiled(blueprints_client, mock_client):
     )
 
 
+@pytest.mark.asyncio
+async def test_download_device_profiles(blueprints_client, mock_client):
+    """Test downloading device profiles."""
+    mock_response = MagicMock(spec=httpx.Response)
+    mock_response.status_code = 200
+    mock_response.content = b"device profiles zip content"
+    mock_client.get = AsyncMock(return_value=mock_response)
+
+    content = await blueprints_client.download_device_profiles()
+
+    assert content == b"device profiles zip content"
+    mock_client.get.assert_called_once_with("v3/blueprints/device_profiles/download")
+
+
 @pytest.mark.asyncio
 async def test_validate_blueprint(blueprints_client, mock_client):
     """Test validating a blueprint from bytes."""