add wac_plus community

1 files changed, 8 insertions, 0 deletions

diff --git a/news_feed.py b/news_feed.py
index c9f5131..a5a1ade 100644
--- a/news_feed.py
+++ b/news_feed.py
@@ -30,6 +30,8 @@ import sega.maimaidx_intl as maimaidx_intl
 import sega.ongeki_jp as ongeki_jp
 import taito.music_diver as music_diver
 import bandai_namco.taiko as taiko
+import community.disc as disc
+import community.wacca_plus.wacca_plus as wac_plus
 import constants
 import translate
 
@@ -135,6 +137,12 @@ def get_news(news_url: str, version=None) -> list:
         site_data = download_site_as_html(news_url)
         news_posts = sorted(taiko.parse_taiko_blog_site(site_data), key=lambda x: x['timestamp'], reverse=True)
 
+    elif news_url == constants.WACCA_PLUS_MAGIC_STRING:
+        if not wac_plus.check_is_generation_possible():
+            news_posts = []
+        else:
+            messages = disc.fetch_messages(constants.WACCA_PLUS_MAGIC_STRING)
+            news_posts = sorted(wac_plus.parse_announcement_messages(messages), key=lambda x: x['timestamp'], reverse=True)
     else:
         news_posts = []
     return news_posts