Agent/examples/content_finder/db/__init__.py

"""
数据库封装模块

- connection: 共享数据库连接
- schedule: 定时任务相关（demand_content、demand_find_task）
- store_results: 推荐结果写入（demand_find_author、demand_find_content_result）
"""

from .connection import get_connection
from .open_aigc_pattern_connection import get_open_aigc_pattern_connection
from .schedule import (
    get_next_unprocessed_demand,
    get_daily_unprocessed_pool,
    get_first_running_task,
    get_one_today_unprocessed_demand,
    create_task_record,
    fetch_trace_ids_created_after,
    update_task_status,
    update_task_on_complete,
)
from .store_results import (
    fetch_demand_content_dt,
    upsert_good_authors,
    insert_contents,
    update_content_plan_ids,
    update_web_html_url,
    update_process_trace_by_aweme_id,
)

__all__ = [
    "get_connection",
    "get_open_aigc_pattern_connection",
    "get_next_unprocessed_demand",
    "get_daily_unprocessed_pool",
    "get_first_running_task",
    "get_one_today_unprocessed_demand",
    "create_task_record",
    "fetch_trace_ids_created_after",
    "update_task_status",
    "update_task_on_complete",
    "fetch_demand_content_dt",
    "upsert_good_authors",
    "insert_contents",
    "update_content_plan_ids",
    "update_web_html_url",
    "update_process_trace_by_aweme_id",
]