localstack-sdk-python 0.0.2__py3-none-any.whl → 0.0.3__py3-none-any.whl

localstack/sdk/aws/__init__.py

@@ -0,0 +1,3 @@
+from localstack.sdk.aws.client import AWSClient
+
+__all__ = ["AWSClient"]

localstack/sdk/aws/client.py

@@ -0,0 +1,68 @@
+import json
+
+from localstack.clients import BaseClient
+from localstack.sdk.api.aws_api import AwsApi
+from localstack.sdk.models import Message, SesSentEmail
+
+
+def _from_sqs_query_to_json(xml_dict: dict) -> list[Message]:
+    """
+    todo: developer endpoint implements sqs-query protocol. Remove this workaround one we move them to json.
+    """
+    raw_messages = (
+        xml_dict.get("ReceiveMessageResponse", {}).get("ReceiveMessageResult", {}) or {}
+    ).get("Message", [])
+    if isinstance(raw_messages, dict):
+        raw_messages = [raw_messages]
+    messages = []
+    for msg in raw_messages:
+        _attributes = msg.get("Attribute", [])
+        attributes = {i["Name"]: i["Value"] for i in _attributes}
+        _m = {
+            "MessageId": msg.get("MessageId"),
+            "ReceiptHandle": msg.get("ReceiptHandle"),
+            "MD5OfBody": msg.get("MD5OfBody"),
+            "Body": msg.get("Body"),
+            "Attributes": attributes,
+        }
+        m = Message.from_dict(_m)
+        messages.append(m)
+    return messages
+
+
+class AWSClient(BaseClient):
+    def __init__(self, **kwargs) -> None:
+        super().__init__(**kwargs)
+        self._client = AwsApi(self._api_client)
+
+    ########
+    # SQS
+    ########
+
+    def list_sqs_messages(self, account_id: str, region: str, queue_name: str) -> list[Message]:
+        response = self._client.list_sqs_messages_with_http_info(
+            account_id=account_id, region=region, queue_name=queue_name
+        )
+        return _from_sqs_query_to_json(json.loads(response.raw_data))
+
+    def list_sqs_messages_from_queue_url(self, queue_url) -> list[Message]:
+        response = self._client.list_all_sqs_messages_with_http_info(queue_url=queue_url)
+        return _from_sqs_query_to_json(json.loads(response.raw_data))
+
+    ########
+    # SES
+    ########
+
+    def get_ses_messages(
+        self, id_filter: str | None = None, email_filter: str | None = None
+    ) -> list[SesSentEmail]:
+        response = self._client.get_ses_messages(id=id_filter, email=email_filter)
+        return response.messages
+
+    def discard_ses_messages(self, id_filter: str | None = None) -> None:
+        return self._client.discard_ses_messages(id=id_filter)
+
+
+def get_default(**args) -> AwsApi:
+    """Return a client with a default configuration"""
+    return AwsApi(args)

localstack/sdk/state/__init__.py

@@ -0,0 +1,3 @@
+from localstack.sdk.state.client import StateClient
+
+__all__ = ["StateClient"]

localstack/sdk/state/client.py

@@ -0,0 +1,11 @@
+from localstack.clients import BaseClient
+from localstack.sdk.api import StateApi
+
+
+class StateClient(BaseClient):
+    def __init__(self, **args) -> None:
+        super().__init__(**args)
+        self._client = StateApi(self._api_client)
+
+    def reset_state(self) -> None:
+        self._client.localstack_state_reset_post_0()

