diff options
Diffstat (limited to 'news_feed.py')
| -rw-r--r-- | news_feed.py | 22 |
1 files changed, 18 insertions, 4 deletions
diff --git a/news_feed.py b/news_feed.py index 602b5a6..e7c1a4a 100644 --- a/news_feed.py +++ b/news_feed.py @@ -17,7 +17,7 @@ Generic format for a news entry. All keys are considered to be nullable """ from email.utils import parsedate_to_datetime -from site_scraper import SiteScraper +from site_scraper import SiteScraper, download_site_as_html import bemani.sdvx as sound_voltex import bemani.iidx as iidx import sega.chuni_jp as chunithm_jp @@ -28,28 +28,42 @@ import sega.ongeki_jp as ongeki_jp import constants def get_news(news_url: str, version=None) -> list: - scraper = SiteScraper(headless=True) - site_data = scraper.get_page_source(news_url) + # As of right now all supported games don't require JS to pull data from + # scraper = SiteScraper(headless=True) + # site_data = scraper.get_page_source(news_url) if news_url == constants.SOUND_VOLTEX_EXCEED_GEAR_NEWS_SITE: + site_data = download_site_as_html(news_url) news_posts = sorted(sound_voltex.parse_exceed_gear_news_site(site_data, constants.EAMUSEMENT_BASE_URL), key=lambda x: x['timestamp'], reverse=True) + elif news_url == constants.IIDX_PINKY_CRUSH_NEWS_SITE: + site_data = download_site_as_html(news_url) news_posts = sorted(iidx.parse_pinky_crush_news_site(site_data, constants.EAMUSEMENT_BASE_URL), key=lambda x: x['timestamp'], reverse=True) + elif news_url == constants.CHUNITHM_JP_NEWS_SITE: + site_data = download_site_as_html(news_url) if version == constants.CHUNITHM_VERSION.VERSE: news_posts = sorted(chunithm_jp.parse_chuni_jp_verse_news_site(site_data), key=lambda x: x['timestamp'], reverse=True) + elif news_url == constants.CHUNITHM_INTL_NEWS_SITE: + site_data = download_site_as_html(news_url) if version == constants.CHUNITHM_VERSION.LUMINOUS_PLUS: news_posts = sorted(chuni_intl.parse_chuni_intl_luminous_plus_news_site(site_data), key=lambda x: x['timestamp'], reverse=True) + elif news_url == constants.MAIMAIDX_JP_NEWS_SITE: + site_data = download_site_as_html(news_url) if version == constants.MAIMAIDX_VERSION.PRISM_PLUS: news_posts = sorted(maimaidx_jp.parse_maimaidx_jp_prism_plus_news_site(site_data), key=lambda x: x['timestamp'], reverse=True) + elif news_url == constants.MAIMAIDX_INTL_NEWS_SITE: + site_data = download_site_as_html(news_url) if version == constants.MAIMAIDX_VERSION.PRISM: news_posts = sorted(maimaidx_intl.parse_maimaidx_intl_prism_news_site(site_data), key=lambda x: x['timestamp'], reverse=True) + elif news_url == constants.ONGEKI_JP_NEWS_SITE: + site_data = download_site_as_html(news_url) if version == constants.ONGEKI_VERSION.REFRESH: news_posts = sorted(ongeki_jp.parse_ongeki_refresh_news_site(site_data), key=lambda x: x['timestamp'], reverse=True) else: news_posts = [] - scraper.close() + # scraper.close() return news_posts |