berryworld 1.0.0.192942__py3-none-any.whl → 1.0.0.193968__py3-none-any.whl

berryworld/generate_env.py

@@ -1,10 +1,92 @@
-from .handy_mix import HandyMix
+import ast
 import yaml
+from .handy_mix import HandyMix
+from azure.identity import ClientSecretCredential
+from azure.keyvault.secrets import SecretClient
 
 
 class EnvVariables:
     """ Generate the environmental variables to be used by kubernetes """
 
+    def extract_akv_config_from_tf(self, config_path):
+        """ Extract the AKV configuration from the Terraform file
+        :param config_path: Path to the Terraform file
+        """
+        with open(config_path, "r") as f:
+            lines = f.readlines()
+
+        start, end = None, None
+        for idx, line in enumerate(lines):
+            if "akv_config" in line and "=" in line and "{" in line:
+                start = idx + 1
+            if start and "}" in line:
+                end = idx
+                break
+
+        if start is None or end is None:
+            raise ValueError("akv_config block not found in the tf file")
+
+        kv_lines = lines[start:end]
+        config = {}
+        for line in kv_lines:
+            if "=" in line:
+                key, value = line.strip().split("=", 1)
+                key = key.strip()
+                value = value.strip()
+
+                # Use ast.literal_eval to safely evaluate lists or strings
+                try:
+                    parsed_value = ast.literal_eval(value)
+                except:
+                    # fallback if parsing fails
+                    parsed_value = value.strip('"')
+
+                config[key] = parsed_value
+
+        return config
+
+    def fetch_secrets_by_tag(self, config):
+        """ Fetch secrets from Azure Key Vault based on tags
+        :param config: Dictionary containing Azure Key Vault configuration
+        """
+        tenant_id = config["AZURE-TENANT-ID"]
+        client_id = config["AZURE-CLIENT-ID"]
+        client_secret = config["AZURE-CLIENT-SECRET"]
+        vault_url = f"https://{config['AZURE-KEY-VAULT-NAME']}.vault.azure.net/"
+        project_tags = config["AZURE-KEY-VAULT-PROJECT-TAGS"]
+
+        credential = ClientSecretCredential(tenant_id, client_id, client_secret)
+        client = SecretClient(vault_url=vault_url, credential=credential)
+
+        secrets = {}
+        for prop in client.list_properties_of_secrets():
+            if prop.tags:
+                if any("project" in key and value in project_tags for key, value in prop.tags.items()):
+                    secret = client.get_secret(prop.name)
+                    key = str(prop.name)
+                    secrets[key] = secret.value
+        return secrets
+
+    def generate_akv_env_file(self, config_path=None, env_path=None):
+        """ Generate the environment variables file from Azure Key Vault secrets
+        :param config_path: Path to the Terraform configuration file (optional)
+        :param env_path: Path to the destination .env file (optional)
+        """
+        config = self.extract_akv_config_from_tf(config_path)
+        secrets = self.fetch_secrets_by_tag(config)
+
+        # Create .env file for UnitTest
+        with open(env_path, "w") as f:
+            for key, value in secrets.items():
+                f.write(f"{key}={value}\n")
+
+    def generate_env_tf(self, config_path='./environments/prod/step01.tf', env_path='.env'):
+        """ Main function to parse arguments and generate AKV environment variables file.
+        :param config_path: Path to the Terraform configuration file
+        :param env_path: Path to the destination .env file
+        """
+        self.generate_akv_env_file(config_path=config_path, env_path=env_path)
+
     @staticmethod
     def generate_env_file(yaml_path='./environments/prd1/step01.yaml', env_path='./.env'):
         """ Save the data from the yaml file into a .env file to be used by decouple

{berryworld-1.0.0.192942.dist-info → berryworld-1.0.0.193968.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: berryworld
-Version: 1.0.0.192942
+Version: 1.0.0.193968
 Summary: Handy classes to improve ETL processes
 Home-page: https://www.berryworld.com
 Author: BerryWorld ltd
@@ -11,6 +11,8 @@ Classifier: Operating System :: OS Independent
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown
 License-File: LICENSE
+Requires-Dist: azure-identity>=1.15.0
+Requires-Dist: azure-keyvault-secrets>=4.8.0
 Requires-Dist: cryptography>=3.4.8
 Requires-Dist: msal>=1.14.0
 Requires-Dist: numpy>=1.25.2

{berryworld-1.0.0.192942.dist-info → berryworld-1.0.0.193968.dist-info}/RECORD

@@ -8,7 +8,7 @@ berryworld/credentials.py,sha256=jJENjLE-qxR0yBtZPMPEOJ6H83bZeo4hI265AerwcUI,954
 berryworld/devops.py,sha256=7mRbyZPGzk5ox-6J2etv3NxCcfP4cprG0_Xelh7prn8,9776
 berryworld/email_con.py,sha256=uSBzs_Ijz9pUPWt9e7U3TCB7i6q7hU1bB5vhsTB_tmw,14448
 berryworld/email_logging.py,sha256=LeSrTExhQhar49gJR2wGC1dS0lqsNpFl9pS3eYWqnuo,4936
-berryworld/generate_env.py,sha256=Tk9Z_u7cA4Ve8YYTyLH2qwmLVAuYoTIWoFc0h8Va8lY,7842
+berryworld/generate_env.py,sha256=B-d4Tu5M69j4CFfQjby3kNXxndIJhb4O9H-v94L6iwI,11204
 berryworld/handy_mix.py,sha256=ZPr5Z4l5TfyXQjhuLjttGi90whBooTQhB7M-Lq5L1DQ,10683
 berryworld/logging.py,sha256=tOxzaFALQm3aVVEc3I7t8MU6PwgUI9VSnzNCH9yRryo,1013
 berryworld/logic_apps.py,sha256=a0uU4tNO3v2w7grdBv-OOx4hUf7VBIerJpwZ9U-29dQ,14591
@@ -26,12 +26,12 @@ berryworld/verify_keys.py,sha256=X4Nuz3o0XbRDYofbJGvxIDeN5gfWj19PN7lhO6T3hR8,435
 berryworld/vivantio.py,sha256=QfZo0UKqkzVRg_LyiwivNd3aEup4TH57x4KxLZkCJwc,10627
 berryworld/vivantio_logging.py,sha256=ciy7gA4u3FrgUIpEBnMgocbNPp6jcu9TPoy-kLcrTZU,5736
 berryworld/xml_parser.py,sha256=HWD71NaTN3DaIOGT6Wzxs4CEsroFhGQwe9iPLIL80Co,957
-berryworld-1.0.0.192942.dist-info/licenses/LICENSE,sha256=vtkVCJM6E2af2gnsi2XxKPr4WY-uIbvzVLXieFND0UU,1074
+berryworld-1.0.0.193968.dist-info/licenses/LICENSE,sha256=vtkVCJM6E2af2gnsi2XxKPr4WY-uIbvzVLXieFND0UU,1074
 tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/test_allocation_config.py,sha256=e12l6fE9U57eSPS35g6ekJ_hol7-RHg89JV60_m1BlE,4633
 tests/test_handy_mix_config.py,sha256=Un56mz9KJmdn4K4OwzHAHLSRzDU1Xv2nFrONNuzOG04,2594
 tests/test_xml_parser.py,sha256=3QTlhFEd6KbK6nRFKZnc35tad6wqukTbe4QrFi8mr_8,859
-berryworld-1.0.0.192942.dist-info/METADATA,sha256=LurenorkZUosMBxMLVh2-RfDnb2ISt1QBCAKmi3KRWY,1362
-berryworld-1.0.0.192942.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-berryworld-1.0.0.192942.dist-info/top_level.txt,sha256=GIZ5qy-P5oxfEH755vA1IMFeTVdX3-40JxMe6nOe5I8,17
-berryworld-1.0.0.192942.dist-info/RECORD,,
+berryworld-1.0.0.193968.dist-info/METADATA,sha256=wwolMjQBxtyUn1NAmEF6jipmUumebMCqM2kwBfqNg_I,1445
+berryworld-1.0.0.193968.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+berryworld-1.0.0.193968.dist-info/top_level.txt,sha256=GIZ5qy-P5oxfEH755vA1IMFeTVdX3-40JxMe6nOe5I8,17
+berryworld-1.0.0.193968.dist-info/RECORD,,