Spaces:
Running
Running
File size: 1,944 Bytes
5301c48 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 |
"""Environment variable loader utility.
This module provides functionality to load environment variables from a .env file
in non-production environments. In production, environment variables should be
set through the system/platform instead of using .env files for security reasons.
Uses python-dotenv for loading environment variables from .env files.
"""
import os
from typing import Optional
# Import python-dotenv
from dotenv import dotenv_values
from dotenv import find_dotenv as dotenv_find_dotenv
from dotenv import load_dotenv as dotenv_load_dotenv
from starfish.common.logger import get_logger
logger = get_logger(__name__)
def load_env_file(env_path: Optional[str] = None, override: bool = False) -> bool:
"""Load environment variables from .env file for non-production environments.
Args:
env_path: Path to the .env file. If None, looks for .env file in the current
working directory and parent directories.
override: Whether to override existing environment variables. Default is False.
Returns:
True if environment variables were loaded, False otherwise.
"""
# Skip loading in production environments
if os.getenv("ENV") == "PROD":
logger.info("Production environment detected. Skipping .env file loading.")
# Find the .env file if path not provided
if env_path is None:
env_path = dotenv_find_dotenv(usecwd=True)
if not env_path:
logger.warning("No .env file found in the current or parent directories.")
# Load environment variables
loaded = dotenv_load_dotenv(dotenv_path=env_path, override=override)
if loaded:
# Get the loaded variables to count and log them
loaded_vars = dotenv_values(env_path)
logger.debug(f"Loaded {len(loaded_vars)} environment variables from {env_path}")
else:
logger.warning(f"Failed to load environment variables from {env_path}")
|