102 lines
4.1 KiB
Python
102 lines
4.1 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
"""
|
|
Генерация docker-compose-prod.yaml из шаблона и env.example.
|
|
|
|
Автор: Сергей Антропов
|
|
Сайт: https://devops.org.ru
|
|
|
|
Поведение:
|
|
- Читает env.example и подставляет значения в docker-compose-prod.tmpl.yaml
|
|
- Запрашивает (или берет из ENV/CLI) REGISTRY_HOST, IMAGE_NAME_FULL, IMAGE_TAG
|
|
- Заменяет плейсхолдеры REGISTRY_PLACEHOLDER/IMAGE_NAME_PLACEHOLDER/IMAGE_TAG_PLACEHOLDER
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
import io
|
|
import os
|
|
import re
|
|
from typing import Dict
|
|
|
|
|
|
ROOT = os.path.abspath(os.path.join(os.path.dirname(__file__), os.pardir))
|
|
|
|
|
|
def parse_env_file(env_path: str) -> Dict[str, str]:
|
|
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:
|
|
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:
|
|
parser = argparse.ArgumentParser(description="Генерация docker-compose-prod.yaml")
|
|
parser.add_argument("--template", default=os.path.join(ROOT, "release", "docker-compose-prod.tmpl.yaml"))
|
|
parser.add_argument("--output", default=os.path.join(ROOT, "docker-compose-prod.yaml"))
|
|
parser.add_argument("--env", dest="env_path", default=os.path.join(ROOT, "env.example"))
|
|
parser.add_argument("--registry", default=os.getenv("REGISTRY_HOST", ""))
|
|
parser.add_argument("--image", dest="image", default=os.getenv("IMAGE_NAME_FULL", ""))
|
|
parser.add_argument("--tag", dest="tag", default=os.getenv("IMAGE_TAG", ""))
|
|
args = parser.parse_args()
|
|
|
|
env = parse_env_file(args.env_path)
|
|
# Подстановка ${VAR} из env.example
|
|
with io.open(args.template, "r", encoding="utf-8") as fh:
|
|
content = fh.read()
|
|
content = substitute_placeholders(content, env)
|
|
|
|
# Комментирование строк, где остались неразрешенные ${VAR}
|
|
# Ищем строки формата 'KEY: "${VAR}"' или 'KEY: ${VAR}' и комментируем их
|
|
lines = content.splitlines()
|
|
commented: list[str] = []
|
|
unresolved_pattern = re.compile(r"^([ \t-]*[^:#\n]+:\s*)(\"?\$\{[A-Za-z_][A-Za-z0-9_]*\}\"?)\s*$")
|
|
for line in lines:
|
|
if unresolved_pattern.match(line):
|
|
commented.append("# " + line)
|
|
else:
|
|
commented.append(line)
|
|
content = "\n".join(commented) + ("\n" if content.endswith("\n") else "")
|
|
|
|
# Плейсхолдеры реестра/имени/тега
|
|
registry = args.registry.strip() or input("Введите Docker Registry (например, ghcr.io или docker.io): ").strip() or "docker.io"
|
|
if registry == "registry.hub.docker.com":
|
|
registry = "docker.io"
|
|
image = args.image.strip() or input(f"Введите имя образа (например, inecs/logboard) (по умолчанию: logboard): ").strip() or "logboard"
|
|
tag = args.tag.strip() or input("Введите тег образа (по умолчанию: latest): ").strip() or "latest"
|
|
|
|
content = content.replace("REGISTRY_PLACEHOLDER", registry)
|
|
content = content.replace("IMAGE_NAME_PLACEHOLDER", image)
|
|
content = content.replace("IMAGE_TAG_PLACEHOLDER", tag)
|
|
|
|
with io.open(args.output, "w", encoding="utf-8") as fh:
|
|
fh.write(content)
|
|
|
|
print(f"Файл {os.path.relpath(args.output, ROOT)} сгенерирован")
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
raise SystemExit(main())
|
|
|
|
|