updating some stuff

Author: stephenbawks

Diff for: aws_lambda_powertools/utilities/iam/__init__.py

@@ -1,6 +1,6 @@
 """Advanced feature flags utility"""
-from .auth import SigV4AuthFactory
+from .aws_auth import AwsSignedRequest
 
 __all__ = [
-    "SigV4AuthFactory",
+    "AwsSignedRequest",
 ]

Diff for: aws_lambda_powertools/utilities/iam/auth.py renamed to aws_lambda_powertools/utilities/iam/aws_auth.py

@@ -1,13 +1,14 @@
 
 from typing import Optional
 from botocore.auth import SigV4Auth
+from botocore.awsrequest import AWSRequest
 from botocore.credentials import Credentials
 import botocore.session
 
 
-class SigV4AuthFactory:
+class AwsSignedRequest:
     """
-    SigV4 authentication utility
+    Authenticating Requests (AWS Signature Version 4)
 
     Args:
         region (str): AWS region
@@ -24,28 +25,47 @@ class SigV4AuthFactory:
     **Using default credentials**
     >>> from aws_lambda_powertools.utilities.iam import SigV4AuthFactory
     >>> auth = SigV4AuthFactory(region="us-east-2", service="vpc-lattice-svcs")
+    """
 
 
-
-    """
     def __init__(
         self,
-        region: str,
         service: str,
+        method: str,
+        url: str,
+        data: Optional[str],
+        params: Optional[str],
+        headers: Optional[str],
         access_key: Optional[str],
         secret_key: Optional[str],
         token: Optional[str],
+        region: Optional[str],
+        sign_payload: Optional[bool] = False,
     ):
-        self._region = region
+
         self._service = service
+        self._method = method
+        self._url = url
+        self._data = data
+        self._params = params
+        self._headers = headers
+
+        if not region:
+            self._region = botocore.session.Session().get_config_variable("region")
+        else:
+            self._region = region
 
         if access_key and secret_key or token:
             self._access_key = access_key
             self._secret_key = secret_key
             self._credentials = Credentials(access_key=self._access_key, secret_key=self._secret_key, token=token)
-
         else:
             self._credentials = botocore.session.Session().get_credentials()
 
         def __call__(self):
-            return SigV4Auth(credentials=self._credentials, service=self._service, region=self._region)
+            request = AWSRequest(method=self._method, url=self._url, data=self._data, params=self._params, headers=self._headers)
+            if sign_payload is False:
+                request.context["payload_signing_enabled"] = False
+
+            signed_request = SigV4Auth(credentials=self._credentials, service_name=self._service, region_name=self._region).add_auth(request)
+            return signed_request.prepare()