add DanceRush and DanceARound eamuse app feeds

3 files changed, 21 insertions, 1 deletions

diff --git a/constants.py b/constants.py
index f69a610..e990f87 100644
--- a/constants.py
+++ b/constants.py
@@ -15,6 +15,8 @@ GITADORA_EAMUSE_APP_ID="9orw5gze0d1tkyhm"
 POPN_MUSIC_EAMUSE_APP_ID="17ua1w2bg3aggz00"
 NOSTALGIA_EAMUSE_APP_ID="mql0c9jmkhzf02r3"
 JUBEAT_EAMUSE_APP_ID="bz6w0u3gp546fpnq"
+DANCE_AROUND_APP_ID="kmhqpindcodm0mkh"
+DANCE_RUSH_APP_ID="ns3maqirvf08ddhp"
 
 CHUNITHM_JP_NEWS_SITE="https://info-chunithm.sega.jp/"
 CHUNITHM_INTL_NEWS_SITE="https://info-chunithm.sega.com/"
diff --git a/generate.py b/generate.py
index 84033b7..293cfc1 100644
--- a/generate.py
+++ b/generate.py
@@ -89,6 +89,12 @@ def generate_ddr_news_file(eamuse_feed: bool=False):
 def generate_polaris_chord_news_file():
     return generate_news_file("polaris_chord_news", constants.POLARIS_CHORD_NEWS_SITE)
 
+def generate_dance_around_news_file():
+    return generate_news_file("dance_around_news", constants.EAMUSE_APP_FEED, constants.DANCE_AROUND_APP_ID)
+
+def generate_dance_rush_news_file():
+    return generate_news_file("dance_rush_news", constants.EAMUSE_APP_FEED, constants.DANCE_RUSH_APP_ID)
+
 def generate_popn_music_news_file():
     return generate_news_file("popn_music_news", constants.EAMUSE_APP_FEED, constants.POPN_MUSIC_EAMUSE_APP_ID)
 
@@ -140,6 +146,8 @@ if __name__ == "__main__":
     sdvx_news_data = generate_sdvx_news_file()
     ddr_news_data = generate_ddr_news_file(eamuse_feed=True)
     polaris_news_data = generate_polaris_chord_news_file()
+    dance_rush_news_data = generate_dance_rush_news_file()
+    dance_around_news_data = generate_dance_around_news_file()
     gitadora_news_data = generate_gitadora_news_file()
     popn_music_news_data = generate_popn_music_news_file()
     jubeat_news_data = generate_jubeat_news_file()
@@ -156,6 +164,7 @@ if __name__ == "__main__":
     generate_rbdx_plus_news_file()
 
 
+
     news = create_merged_feed(
         iidx_news_data,
         sdvx_news_data,
@@ -172,7 +181,10 @@ if __name__ == "__main__":
         music_diver_news_data,
         taiko_news_data,
         wacca_plus_news,
-        museca_plus_news
+        museca_plus_news,
+        polaris_news_data,
+        dance_rush_news_data,
+        dance_around_news_data
     )
     log_output("Creating merged news.json file for all news that are within " + str(constants.DAYS_LIMIT) + " days old")
     with open(OUTPUT_DIR+'/news.json', 'w') as json_file:
diff --git a/news_feed.py b/news_feed.py
index 4896322..0a49707 100644
--- a/news_feed.py
+++ b/news_feed.py
@@ -93,6 +93,12 @@ def get_news(news_url: str, version=None) -> list:
             case constants.NOSTALGIA_EAMUSE_APP_ID:
                 news_posts= sorted(eamuse_app.parse_news_page(site_data, "NOSTALGIA_EAMUSEMENT"), key=lambda x: x['timestamp'], reverse=True)
                 news_posts = translate.add_translate_text_to_en(news_posts)
+            case constants.DANCE_RUSH_APP_ID:
+                news_posts= sorted(eamuse_app.parse_news_page(site_data, "DANCE_RUSH_EAMUSEMENT"), key=lambda x: x['timestamp'], reverse=True)
+                news_posts = translate.add_translate_text_to_en(news_posts)
+            case constants.DANCE_AROUND_APP_ID:
+                news_posts= sorted(eamuse_app.parse_news_page(site_data, "DANCE_AROUND_EAMUSEMENT"), key=lambda x: x['timestamp'], reverse=True)
+                news_posts = translate.add_translate_text_to_en(news_posts)
             case _:
                 raise ValueError("Cannot find provided e-amuse app gameId", version)