o2o-site-AEO/solution/backend/scripts/demo_build.py

"""생성 사이트 데모 — 실 DB 에 데이터가 갖춰진 사업장을 만들고 빌드해서 HTML 을 파일로 뽑는다.

    python scripts/demo_build.py     (backend/ 에서 실행)

★ 실 DB(web4ai_db)에 데모 회사·계정·사업장을 만든다. 개발 DB 에서만 쓸 것.
"""
import asyncio, os, sys, uuid
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
os.environ["APP_ENV"] = "local"

from httpx import ASGITransport, AsyncClient
from sqlalchemy import text
from sqlalchemy.ext.asyncio import create_async_engine

from common.enums import FactStatus, MediaStatus, PlaceCategory, SourceType, UserRole, UserStatus
from config.server_configs import main_db_config

NAME = "핑크비치펜션"
PHOTOS = [
    ("외관", "https://picsum.photos/seed/pension-out/1200/800", "2층 목조 건물 외관과 앞마당 잔디"),
    ("A동 침실", "https://picsum.photos/seed/pension-bed/1200/800", "퀸 침대와 창밖으로 바다가 보이는 침실"),
    ("A동 거실", "https://picsum.photos/seed/pension-liv/1200/800", "소파와 원목 테이블이 있는 거실"),
    ("바비큐장", "https://picsum.photos/seed/pension-bbq/1200/800", "지붕이 있는 야외 바비큐 공간"),
    ("B동 복층", "https://picsum.photos/seed/pension-duo/1200/800", "계단으로 이어진 복층 구조의 객실"),
]
PLACE_FACTS = {
    "check_in_time": "15:00", "check_out_time": "11:00",
    "cancel_policy": "이용 7일 전 100% 환불, 3일 전 50% 환불, 당일 취소 불가",
    "cooking_allowed": "true", "pet_allowed": "false", "smoking": "false",
    "parking": "true", "parking_capacity": "6", "wifi": "true",
    "bbq_available": "true", "bbq_fee": "20000", "breakfast": "false",
    "reception_hours": "09:00 - 21:00",
    "intro": "하조대 해변에서 도보 3분 거리에 있는 2개 동 규모의 펜션입니다. 모든 객실에서 바다가 보이며, 취사가 가능하고 야외 바비큐 공간을 갖추고 있습니다.",
}
UNITS = {
    "A동 스탠다드": {"room_type": "스탠다드", "standard_capacity": "2", "max_capacity": "4",
                  "bed_type": "퀸 1", "has_kitchen": "true", "view": "오션뷰", "weekday_price": "150000"},
    "B동 복층": {"room_type": "복층", "standard_capacity": "4", "max_capacity": "6",
               "bed_type": "퀸 1 + 싱글 2", "has_kitchen": "true", "view": "오션뷰", "weekday_price": "220000"},
}
FAQS = [
    ("체크인·체크아웃은 몇 시인가요?", "체크인은 15:00, 체크아웃은 11:00 입니다.", ["check_in_time", "check_out_time"]),
    ("반려동물과 함께 갈 수 있나요?", "반려동물 동반은 불가합니다.", ["pet_allowed"]),
    ("객실에서 취사가 가능한가요?", "취사가 가능하며 모든 객실에 주방이 있습니다.", ["cooking_allowed", "has_kitchen"]),
    ("바비큐를 이용하려면 얼마인가요?", "바비큐 이용료는 20,000원입니다.", ["bbq_available", "bbq_fee"]),
    ("주차는 가능한가요?", "주차가 가능하며 6대까지 주차할 수 있습니다.", ["parking", "parking_capacity"]),
]


