Add SES operations to the SDK (#7)

Author: giograno

.github/workflows/test.yml

@@ -10,6 +10,7 @@ on:
   push:
     branches:
       - main
+  workflow_dispatch:
 
 jobs:
 

bin/generate.sh

@@ -3,7 +3,7 @@
 version=$(cat VERSION)
 
 docker run --rm -v "${PWD}:/local" openapitools/openapi-generator-cli:v7.9.0 generate \
-    -i /local/openapi.yaml \
+    -i https://raw.github.com/localstack/openapi/refs/heads/main/openapi/emulators/localstack-spec-latest.yml \
     --skip-validate-spec \
     -g python \
     -o /local//packages/localstack-sdk-generated \

localstack-sdk-python/localstack/sdk/aws/client.py

@@ -2,7 +2,7 @@
 
 from localstack.clients import BaseClient
 from localstack.sdk.api.aws_api import AwsApi
-from localstack.sdk.models import Message
+from localstack.sdk.models import Message, SesSentEmail
 
 
 def _from_sqs_query_to_json(xml_dict: dict) -> list[Message]:
@@ -35,6 +35,10 @@ 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
@@ -45,6 +49,19 @@ 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"""