long_article_mapper.py 665 B

12345678910111213141516171819
  1. from typing import Set, List
  2. from app.core.database import DatabaseManager
  3. class LongArticleDatabaseMapper:
  4. class Basic:
  5. @staticmethod
  6. async def get_unsafe_articles(pool: DatabaseManager) -> Set[str]:
  7. query = """
  8. SELECT title FROM article_unsafe_title WHERE status = 1;
  9. """
  10. response = await pool.async_fetch(query=query)
  11. return set([i["title"] for i in response])
  12. class Recommend(Basic):
  13. @staticmethod
  14. async def get_top_articles(pool: DatabaseManager, query: str, gh_id: str):
  15. return await pool.async_fetch(query=query, params=(gh_id,))