PDM-892: adding HealthLake, STS and ECS client creation function

Author: adrianwalker5-nhs

nhs_aws_helpers/__init__.py

@@ -50,9 +50,11 @@
 from mypy_boto3_dynamodb.client import DynamoDBClient
 from mypy_boto3_dynamodb.service_resource import DynamoDBServiceResource, Table
 from mypy_boto3_dynamodb.type_defs import KeysAndAttributesTypeDef
+from mypy_boto3_ecs.client import ECSClient
 from mypy_boto3_events import EventBridgeClient
 from mypy_boto3_firehose import FirehoseClient
 from mypy_boto3_glue import GlueClient
+from mypy_boto3_healthlake.client import HealthLakeClient
 from mypy_boto3_kms.client import KMSClient
 from mypy_boto3_lambda.client import LambdaClient
 from mypy_boto3_logs.client import CloudWatchLogsClient
@@ -72,6 +74,7 @@
 from mypy_boto3_sqs.client import SQSClient
 from mypy_boto3_ssm.client import SSMClient
 from mypy_boto3_stepfunctions import SFNClient
+from mypy_boto3_sts.client import STSClient
 
 from nhs_aws_helpers.common import run_in_executor
 from nhs_aws_helpers.s3_object_writer import S3ObjectWriter
@@ -349,6 +352,20 @@ def events_client(session: Optional[Session] = None, config: Optional[Config] =
     return _aws("events", "client", session, config)  # type: ignore[no-any-return]
 
 
+def healthlake_client(session: Optional[Session] = None, config: Optional[Config] = None) -> HealthLakeClient:
+    return _aws("healthlake", "client", session, config)  # type: ignore[no-any-return]
+
+
+def sts_client(
+    session: Optional[Session] = None, config: Optional[Config] = None, endpoint_url: Optional[str] = None
+) -> STSClient:
+    return _aws("sts", "client", session, config)  # type: ignore[no-any-return]
+
+
+def ecs_client(session: Optional[Session] = None, config: Optional[Config] = None) -> ECSClient:
+    return _aws("ecs", "client", session, config)  # type: ignore[no-any-return]
+
+
 def s3_bucket(bucket: str, session: Optional[Session] = None, config: Optional[Config] = None) -> Bucket:
     return s3_resource(session=session, config=config).Bucket(bucket)
 

poetry.lock

@@ -15,7 +15,7 @@ repository = "https://github.com/NHSDigital/nhs-aws-helpers"
 # core dependencies
 python = ">=3.9,<4.0"
 boto3 = "^1.38.14"
-boto3-stubs = {extras = ["s3", "ssm", "secretsmanager", "dynamodb", "stepfunctions", "sqs", "lambda", "logs", "ses", "sns", "events", "kms", "firehose", "athena", "glue", "ce", "cloudwatch", "backup"], version = "^1.38.6"}
+boto3-stubs = {extras = ["s3", "ssm", "secretsmanager", "dynamodb", "stepfunctions", "sqs", "lambda", "logs", "ses", "sns", "events", "kms", "firehose", "athena", "glue", "ce", "cloudwatch", "backup", "healthlake", "sts", "ecs"], version = "^1.38.6"}
 botocore-stubs = "^1.38.46"