try_topic / scraper.py
Supitn's picture
Create scraper.py
d08f4c4 verified
import requests
from bs4 import BeautifulSoup
import feedparser
from datetime import datetime
import time
class YahooFinanceScraper:
def __init__(self):
self.headers = {
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36'
}
self.base_url = "https://finance.yahoo.com"
def get_latest_news(self, symbol="", max_articles=10):
"""
ดึงข่าวล่าสุดจาก Yahoo Finance
Args:
symbol: ticker symbol (เช่น AAPL, TSLA) หรือเว้นว่างสำหรับข่าวทั่วไป
max_articles: จำนวนข่าวที่ต้องการ
Returns:
list: รายการข่าว [{title, link, summary, published}]
"""
news_list = []
try:
if symbol:
# ดึงข่าวเฉพาะหุ้น
url = f"{self.base_url}/quote/{symbol}"
response = requests.get(url, headers=self.headers, timeout=10)
soup = BeautifulSoup(response.content, 'html.parser')
# หา news section
news_items = soup.find_all('li', class_='stream-item', limit=max_articles)
for item in news_items:
try:
title_tag = item.find('h3')
link_tag = item.find('a')
summary_tag = item.find('p')
time_tag = item.find('time')
if title_tag and link_tag:
news_list.append({
'title': title_tag.get_text(strip=True),
'link': link_tag.get('href', ''),
'summary': summary_tag.get_text(strip=True) if summary_tag else '',
'published': time_tag.get_text(strip=True) if time_tag else 'N/A'
})
except Exception as e:
continue
else:
# ดึงข่าวทั่วไปผ่าน RSS Feed
rss_url = "https://finance.yahoo.com/news/rssindex"
feed = feedparser.parse(rss_url)
for entry in feed.entries[:max_articles]:
news_list.append({
'title': entry.get('title', 'No title'),
'link': entry.get('link', ''),
'summary': entry.get('summary', '')[:300], # จำกัด 300 ตัวอักษร
'published': entry.get('published', 'N/A')
})
# ถ้าไม่มีข่าว ลองวิธีสำรอง
if not news_list:
news_list = self._get_fallback_news(max_articles)
except Exception as e:
print(f"Error scraping news: {e}")
news_list = self._get_fallback_news(max_articles)
return news_list
def _get_fallback_news(self, max_articles):
"""วิธีสำรองในกรณีที่ดึงข่าวไม่ได้"""
try:
# ใช้ Yahoo Finance RSS feed แบบง่าย
url = "https://finance.yahoo.com/rss/topstories"
feed = feedparser.parse(url)
news_list = []
for entry in feed.entries[:max_articles]:
news_list.append({
'title': entry.get('title', 'No title'),
'link': entry.get('link', ''),
'summary': entry.get('summary', '')[:300],
'published': entry.get('published', 'N/A')
})
return news_list
except:
return [{
'title': 'Unable to fetch news',
'link': '',
'summary': 'Please try again later',
'published': 'N/A'
}]
def search_news(self, keyword, max_articles=10):
"""
ค้นหาข่าวด้วย keyword
Args:
keyword: คำค้นหา
max_articles: จำนวนข่าว
Returns:
list: รายการข่าวที่เกี่ยวข้อง
"""
try:
search_url = f"{self.base_url}/search?p={keyword}"
response = requests.get(search_url, headers=self.headers, timeout=10)
soup = BeautifulSoup(response.content, 'html.parser')
news_list = []
articles = soup.find_all('div', class_='Ov(h)', limit=max_articles)
for article in articles:
try:
title = article.find('h3')
link = article.find('a')
if title and link:
news_list.append({
'title': title.get_text(strip=True),
'link': link.get('href', ''),
'summary': '',
'published': 'Recent'
})
except:
continue
return news_list if news_list else self.get_latest_news(max_articles=max_articles)
except Exception as e:
print(f"Search error: {e}")
return self.get_latest_news(max_articles=max_articles)