reach_commons 0.18.2__tar.gz → 0.18.3__tar.gz

{reach_commons-0.18.2 → reach_commons-0.18.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: reach_commons
-Version: 0.18.2
+Version: 0.18.3
 Summary: Reach Commons is a versatile utility library designed to streamline and enhance development workflows within the Reach ecosystem.
 License: MIT
 Author: Wilson Moraes

{reach_commons-0.18.2 → reach_commons-0.18.3}/pyproject.toml

@@ -1,7 +1,7 @@
 # isort .; black .; poetry build; poetry publish
 [tool.poetry]
 name = "reach_commons"
-version = "0.18.2"
+version = "0.18.3"
 description = "Reach Commons is a versatile utility library designed to streamline and enhance development workflows within the Reach ecosystem."
 authors = ["Wilson Moraes <wmoraes@getreach.ai>"]
 license = "MIT"

{reach_commons-0.18.2 → reach_commons-0.18.3}/reach_commons/reach_aws/kms.py

@@ -17,10 +17,12 @@ class BaseKMSClient:
         logger=get_reach_logger(),
         region_name="us-east-1",
         profile_name=None,
+        enviroment=None,
     ):
         self.region_name = region_name
         self.profile_name = profile_name
         self.logger = logger
+        self.enviroment = enviroment or os.environ.get("ENV", "staging")
 
     def handle_exception(self, exc, secret, ciphertext):
         error_msg = (
@@ -55,7 +57,7 @@ class KMSClient(BaseKMSClient):
             )
         )
 
-        key_alias = f"alias/{os.environ.get('ENV', 'staging')}-db-data".lower()
+        key_alias = f"alias/{self.enviroment}-db-data".lower()
 
         return self.encrypt(custom_context, data_to_encrypt, key_alias)