Adding some small changes

Author: leandrodamascena

aws_lambda_powertools/utilities/feature_flags/appconfig.py

@@ -2,6 +2,7 @@
 import traceback
 from typing import TYPE_CHECKING, Any, Dict, Optional, Union, cast
 
+import boto3
 from botocore.config import Config
 
 from aws_lambda_powertools.utilities import jmespath_utils
@@ -30,7 +31,9 @@ def __init__(
         envelope: Optional[str] = "",
         jmespath_options: Optional[Dict] = None,
         logger: Optional[Union[logging.Logger, Logger]] = None,
-        sdk_client: Optional["AppConfigDataClient"] = None,
+        boto_config: Optional[Config] = None,
+        boto3_session: Optional[boto3.session.Session] = None,
+        boto3_client: Optional["AppConfigDataClient"] = None,
     ):
         """This class fetches JSON schemas from AWS AppConfig
 
@@ -52,23 +55,28 @@ def __init__(
             Alternative JMESPath options to be included when filtering expr
         logger: A logging object
             Used to log messages. If None is supplied, one will be created.
-        sdk_client: Optional[AppConfigDataClient]
-            AppConfigData boto3 client, `sdk_config` will be ignored if this value is provided.
+        boto_config: botocore.config.Config, optional
+            Botocore configuration to pass during client initialization
+        boto3_session : boto3.Session, optional
+            Boto3 session to use for AWS API communication
+        boto3_client : AppConfigDataClient, optional
+            Boto3 AppConfigDataClient Client to use, boto3_session and boto_config will be ignored if both are provided
         """
         super().__init__()
         self.logger = logger or logging.getLogger(__name__)
         self.environment = environment
         self.application = application
         self.name = name
         self.cache_seconds = max_age
-        self.config = sdk_config
+        self.config = sdk_config or boto_config
         self.envelope = envelope
         self.jmespath_options = jmespath_options
         self._conf_store = AppConfigProvider(
             environment=environment,
             application=application,
-            config=sdk_config,
-            boto3_client=sdk_client,
+            config=sdk_config or boto_config,
+            boto3_client=boto3_client,
+            boto3_session=boto3_session,
         )
 
     @property

docs/utilities/feature_flags.md

@@ -503,10 +503,11 @@ These are the available options for further customization.
 | **name**             | `""`             | AWS AppConfig Configuration name, e.g `features`                                                                                                                      |
 | **envelope**         | `None`           | JMESPath expression to use to extract feature flags configuration from AWS AppConfig configuration                                                                    |
 | **max_age**          | `5`              | Number of seconds to cache feature flags configuration fetched from AWS AppConfig                                                                                     |
