Spaces:
Sleeping
Sleeping
| # parameter_utils.py | |
| import os | |
| import boto3 | |
| ssm = boto3.client('ssm', region_name='ap-southeast-3') | |
| def init_secret(path): | |
| next_token = None | |
| while True: | |
| response = ssm.get_parameters_by_path( | |
| Path=path, | |
| Recursive=True, | |
| WithDecryption=True, | |
| NextToken=next_token if next_token else "", | |
| ) | |
| parameters = response.get("Parameters", []) | |
| for parameter in parameters: | |
| key = parameter.get("Name").split("/")[-1] | |
| value = parameter.get("Value") | |
| os.environ[key] = value | |
| print(f"Set env variable {key}") | |
| next_token = response.get("NextToken") | |
| if not next_token: | |
| break | |