Add better pricing typings

Pricing functions can be better typed with types-boto3

Author: adanaja

pyproject.toml

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

src/e3/aws/__init__.py

@@ -39,6 +39,7 @@
     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
@@ -321,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 = ...
@@ -339,6 +345,7 @@ def client(
         | 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.