From a5b15484423f9c9c9518a7be49845f018a8ff46f Mon Sep 17 00:00:00 2001 From: Pinapelz Date: Wed, 16 Apr 2025 23:55:00 -0700 Subject: feat: add support for DDR World --- news_feed.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) (limited to 'news_feed.py') diff --git a/news_feed.py b/news_feed.py index ac90f0f..75a3678 100644 --- a/news_feed.py +++ b/news_feed.py @@ -19,6 +19,7 @@ Generic format for a news entry. All keys are considered to be nullable from site_scraper import SiteScraper, download_site_as_html import bemani.sdvx as sound_voltex import bemani.iidx as iidx +import bemani.ddr as ddr import sega.chuni_jp as chunithm_jp import sega.chuni_intl as chuni_intl import sega.maimaidx_jp as maimaidx_jp @@ -30,14 +31,21 @@ import translate def get_news(news_url: str, version=None) -> list: 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) + news_posts = sorted(sound_voltex.parse_exceed_gear_news_site(site_data), key=lambda x: x['timestamp'], reverse=True) news_posts = translate.add_translate_text_to_en(news_posts) 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) + news_posts = sorted(iidx.parse_pinky_crush_news_site(site_data), key=lambda x: x['timestamp'], reverse=True) news_posts = translate.add_translate_text_to_en(news_posts, iidx.KEY_TERMS_TL) + elif news_url == constants.DDR_WORLD_NEWS_SITE: + scraper = SiteScraper(headless=True) + site_data = scraper.get_page_source(news_url) + scraper.close() + news_posts = sorted(ddr.parse_ddr_world_news_site(site_data), key=lambda x: x['timestamp'], reverse=True) + news_posts = translate.add_translate_text_to_en(news_posts) + elif news_url == constants.CHUNITHM_JP_NEWS_SITE: site_data = download_site_as_html(news_url) if version == constants.CHUNITHM_VERSION.VERSE: -- cgit v1.2.3