Auto commit from main repo: manual-sync

Author: GitHub Actions

__init__.py

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

armis_sdk/clients/__init__.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/assets_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.
 
@@ -71,15 +71,12 @@ async def main():
             async with client.stream("GET", collector_image.url) as response:
                 response.raise_for_status()
                 total_size = int(response.headers.get("Content-Length", "0"))
-                # pylint: disable-next=contextmanager-generator-missing-cleanup
                 with self.open_file(destination) as file:
                     async for chunk in response.aiter_bytes():
                         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 +105,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/clients/collectors_client.py

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

armis_sdk/core/__init__.py

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

armis_sdk/entities/__init__.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/collector_image.py

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

armis_sdk/entities/data_export/__init__.py

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