{localstack_sdk_python-0.0.2.dist-info → localstack_sdk_python-0.0.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: localstack-sdk-python
-Version: 0.0.2
+Version: 0.0.3
 Summary: Python SDK for LocalStack
 Author-email: LocalStack Team <info@localstack.cloud>
 Project-URL: Homepage, https://localstack.cloud

localstack_sdk_python-0.0.3.dist-info/RECORD

@@ -0,0 +1,15 @@
+localstack/clients.py,sha256=UFBz23vDc48uGl1ton1-ejjQb-nnHnj0YEGVBelz57E,680
+localstack/sdk/aws/__init__.py,sha256=Das2FPp0t5hl_gxEP8Nw6MigVxBU8HxePF5ZKq_YNkM,73
+localstack/sdk/aws/client.py,sha256=TGemZuxjbojlq-kWEJNoVBGGg3a3Jshk2XF-PXd97RE,2333
+localstack/sdk/chaos/__init__.py,sha256=czMV7U_iDgUPBObl6nmWQxl78cF-zVl1c8QPY5tnQg4,79
+localstack/sdk/chaos/client.py,sha256=l5hJ8NNLNv2FaSp8lol8fOXZCcAL7DRDV2X_wFGYwxc,1506
+localstack/sdk/chaos/managers.py,sha256=v9oKvotSmz3npEadYtrYBumCuISdxPeol_YGYYQQlGw,386
+localstack/sdk/pods/__init__.py,sha256=49ilbKnbGpIRcv0w71nEp4RvDFtkxfrvrUVdrSLFltI,76
+localstack/sdk/pods/client.py,sha256=YzHQBLbEaDb-veNS2NwDZrND12kFSAVoy5cm-hmS4-M,2909
+localstack/sdk/state/__init__.py,sha256=gHCAz6H0FLwsg72361MbVpdL-YsuTH8GdaTmhKUgkL4,79
+localstack/sdk/state/client.py,sha256=BISI3O-lM0hRIKJVGOPstlPcO6G9uFfYHp5vPOE6ztw,327
+localstack_sdk_python-0.0.3.dist-info/LICENSE.txt,sha256=3PC-9Z69UsNARuQ980gNR_JsLx8uvMjdG6C7cc4LBYs,606
+localstack_sdk_python-0.0.3.dist-info/METADATA,sha256=cZSAkz6TRLGu1R0_IdrtNI-SdXxkDxPfhUA9WXcFZMA,1740
+localstack_sdk_python-0.0.3.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
+localstack_sdk_python-0.0.3.dist-info/top_level.txt,sha256=3sqmK2lGac8nCy8nwsbS5SpIY_izmtWtgaTFKHYVHbI,11
+localstack_sdk_python-0.0.3.dist-info/RECORD,,

{localstack_sdk_python-0.0.2.dist-info → localstack_sdk_python-0.0.3.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.1.0)
+Generator: setuptools (75.3.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

localstack_sdk_python-0.0.2.dist-info/RECORD

@@ -1,11 +0,0 @@
-localstack/clients.py,sha256=UFBz23vDc48uGl1ton1-ejjQb-nnHnj0YEGVBelz57E,680
-localstack/sdk/chaos/__init__.py,sha256=czMV7U_iDgUPBObl6nmWQxl78cF-zVl1c8QPY5tnQg4,79
-localstack/sdk/chaos/client.py,sha256=l5hJ8NNLNv2FaSp8lol8fOXZCcAL7DRDV2X_wFGYwxc,1506
-localstack/sdk/chaos/managers.py,sha256=v9oKvotSmz3npEadYtrYBumCuISdxPeol_YGYYQQlGw,386
-localstack/sdk/pods/__init__.py,sha256=49ilbKnbGpIRcv0w71nEp4RvDFtkxfrvrUVdrSLFltI,76
-localstack/sdk/pods/client.py,sha256=YzHQBLbEaDb-veNS2NwDZrND12kFSAVoy5cm-hmS4-M,2909
-localstack_sdk_python-0.0.2.dist-info/LICENSE.txt,sha256=3PC-9Z69UsNARuQ980gNR_JsLx8uvMjdG6C7cc4LBYs,606
-localstack_sdk_python-0.0.2.dist-info/METADATA,sha256=3QTB1Pl1CrBrsbW42Ep2oDywzCAJ93C2khujATfCllg,1740
-localstack_sdk_python-0.0.2.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-localstack_sdk_python-0.0.2.dist-info/top_level.txt,sha256=3sqmK2lGac8nCy8nwsbS5SpIY_izmtWtgaTFKHYVHbI,11
-localstack_sdk_python-0.0.2.dist-info/RECORD,,