pyyaml_env_tag_inject_environment_variables_with_defaults.py

1import os
2import yaml
3from yaml_env_tag import construct_env_tag
4
5# Register the !env tag constructor
6yaml.SafeLoader.add_constructor('!env', construct_env_tag)
7
8# Set an environment variable for the example
9os.environ['DATABASE_URL'] = 'postgresql://user:pass@localhost/db'
10
11# YAML content using the !env tag
12config_data = """
13database:
14  url: !env DATABASE_URL
15  timeout: !env [DB_TIMEOUT, 30]
16"""
17
18# Load the configuration
19config = yaml.safe_load(config_data)
20
21print(config)
22# Output: {'database': {'url': 'postgresql://user:pass@localhost/db', 'timeout': 30}}