diff options
Diffstat (limited to 'news_feed.py')
| -rw-r--r-- | news_feed.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/news_feed.py b/news_feed.py index 8bd1179..01ee3d1 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.maimaidx_jp as maimaidx_jp import constants def get_news(news_url: str, version=None) -> list: @@ -33,8 +34,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.MAIMAIDX_NEWS_SITE: - pass + 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) else: news_posts = [] scraper.close() |