From 8a68a0f554f780f0e973df181406eb25bb94b8f5 Mon Sep 17 00:00:00 2001 From: Pinapelz Date: Tue, 15 Apr 2025 00:33:25 -0700 Subject: add chuni_intl to generation script --- constants.py | 4 +++- generate.py | 13 ++++++++++++- news_feed.py | 4 ++++ sega/chuni_intl.py | 2 +- 4 files changed, 20 insertions(+), 3 deletions(-) diff --git a/constants.py b/constants.py index 6ef036e..6d13daf 100644 --- a/constants.py +++ b/constants.py @@ -7,12 +7,14 @@ SOUND_VOLTEX_EXCEED_GEAR_NEWS_SITE ="https://p.eagate.573.jp/game/sdvx/vi/news/i IIDX_PINKY_CRUSH_NEWS_SITE="https://p.eagate.573.jp/game/2dx/32/info/index.html" CHUNITHM_JP_NEWS_SITE="https://info-chunithm.sega.jp/" +CHUNITHM_INTL_NEWS_SITE="https://info-chunithm.sega.com/" MAIMAIDX_JP_NEWS_SITE="https://info-maimai.sega.jp/" MAIMAIDX_INTL_NEWS_SITE="https://maimai.sega.com/download/" ONGEKI_JP_NEWS_SITE="https://info-ongeki.sega.jp/" class CHUNITHM_VERSION(Enum): - VERSE = 1 + LUMINOUS_PLUS = 1 + VERSE = 2 class MAIMAIDX_VERSION(Enum): PRISM = 1 diff --git a/generate.py b/generate.py index e91c8ee..a136fa5 100644 --- a/generate.py +++ b/generate.py @@ -102,6 +102,17 @@ if __name__ == "__main__": maimaidx_intl_news_data = json.load(json_file)['news_posts'] print("MAIMAIDX INTL Data not fetched, using existing data.") - news = create_merged_feed(iidx_news_data, sdvx_news_data, chunithm_jp_news_data, maimaidx_jp_news_data, ongeki_jp_news_data, maimaidx_intl_news_data) + print("Fetching CHUNITHM INTL Data") + chunithm_intl_news_data = feed.get_news(constants.CHUNITHM_INTL_NEWS_SITE, constants.CHUNITHM_VERSION.LUMINOUS_PLUS) + if len(chunithm_intl_news_data) != 0: + with open(OUTPUT_DIR+'/chunithm_intl_news.json', 'w') as json_file: + json.dump(attach_news_meta_data(chunithm_intl_news_data), json_file) + print("CHUNITHM INTL Data fetched and saved.") + elif len(chunithm_intl_news_data) == 0 and os.path.exists(OUTPUT_DIR+'/chunithm_intl_news.json'): + with open(OUTPUT_DIR+'/chunithm_intl_news.json', 'r') as json_file: + chunithm_intl_news_data = json.load(json_file)['news_posts'] + print("CHUNITHM INTL Data not fetched, using existing data.") + + news = create_merged_feed(iidx_news_data, sdvx_news_data, chunithm_jp_news_data, maimaidx_jp_news_data, ongeki_jp_news_data, maimaidx_intl_news_data, chunithm_intl_news_data) with open(OUTPUT_DIR+'/news.json', 'w') as json_file: json.dump(attach_news_meta_data(news), json_file) diff --git a/news_feed.py b/news_feed.py index 13720a8..602b5a6 100644 --- a/news_feed.py +++ b/news_feed.py @@ -21,6 +21,7 @@ from site_scraper import SiteScraper import bemani.sdvx as sound_voltex import bemani.iidx as iidx import sega.chuni_jp as chunithm_jp +import sega.chuni_intl as chuni_intl import sega.maimaidx_jp as maimaidx_jp import sega.maimaidx_intl as maimaidx_intl import sega.ongeki_jp as ongeki_jp @@ -36,6 +37,9 @@ def get_news(news_url: str, version=None) -> list: elif news_url == constants.CHUNITHM_JP_NEWS_SITE: 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: + 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: 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) diff --git a/sega/chuni_intl.py b/sega/chuni_intl.py index 1421773..44bab1e 100644 --- a/sega/chuni_intl.py +++ b/sega/chuni_intl.py @@ -53,7 +53,7 @@ def make_chuni_intl_parser(identifier: str, parser: ParserVersion): return results - if parser == ParserVersion.ALPHA + if parser == ParserVersion.ALPHA: return alpha_parser parse_chuni_intl_luminous_plus_news_site = make_chuni_intl_parser("CHUNITHM_INTL_LUMINOUS_PLUS", ParserVersion.ALPHA) -- cgit v1.2.3