add chuni_intl to generation script

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)