실습 5 — 컨테이너에 넣을 준비

import shutil
from pathlib import Path

DEPLOY = Path("out_deploy")
shutil.rmtree(DEPLOY, ignore_errors=True)
(DEPLOY / "api").mkdir(parents=True)
(DEPLOY / "ui").mkdir(parents=True)
shutil.copy("out_delay_api.py", DEPLOY / "api" / "main.py")            # API 코드는 main.py 로
shutil.copy("out_delay_model.joblib", DEPLOY / "api")                  # 모델 파일도 함께
shutil.copy("out_app_api.py", DEPLOY / "ui" / "app.py")                # 화면은 app.py 로. 모델은 넣지 않는다
for p in sorted(DEPLOY.rglob("*")):
    print(p.as_posix(), f"{p.stat().st_size:,}바이트" if p.is_file() else "")

import ast, sys
from importlib.metadata import packages_distributions, version


def imports_of(path):
    """파이썬 파일을 읽어 import 하는 맨 위 모듈 이름들을 모은다(실행하지 않고 문법만 읽는다)"""
    tree = ast.parse(Path(path).read_text(encoding="utf-8"))
    names = set()
    for node in ast.walk(tree):
        if isinstance(node, ast.Import):
            names |= {a.name.split(".")[0] for a in node.names}
        elif isinstance(node, ast.ImportFrom) and node.module:
            names.add(node.module.split(".")[0])
    return names - set(sys.stdlib_module_names)                  # 표준 라이브러리는 설치할 필요가 없다


dist = packages_distributions()                                   # 모듈 이름 → 패키지(배포판) 이름
found = {}
for service in ("api", "ui"):
    mods = imports_of(DEPLOY / service / ("main.py" if service == "api" else "app.py"))
    found[service] = sorted({dist[m][0] for m in mods})
    print(service, "의 import:", sorted(mods), "→ 패키지:", found[service])

# import 에는 없지만 실행에 필요한 것: 서버를 띄우는 uvicorn, joblib 파일을 풀 때 필요한 scikit-learn
extra = {"api": ["uvicorn", "scikit-learn"], "ui": []}
for service in ("api", "ui"):
    names = sorted(set(found[service]) | set(extra[service]), key=str.lower)
    lines = [f"{n}=={version(n)}" for n in names]
    (DEPLOY / service / "requirements.txt").write_text("\n".join(lines) + "\n", encoding="utf-8")
    print(f"--- {service}/requirements.txt", *lines, sep="\n")

DOCKERFILE = {
    "api": """FROM python:3.14-slim
WORKDIR /app
# 패키지 목록만 먼저 복사해 설치한다. 코드만 바뀌면 이 층을 다시 쓰지 않는다
COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt
COPY . .
EXPOSE 8013
# 컨테이너 바깥에서 연결할 수 있게 0.0.0.0 으로 띄운다
CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8013"]
""",
    "ui": """FROM python:3.14-slim
WORKDIR /app
COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt
COPY . .
ENV API_URL=http://api:8013
EXPOSE 8501
CMD ["streamlit", "run", "app.py", "--server.port", "8501", "--server.address", "0.0.0.0", "--server.headless", "true", "--browser.gatherUsageStats", "false"]
""",
}
IGNORE = "__pycache__/\n*.pyc\n.venv/\n*.log\n.env\n"            # 이미지에 넣지 않을 것. .env 에는 비밀이 들기 쉽다
for service, text in DOCKERFILE.items():
    (DEPLOY / service / "Dockerfile").write_text(text, encoding="utf-8")
    (DEPLOY / service / ".dockerignore").write_text(IGNORE, encoding="utf-8")
    files = sorted(p.name for p in (DEPLOY / service).iterdir())
    size = sum(p.stat().st_size for p in (DEPLOY / service).iterdir())
    print(service, files, f"· 빌드에 보낼 파일 {size / 1024:.1f}KB")