Berack96
01e7bf58f1
- update NewsAPIsTool - update SocialAPIsTool - add tests for NewsAPIsTool - added some missing docs
37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
from gnews import GNews
|
|
from .base import Article, NewsWrapper
|
|
|
|
def result_to_article(result: dict) -> Article:
|
|
article = Article()
|
|
article.source = result.get("source", "")
|
|
article.time = result.get("publishedAt", "")
|
|
article.title = result.get("title", "")
|
|
article.description = result.get("description", "")
|
|
return article
|
|
|
|
class GoogleNewsWrapper(NewsWrapper):
|
|
"""
|
|
A wrapper for the Google News RSS Feed (Documentation: https://github.com/ranahaani/GNews/?tab=readme-ov-file#about-gnews)
|
|
It does not require an API key and is free to use.
|
|
"""
|
|
|
|
def get_top_headlines(self, total: int = 100) -> list[Article]:
|
|
gnews = GNews(language='en', max_results=total, period='7d')
|
|
results = gnews.get_top_news()
|
|
|
|
articles = []
|
|
for result in results:
|
|
article = result_to_article(result)
|
|
articles.append(article)
|
|
return articles
|
|
|
|
def get_latest_news(self, query: str, total: int = 100) -> list[Article]:
|
|
gnews = GNews(language='en', max_results=total, period='7d')
|
|
results = gnews.get_news(query)
|
|
|
|
articles = []
|
|
for result in results:
|
|
article = result_to_article(result)
|
|
articles.append(article)
|
|
return articles
|