__init__.py 1.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445
  1. """
  2. 数据库封装模块
  3. - connection: 共享数据库连接
  4. - schedule: 定时任务相关(demand_content、demand_find_task)
  5. - store_results: 推荐结果写入(demand_find_author、demand_find_content_result)
  6. """
  7. from .connection import get_connection
  8. from .open_aigc_pattern_connection import get_open_aigc_pattern_connection
  9. from .schedule import (
  10. get_next_unprocessed_demand,
  11. get_daily_unprocessed_pool,
  12. get_first_running_task,
  13. get_one_today_unprocessed_demand,
  14. create_task_record,
  15. fetch_trace_ids_created_after,
  16. update_task_status,
  17. update_task_on_complete,
  18. )
  19. from .store_results import (
  20. upsert_good_authors,
  21. insert_contents,
  22. update_content_plan_ids,
  23. update_web_html_url,
  24. update_process_trace_by_aweme_id,
  25. )
  26. __all__ = [
  27. "get_connection",
  28. "get_open_aigc_pattern_connection",
  29. "get_next_unprocessed_demand",
  30. "get_daily_unprocessed_pool",
  31. "get_first_running_task",
  32. "get_one_today_unprocessed_demand",
  33. "create_task_record",
  34. "fetch_trace_ids_created_after",
  35. "update_task_status",
  36. "update_task_on_complete",
  37. "upsert_good_authors",
  38. "insert_contents",
  39. "update_content_plan_ids",
  40. "update_web_html_url",
  41. "update_process_trace_by_aweme_id",
  42. ]