1 files changed, 33 insertions, 0 deletions
diff --git a/konami.py b/konami.py
new file mode 100644
index 0000000..438b1ed
--- /dev/null
+++ b/konami.py
@@ -0,0 +1,33 @@
+"""
+Fetching data for Konami/Bemani games
+{
+    'date': JST date of news post
+    'type': Type of post if available, otherwise if not provided it will be None (aka Generic news)
+    'timestamp': Unixtime of date above,
+    'headline': Headline,
+    'content': All text content of news,
+    'images': {
+        'image': URL to image,
+        'link': If there's an associated href. Else None
+
+    }
+}
+"""
+
+from email.utils import parsedate_to_datetime
+from site_scraper import SiteScraper
+import bemani.sdvx as sound_voltex
+import bemani.iidx as iidx
+import constants
+
+def get_news(news_url: str) -> list:
+    scraper = SiteScraper(headless=True)
+    site_data = scraper.get_page_source(news_url)
+    if news_url == constants.SOUND_VOLTEX_EXCEED_GEAR_NEWS_SITE:
+        news_posts = sorted(sound_voltex.parse_exceed_gear_news_site(site_data, constants.EAMUSEMENT_BASE_URL), key=lambda x: x['timestamp'], reverse=True)
+    elif news_url == constants.IIDX_PINKY_CRUSH_NEWS_SITE:
+        news_posts = sorted(iidx.parse_pinky_crush_news_site(site_data, constants.EAMUSEMENT_BASE_URL), key=lambda x: x['timestamp'], reverse=True)
+    else:
+        news_posts = []
+    scraper.close()
+    return news_posts