core/data_fetchers.py

"""캐시된 API 데이터 페처.

Streamlit cache를 활용한 API 호출 함수들.
"""
import streamlit as st

from .api_client import ChainShiftClient


@st.cache_data(ttl=300)
def get_campaigns(api_key: str = "", access_token: str = ""):
    """Fetch all campaigns with pagination and caching."""
    client = ChainShiftClient(api_key=api_key or None, access_token=access_token or None)
    all_items: list[dict] = []
    page = 1
    while True:
        resp = client.get_campaigns(page=page, page_size=100)
        items = resp.get("data", {}).get("items", [])
        all_items.extend(items)
        total = resp.get("data", {}).get("total", 0)
        if len(all_items) >= total or not items:
            break
        page += 1
    return {"data": {"items": all_items, "total": len(all_items)}}


@st.cache_data(ttl=60)
def get_nudge_candidates(
    api_key: str = "",
    campaign_id: int = 0,
    page: int = 1,
    page_size: int = 50,
    platform: str | None = None,
    confidence_tier: str | None = None,
    access_token: str = "",
):
    """Fetch nudge candidates (in-house brand negative mentions)."""
    client = ChainShiftClient(api_key=api_key or None, access_token=access_token or None)
    return client.get_nudge_candidates(
        campaign_id,
        page=page,
        page_size=page_size,
        platform=platform if platform and platform != "전체" else None,
    )


@st.cache_data(ttl=60)
def get_brand_mentions(
    api_key: str = "",
    campaign_id: int = 0,
    brand_type: str | None = None,
    polarity: str | None = None,
    access_token: str = "",
):
    """Fetch brand mention analysis."""
    client = ChainShiftClient(api_key=api_key or None, access_token=access_token or None)
    return client.get_brand_mentions(
        campaign_id,
        brand_type=brand_type if brand_type and brand_type != "전체" else None,
        polarity=polarity if polarity and polarity != "전체" else None,
    )


@st.cache_data(ttl=60)
def get_feedback_stats(api_key: str = "", campaign_id: int = 0, access_token: str = ""):
    """Fetch feedback statistics for a campaign."""
    try:
        client = ChainShiftClient(api_key=api_key or None, access_token=access_token or None)
        return client.get_feedback_stats(campaign_id)
    except Exception:
        return None