85 lines
2.7 KiB
Python
85 lines
2.7 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
"""
|
||
Скрипт подстановки переменных окружения из .env/env.example в файл шаблона.
|
||
|
||
Автор: Сергей Антропов
|
||
Сайт: https://devops.org.ru
|
||
|
||
Использование:
|
||
python3 app/scripts/substitute_env.py PATH_TO_ENV PATH_TO_COMPOSE_FILE
|
||
|
||
Логика:
|
||
- Читает пары KEY=VALUE из указанного env-файла (игнорирует комментарии и пустые строки)
|
||
- Не выполняет файл как shell (безопасно парсит)
|
||
- Заменяет плейсхолдеры вида ${KEY} в целевом файле
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
import io
|
||
import os
|
||
import re
|
||
import sys
|
||
from typing import Dict
|
||
|
||
|
||
def parse_env_file(env_path: str) -> Dict[str, str]:
|
||
"""Простой парсер .env файла: KEY=VALUE, без исполнения shell."""
|
||
variables: Dict[str, str] = {}
|
||
with io.open(env_path, "r", encoding="utf-8") as fh:
|
||
for raw in fh:
|
||
line = raw.strip()
|
||
if not line or line.startswith("#"):
|
||
continue
|
||
if "=" not in line:
|
||
continue
|
||
key, val = line.split("=", 1)
|
||
key = key.strip()
|
||
val = val.strip()
|
||
# Снимаем внешние кавычки, если присутствуют
|
||
if (val.startswith('"') and val.endswith('"')) or (
|
||
val.startswith("'") and val.endswith("'")
|
||
):
|
||
val = val[1:-1]
|
||
variables[key] = val
|
||
return variables
|
||
|
||
|
||
def substitute_placeholders(text: str, env: Dict[str, str]) -> str:
|
||
"""Заменяет ${VAR} из словаря env, оставляет плейсхолдер, если VAR не найден."""
|
||
pattern = re.compile(r"\$\{([A-Za-z_][A-Za-z0-9_]*)\}")
|
||
return pattern.sub(lambda m: env.get(m.group(1), m.group(0)), text)
|
||
|
||
|
||
def main() -> int:
|
||
if len(sys.argv) != 3:
|
||
sys.stderr.write(
|
||
"Usage: substitute_env.py <env_path> <compose_path>\n"
|
||
)
|
||
return 2
|
||
|
||
env_path, compose_path = sys.argv[1], sys.argv[2]
|
||
|
||
if not os.path.exists(env_path):
|
||
sys.stderr.write(f"Env file not found: {env_path}\n")
|
||
return 3
|
||
if not os.path.exists(compose_path):
|
||
sys.stderr.write(f"Compose file not found: {compose_path}\n")
|
||
return 4
|
||
|
||
env = parse_env_file(env_path)
|
||
|
||
with io.open(compose_path, "r", encoding="utf-8") as fh:
|
||
content = fh.read()
|
||
new_content = substitute_placeholders(content, env)
|
||
with io.open(compose_path, "w", encoding="utf-8") as fh:
|
||
fh.write(new_content)
|
||
return 0
|
||
|
||
|
||
if __name__ == "__main__":
|
||
raise SystemExit(main())
|
||
|
||
|