async def main():
    pw = f":{main_db_config.write_pw}" if main_db_config.write_pw else ""
    dsn = f"postgresql+asyncpg://{main_db_config.write_id}{pw}@{main_db_config.write_host}:{main_db_config.write_port}/{main_db_config.name}"
    engine = create_async_engine(dsn)

    from router.v1.validator.dependencies import GetHashedPW
    cid, uid, login = uuid.uuid4(), uuid.uuid4(), f"demo{uuid.uuid4().hex[:6]}"
    async with engine.begin() as c:
        await c.execute(text("INSERT INTO company.companies (company_id,name,status) VALUES (:c,:n,1)"),
                        {"c": cid, "n": "데모대행사"})
        await c.execute(text("INSERT INTO company.users (user_id,company_id,id,password,name,status,role,last_accessed_at) "
                             "VALUES (:u,:c,:i,:p,:n,:s,:r,now())"),
                        {"u": uid, "c": cid, "i": login, "p": await GetHashedPW("pw1234"),
                         "n": "데모", "s": UserStatus.ACTIVE.value, "r": UserRole.OWNER.value})

    from router.router import app
    async with AsyncClient(transport=ASGITransport(app=app), base_url="http://demo") as cl:
        tok = (await cl.post("/v1/auth/login", json={"id": login, "password": "pw1234"})).json()["access_token"]
        h = {"Authorization": f"Bearer {tok}"}

        pid = (await cl.post("/v1/place", headers=h,
                             json={"name": NAME, "category": PlaceCategory.LODGING.value})).json()["place"]["place_id"]
        await cl.post(f"/v1/place/{pid}/verify", headers=h, json={
            "source": 2, "road_address": "강원특별자치도 양양군 현북면 하조대3길 11",
            "phone": "033-672-0000", "latitude": "38.0219217", "longitude": "128.7221449"})
        print(f"사업장 등록·검증 완료  place_id={pid}")

        for k, v in PLACE_FACTS.items():
            await cl.post(f"/v1/place/{pid}/fact", headers=h, json={"key": k, "value": v})

        unit_ids = {}
        for order, (name, kv) in enumerate(UNITS.items()):
            u = (await cl.post(f"/v1/place/{pid}/unit", headers=h,
                               json={"name": name, "sort_order": order})).json()["unit"]
            unit_ids[name] = u["unit_id"]
            for k, v in kv.items():
                await cl.post(f"/v1/place/{pid}/fact", headers=h,
                              json={"key": k, "value": v, "unit_id": u["unit_id"]})

        async with engine.begin() as c:
            for order, (label, url, alt) in enumerate(PHOTOS):
                unit_id = None
                for n, i in unit_ids.items():
                    if label.startswith(n.split()[0]):
                        unit_id = uuid.UUID(i)
                await c.execute(
                    text("INSERT INTO place.media (media_id,place_id,unit_id,url,origin_url,source_type,status,label,alt_text,sort_order) "
                         "VALUES (:m,:p,:u,:url,:url,:s,:st,:l,:a,:o)"),
                    {"m": uuid.uuid4(), "p": uuid.UUID(pid), "u": unit_id, "url": url,
                     "s": SourceType.OWNER.value, "st": MediaStatus.APPROVED.value,
                     "l": label, "a": alt, "o": order},
                )
            for order, (q, a, keys) in enumerate(FAQS):
                await c.execute(
                    text("INSERT INTO fact.faqs (faq_id,place_id,question,answer,source_fact_ids,generated_by,status,sort_order) "
                         "VALUES (:f,:p,:q,:a,CAST(:k AS jsonb),:g,:st,:o)"),
                    {"f": uuid.uuid4(), "p": uuid.UUID(pid), "q": q, "a": a,
                     "k": __import__("json").dumps(keys), "g": SourceType.LLM.value,
                     "st": FactStatus.VERIFIED.value, "o": order},
                )
        print(f"fact {len(PLACE_FACTS)}건 · 객실 {len(UNITS)}개 · 사진 {len(PHOTOS)}장 · FAQ {len(FAQS)}건 적재")

        job_id = (await cl.post(f"/v1/place/{pid}/site/build", headers=h,
                                json={"publish": True})).json()["job_id"]
        from crud.job_crud import JobQueue
        from worker.handlers import build_handler
        from worker.runner import Worker
        await Worker("demo", JobQueue(), build_handler(), job_deadline_sec=120).process_one()

        job = (await cl.get(f"/v1/job/{job_id}", headers=h)).json()["job"]
        r = job.get("result") or {}
        print(f"\n빌드: {r.get('build_status')} · 게이트 {'통과' if r.get('gate',{}).get('passed') else '거부'}"
              f" · 고유콘텐츠 {r.get('unique_content_count')}건 · 발행 {r.get('published')}")
        if not r.get("gate", {}).get("passed"):
            print("게이트 사유:", r.get("gate"))
            return

        # ★ HTML 은 백엔드가 만들지 않는다. payload JSON 을 쓰는 것까지가 백엔드의 일이고,
        #   그걸 정적 페이지로 굽는 것은 solution/site 의 렌더러다(그게 방문자가 보는 유일한 페이지).
        print(f"\npayload: {r.get('payload_path')}")
        print(f"페이지: {r.get('routes')}개 · JSON-LD 노드 {len(r.get('mismatches') or []) == 0 and '검증 통과' or '불일치'}")
        print("정적 파일: solution/site/out/s/<slug>/  (렌더러가 굽는다)")
        await engine.dispose()

asyncio.run(main())