refactor(jmespath_utils): add from __future__ import annotations (aws-powertools#4962)

ericbn · leandrodamascena · web-flow · commit a57bbfd4ef60 · 2024-08-15T13:30:24.000+01:00
and update code according to ruff rules TCH, UP006, UP007, UP037 and
FA100.

Co-authored-by: Leandro Damascena &lt;lcdama@amazon.pt&gt;
diff --git a/aws_lambda_powertools/utilities/jmespath_utils/__init__.py b/aws_lambda_powertools/utilities/jmespath_utils/__init__.py
@@ -1,9 +1,11 @@
+from __future__ import annotations
+
 import base64
 import gzip
 import json
 import logging
 import warnings
-from typing import Any, Dict, Optional, Union
+from typing import Any
 
 import jmespath
 from jmespath.exceptions import LexerError
@@ -33,7 +35,7 @@ def _func_powertools_base64_gzip(self, value):
         return uncompressed.decode()
 
 
-def query(data: Union[Dict, str], envelope: str, jmespath_options: Optional[Dict] = None) -> Any:
+def query(data: dict | str, envelope: str, jmespath_options: dict | None = None) -> Any:
     """Searches and extracts data using JMESPath
 
     Envelope being the JMESPath expression to extract the data you're after
@@ -57,11 +59,11 @@ def handler(event: dict, context: LambdaContext):
 
     Parameters
     ----------
-    data : Dict
+    data : dict | str
         Data set to be filtered
     envelope : str
         JMESPath expression to filter data against
-    jmespath_options : Dict
+    jmespath_options : dict | None
         Alternative JMESPath options to be included when filtering expr
 
 
@@ -82,7 +84,7 @@ def handler(event: dict, context: LambdaContext):
 
 
 @deprecated("`extract_data_from_envelope` is deprecated; use `query` instead.", category=None)
-def extract_data_from_envelope(data: Union[Dict, str], envelope: str, jmespath_options: Optional[Dict] = None) -> Any:
+def extract_data_from_envelope(data: dict | str, envelope: str, jmespath_options: dict | None = None) -> Any:
     """Searches and extracts data using JMESPath
 
     *Deprecated*: Use query instead
