Auto commit from main repo: manual-sync (#41)

Author: read-write-content[bot]

armis_sdk/clients/__init__.py

@@ -0,0 +1 @@
+# Required by mkdocs

armis_sdk/clients/assets_client.py

@@ -1,6 +1,5 @@
 import datetime
 from typing import AsyncIterator
-from typing import Literal
 from typing import Optional
 from typing import Type
 from typing import Union
@@ -15,14 +14,7 @@
 from armis_sdk.entities.asset import Asset
 from armis_sdk.entities.asset import AssetT
 from armis_sdk.entities.device import Device
-
-AssetIdSource = Literal[
-    "ASSET_ID",
-    "IPV4_ADDRESS",
-    "IPV6_ADDRESS",
-    "MAC_ADDRESS",
-    "SERIAL_NUMBER",
-]
+from armis_sdk.types.asset_id_source import AssetIdSource
 
 
 @universalasync.wrap

armis_sdk/clients/collectors_client.py

@@ -11,7 +11,7 @@
 from armis_sdk.core.base_entity_client import BaseEntityClient
 from armis_sdk.entities.collector_image import CollectorImage
 from armis_sdk.entities.download_progress import DownloadProgress
-from armis_sdk.enums.collector_image_type import CollectorImageType
+from armis_sdk.types.collector_image_type import CollectorImageType
 
 
 @universalasync.wrap
@@ -26,7 +26,7 @@ class CollectorsClient(BaseEntityClient):
     async def download_image(
         self,
         destination: Union[str, IO[bytes]],
-        image_type: CollectorImageType = CollectorImageType.OVA,
+        image_type: CollectorImageType = "OVA",
     ) -> AsyncIterator[DownloadProgress]:
         """Download a collector image to a specified destination path / file.
 
@@ -77,9 +77,7 @@ async def main():
                         file.write(chunk)
                         yield DownloadProgress(downloaded=file.tell(), total=total_size)
 
-    async def get_image(
-        self, image_type: CollectorImageType = CollectorImageType.OVA
-    ) -> CollectorImage:
+    async def get_image(self, image_type: CollectorImageType = "OVA") -> CollectorImage:
         """Get collector image information including download URL and credentials.
 
         Args:
@@ -108,7 +106,7 @@ async def main():
         """
         async with self._armis_client.client() as client:
             response = await client.get(
-                "/v3/collectors/_image", params={"image_type": image_type.value}
+                "/v3/collectors/_image", params={"image_type": image_type}
             )
             data = response_utils.get_data_dict(response)
             return CollectorImage.model_validate(data)

armis_sdk/core/__init__.py

@@ -0,0 +1 @@
+# Required by mkdocs

armis_sdk/entities/__init__.py

@@ -0,0 +1 @@
+# Required by mkdocs

armis_sdk/entities/collector_image.py

@@ -3,15 +3,15 @@
 from pydantic import Field
 
 from armis_sdk.core.base_entity import BaseEntity
-from armis_sdk.enums.collector_image_type import CollectorImageType
+from armis_sdk.types.collector_image_type import CollectorImageType
 
 
 class CollectorImage(BaseEntity):
     """
     An entity that represents the details required to download and run a collector image.
     """
 
-    image_type: CollectorImageType = Field(strict=False)
+    image_type: CollectorImageType
     """The type of the image."""
 
     image_password: str

armis_sdk/entities/data_export/__init__.py

@@ -0,0 +1 @@
+# Required by mkdocs

armis_sdk/enums/collector_image_type.py

@@ -0,0 +1 @@
+# Required by mkdocs

armis_sdk/types/__init__.py

@@ -0,0 +1,9 @@
+from typing import Literal
+
+AssetIdSource = Literal[
+    "ASSET_ID",
+    "IPV4_ADDRESS",
+    "IPV6_ADDRESS",
+    "MAC_ADDRESS",
+    "SERIAL_NUMBER",
+]