add chuni_intl to generation script

1 files changed, 12 insertions, 1 deletions

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)