실습 6 — docker compose 로 묶을 준비와 로컬 확인

from pathlib import Path
import yaml

DEPLOY = Path("out_deploy")
compose_text = """services:
  api:
    build: ./api
    ports: ["8013:8013"]
    healthcheck:
      test: ["CMD", "python", "-c", "import urllib.request; urllib.request.urlopen('http://127.0.0.1:8013/health')"]
      interval: 10s
  ui:
    build: ./ui
    ports: ["8501:8501"]
    environment:
      API_URL: http://api:8013
    depends_on:
      api:
        condition: service_healthy
"""
(DEPLOY / "compose.yaml").write_text(compose_text, encoding="utf-8")
compose = yaml.safe_load(compose_text)
print({name: svc.get("ports") for name, svc in compose["services"].items()})

from urllib.parse import urlsplit

problems = []
services = compose["services"]
for name, svc in services.items():
    if "build" in svc and not (DEPLOY / svc["build"] / "Dockerfile").exists():      # build 폴더에 Dockerfile 이 있나
        problems.append(f"{name}: {svc['build']} 에 Dockerfile 이 없다")
host_ports = [p.split(":")[0] for svc in services.values() for p in svc.get("ports", [])]
if len(host_ports) != len(set(host_ports)):                                          # PC 쪽 포트가 겹치나
    problems.append("PC 쪽 포트가 겹친다")
for name, svc in services.items():                                                   # 주소의 호스트 이름이 서비스 이름인가
    for key, value in svc.get("environment", {}).items():
        host = urlsplit(value).hostname
        if value.startswith("http") and host not in services:
            problems.append(f"{name}.{key}: {host} 라는 서비스가 없다")
expose = (DEPLOY / "api" / "Dockerfile").read_text(encoding="utf-8")
print("api Dockerfile 의 포트와 compose 의 포트가 같다:", "--port\", \"8013" in expose and "8013:8013" in services["api"]["ports"])
print("문제:", problems or "없음")

import os, subprocess, sys, time, urllib.request

# 두 서비스를 각자의 폴더에서 띄운다. compose 의 서비스 이름(api) 대신 127.0.0.1 을 준다
api = subprocess.Popen([sys.executable, "-m", "uvicorn", "main:app", "--port", "8013", "--log-level", "warning"],
                       cwd=DEPLOY / "api")
ui = subprocess.Popen([sys.executable, "-m", "streamlit", "run", "app.py", "--server.port", "8501", "--server.headless", "true",
                       "--browser.gatherUsageStats", "false"], cwd=DEPLOY / "ui",
                      env=dict(os.environ, API_URL="http://127.0.0.1:8013"), stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
for url in ("http://127.0.0.1:8013/health", "http://127.0.0.1:8501/_stcore/health"):
    for _ in range(60):
        try:
            print(url, "→", urllib.request.urlopen(url, timeout=2).read().decode())
            break
        except OSError:
            time.sleep(0.5)

from streamlit.testing.v1 import AppTest

os.environ["API_URL"] = "http://127.0.0.1:8013"
at = AppTest.from_file(str((DEPLOY / "ui" / "app.py").resolve()), default_timeout=60).run()
at.button[0].click().run()                                     # 기본값 그대로 예측
print(at.caption[0].value, "·", at.metric[0].value)
for proc in (ui, api):
    proc.terminate()
    proc.wait(timeout=10)
print("두 서버 꺼짐")