trend_engine/utils.py

"""
Trend Engine 공통 유틸리티 — 재시도, 기간 계산, retention 정리

S7: 에러 핸들링 공통 래퍼
S6: 기간 계산 헬퍼
"""

import functools
import logging
import random
import time
from datetime import date, timedelta

logger = logging.getLogger(__name__)


# ------------------------------------------------------------------
# S7: 재시도 데코레이터
# ------------------------------------------------------------------


def retry_on_failure(max_retries=3, base_delay=1.0, exceptions=(Exception,)):
    """API 호출 실패 시 지수 백오프 재시도 데코레이터."""

    def decorator(func):
        @functools.wraps(func)
        def wrapper(*args, **kwargs):
            for attempt in range(max_retries):
                try:
                    return func(*args, **kwargs)
                except exceptions as e:
                    if attempt == max_retries - 1:
                        logger.error(
                            "%s failed after %d retries: %s",
                            func.__name__, max_retries, e,
                        )
                        raise
                    wait = base_delay * (2 ** attempt) + random.uniform(0, 0.5)
                    logger.warning(
                        "%s attempt %d failed: %s. Retrying in %.1fs",
                        func.__name__, attempt + 1, e, wait,
                    )
                    time.sleep(wait)

        return wrapper

    return decorator


# ------------------------------------------------------------------
# S6: 기간 계산 헬퍼
# ------------------------------------------------------------------


def get_week_period() -> tuple[date, date]:
    """이번 주 월요일(period_start)과 오늘(period_end)을 반환한다."""
    today = date.today()
    period_start = today - timedelta(days=today.weekday())  # 이번 주 월요일
    return period_start, today


# ------------------------------------------------------------------
# S6: Retention 정리
# ------------------------------------------------------------------


def safe_upsert_spot_trend(supabase, row: dict) -> bool:
    """PostgREST partial unique index 호환 upsert (DELETE + INSERT).

    spot_trends 테이블의 partial unique index(WHERE spot_id != '__pending__')가
    PostgREST의 on_conflict 파라미터와 호환되지 않아 42P10 에러가 발생한다.
    이를 우회하기 위해 기존 레코드를 먼저 삭제 후 새 레코드를 삽입한다.

    __pending__ 행은 건드리지 않는다 (blog_post INSERT는 별도 경로).
    """
    sid = row.get("spot_id", "")
    if sid == "__pending__":
        # __pending__은 중복 허용 — 일반 INSERT 사용
        supabase.table("spot_trends").insert(row).execute()
        return True

    try:
        supabase.table("spot_trends").delete() \
            .eq("spot_id", sid) \
            .eq("source", row["source"]) \
            .eq("metric_type", row["metric_type"]) \
            .eq("period_start", row["period_start"]) \
            .execute()
    except Exception:
        pass  # 삭제할 레코드 없으면 무시

    supabase.table("spot_trends").insert(row).execute()
    return True


def cleanup_old_trends(supabase, retention_weeks: int = 12) -> int:
    """retention_weeks 이상 된 spot_trends 레코드를 삭제한다.

    __pending__ 블로그 원본 레코드도 함께 정리한다.
    """
    cutoff = (date.today() - timedelta(weeks=retention_weeks)).isoformat()

    try:
        result = (
            supabase.table("spot_trends")
            .delete()
            .lt("collected_at", cutoff)
            .execute()
        )
        deleted = len(result.data) if result.data else 0
        logger.info(
            "retention 정리 완료: %d건 삭제 (cutoff: %s, %d주 이전)",
            deleted, cutoff, retention_weeks,
        )
        return deleted
    except Exception as e:
        logger.warning("retention 정리 실패: %s", e)
        return 0