-| **sdk_config**       | `None`           | [Botocore Config object](https://botocore.amazonaws.com/v1/documentation/api/latest/reference/config.html){target="_blank"}                                           |
 | **jmespath_options** | `None`           | For advanced use cases when you want to bring your own [JMESPath functions](https://github.com/jmespath/jmespath.py#custom-functions){target="_blank" rel="nofollow"} |
 | **logger**           | `logging.Logger` | Logger to use for debug.  You can optionally supply an instance of Powertools for AWS Lambda (Python) Logger.                                                         |
-| **sdk_client**       | `None`           | [AppConfigData boto3 client](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/appconfigdata.html#AppConfigData.Client){target="_blank"}     |
+| **boto3_client**     | `None`           | [AppConfigData boto3 client](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/appconfigdata.html#AppConfigData.Client){target="_blank"}     |
+| **boto3_session**    | `None`           | [Boto3 session](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/core/session.html){target="_blank"}     |
+| **boto_config**     | `None`            | [Botocore config](https://botocore.amazonaws.com/v1/documentation/api/latest/reference/config.html){target="_blank"}     |
 
 === "appconfig_provider_options.py"
 
@@ -526,6 +527,27 @@ These are the available options for further customization.
     --8<-- "examples/feature_flags/src/appconfig_provider_options_features.json"
     ```
 
+#### Customizing boto configuration
+
+<!-- markdownlint-disable MD013 -->
+The **`boto_config`** , **`boto3_session`**, and **`boto3_client`**  parameters enable you to pass in a custom [botocore config object](https://botocore.amazonaws.com/v1/documentation/api/latest/reference/config.html){target="_blank"}, [boto3 session](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/core/session.html){target="_blank"}, or  a [boto3 client](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/core/boto3.html){target="_blank"} when constructing the AppConfig store provider.
+<!-- markdownlint-enable MD013 -->
+
+=== "custom_boto_session_feature_flags.py"
+    ```python hl_lines="8 14"
+    --8<-- "examples/feature_flags/src/custom_boto_session_feature_flags.py"
+    ```
+
+=== "custom_boto_config_feature_flags.py"
+    ```python hl_lines="8 14"
+    --8<-- "examples/feature_flags/src/custom_boto_config_feature_flags.py"
+    ```
+
+=== "custom_boto_client_feature_flags.py"
+    ```python hl_lines="8 14"
+    --8<-- "examples/feature_flags/src/custom_boto_client_feature_flags.py"
+    ```
+
 ### Create your own store provider
 
 You can create your own custom FeatureFlags store provider by inheriting the `StoreProvider` class, and implementing both `get_raw_configuration()` and `get_configuration()` methods to retrieve the configuration from your custom store.

examples/feature_flags/src/appconfig_provider_options.py

@@ -26,7 +26,7 @@ def _func_special_decoder(self, features):
     name="features",
     max_age=120,
     envelope="special_decoder(features)",  # using a custom function defined in CustomFunctions Class
-    sdk_config=boto_config,
+    boto_config=boto_config,
     jmespath_options=custom_jmespath_options,
 )
 

examples/feature_flags/src/custom_boto_client_feature_flags.py

@@ -0,0 +1,29 @@
+from typing import Any
+
+import boto3
+
+from aws_lambda_powertools.utilities.feature_flags import AppConfigStore, FeatureFlags
+from aws_lambda_powertools.utilities.typing import LambdaContext
+
+boto3_client = boto3.client("ssm")
+
+app_config = AppConfigStore(
+    environment="dev",
+    application="product-catalogue",
+    name="features",
+    boto3_client=boto3_client,
+)
+
+feature_flags = FeatureFlags(store=app_config)
+
+
+def lambda_handler(event: dict, context: LambdaContext):
+    apply_discount: Any = feature_flags.evaluate(name="ten_percent_off_campaign", default=False)
+
+    price: Any = event.get("price")
+
+    if apply_discount:
+        # apply 10% discount to product
+        price = price * 0.9
+
+    return {"price": price}

examples/feature_flags/src/custom_boto_config_feature_flags.py

@@ -0,0 +1,29 @@
+from typing import Any
+
+from botocore.config import Config
+
+from aws_lambda_powertools.utilities.feature_flags import AppConfigStore, FeatureFlags
+from aws_lambda_powertools.utilities.typing import LambdaContext
+
+boto_config = Config(read_timeout=10, retries={"total_max_attempts": 2})
+
+app_config = AppConfigStore(
+    environment="dev",
+    application="product-catalogue",
+    name="features",
+    boto_config=boto_config,
+)
+
+feature_flags = FeatureFlags(store=app_config)
+
+
+def lambda_handler(event: dict, context: LambdaContext):
+    apply_discount: Any = feature_flags.evaluate(name="ten_percent_off_campaign", default=False)
+
+    price: Any = event.get("price")
+
+    if apply_discount:
+        # apply 10% discount to product
+        price = price * 0.9
+
+    return {"price": price}

examples/feature_flags/src/custom_boto_session_feature_flags.py

@@ -0,0 +1,29 @@
+from typing import Any
+
+import boto3
+
+from aws_lambda_powertools.utilities.feature_flags import AppConfigStore, FeatureFlags
+from aws_lambda_powertools.utilities.typing import LambdaContext
+
+boto3_session = boto3.session.Session()
+
+app_config = AppConfigStore(
+    environment="dev",
+    application="product-catalogue",
+    name="features",
+    boto3_session=boto3_session,
+)
+
+feature_flags = FeatureFlags(store=app_config)
+
+
+def lambda_handler(event: dict, context: LambdaContext):
+    apply_discount: Any = feature_flags.evaluate(name="ten_percent_off_campaign", default=False)
+
+    price: Any = event.get("price")
+
+    if apply_discount:
+        # apply 10% discount to product
+        price = price * 0.9
+
+    return {"price": price}

tests/functional/feature_flags/_boto3/test_feature_flags.py

@@ -78,10 +78,10 @@ def init_feature_flags(
         application=application,
         name=name,
         max_age=600,
-        sdk_config=config,
         envelope=envelope,
         jmespath_options=jmespath_options,
-        sdk_client=client,
+        boto_config=config,
+        boto3_client=client,
     )
     feature_flags: FeatureFlags = FeatureFlags(store=app_conf_fetcher)
     return feature_flags