forked from AIAnytime/AI-News-Anchor-Generator-App
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathnews_api.py
33 lines (25 loc) · 1 KB
/
news_api.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
# news_api.py
import requests
import os
from dotenv import load_dotenv
load_dotenv()
api_key = os.getenv('NEWS_API_KEY')
class NewsAPI:
def __init__(self, api_key):
self.api_key = api_key
def get_news(self, query, num_news):
url = f'https://newsapi.org/v2/everything?q={query}&from=2023-09-16&sortBy=popularity&pageSize={num_news}&language=en&apiKey={self.api_key}'
response = requests.get(url)
if response.status_code == 200:
news_data = response.json()
return news_data.get("articles", [])
else:
return []
def get_news_descriptions(self, query, num_news):
news = self.get_news(query, num_news)
desc_list = [news_item['description'] for news_item in news]
return desc_list
def get_news_string(self, query, num_news):
desc_list = self.get_news_descriptions(query, num_news)
desc_string = ". ".join(desc_list)
return desc_string