Merge branch 'morosi-feat' into 'master'

Add better secretsmanager|cloudfront|pricing typings

See merge request it/e3-aws!141

Author: adanaja

pyproject.toml

@@ -50,7 +50,7 @@ check = [
   "types-requests",
   "types-setuptools",
   "python-on-whales",
-  "types-boto3[essential,ec2,ecr,iam,lambda,sts]",
+  "types-boto3[essential,cloudfront,ec2,ecr,iam,lambda,pricing,secretsmanager,sts]",
 ]
 
 [tool.setuptools.dynamic]

src/e3/aws/__init__.py

@@ -33,12 +33,15 @@
     from types import TracebackType
 
     from types_boto3_cloudformation import CloudFormationClient
+    from types_boto3_cloudfront import CloudFrontClient
     from types_boto3_dynamodb import DynamoDBClient
     from types_boto3_ec2 import EC2Client
     from types_boto3_ecr import ECRClient
     from types_boto3_iam import IAMClient
     from types_boto3_lambda import LambdaClient
+    from types_boto3_pricing import PricingClient
     from types_boto3_s3 import S3Client
+    from types_boto3_secretsmanager import SecretsManagerClient
     from types_boto3_sts import STSClient
     from types_boto3_sts.type_defs import AssumeRoleRequestTypeDef
 
@@ -282,6 +285,11 @@ def to_boto3(self) -> boto3.Session:
     @overload
     def client(self, name: Literal["s3"], region: str | None = ...) -> S3Client: ...
 
+    @overload
+    def client(
+        self, name: Literal["secretsmanager"], region: str | None = ...
+    ) -> SecretsManagerClient: ...
+
     @overload
     def client(self, name: Literal["sts"], region: str | None = ...) -> STSClient: ...
 
@@ -290,6 +298,11 @@ def client(
         self, name: Literal["cloudformation"], region: str | None = ...
     ) -> CloudFormationClient: ...
 
+    @overload
+    def client(
+        self, name: Literal["cloudfront"], region: str | None = ...
+    ) -> CloudFrontClient: ...
+
     @overload
     def client(
         self, name: Literal["dynamodb"], region: str | None = ...
@@ -309,6 +322,11 @@ def client(
         self, name: Literal["lambda"], region: str | None = ...
     ) -> LambdaClient: ...
 
+    @overload
+    def client(
+        self, name: Literal["pricing"], region: str | None = ...
+    ) -> PricingClient: ...
+
     @overload
     def client(
         self, name: str, region: str | None = ...
@@ -318,13 +336,16 @@ def client(
         self, name: str, region: str | None = None
     ) -> (
         S3Client
+        | SecretsManagerClient
         | STSClient
         | CloudFormationClient
+        | CloudFrontClient
         | DynamoDBClient
         | EC2Client
         | ECRClient
         | IAMClient
         | LambdaClient
+        | PricingClient
         | botocore.client.BaseClient
     ):
         """Get a client.

src/e3/aws/pricing/__init__.py

@@ -9,7 +9,8 @@
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
-    import botocore
+    from types_boto3_pricing import PricingClient
+    from types_boto3_pricing.type_defs import FilterTypeDef
 
     from typing import Any
 
@@ -22,7 +23,7 @@
 class Pricing:
     """Pricing abstraction."""
 
-    def __init__(self, client: botocore.client.BaseClient) -> None:
+    def __init__(self, client: PricingClient) -> None:
         """Initialize Pricing.
 
         :param client: a client for the Pricing API
@@ -51,7 +52,7 @@ def ec2_price_information(
         instance_type: str | None = None,
         os: str | None = None,
         region: str | None = None,
-        filters: list[dict[str, Any]] | None = None,
+        filters: list[FilterTypeDef] | None = None,
     ) -> list[PriceInformation]:
         """Get pricing informations for EC2 instances.