refactor sega games to follow function factory design

unlikely to be much change between each game

1 files changed, 49 insertions, 36 deletions

diff --git a/sega/maimaidx_jp.py b/sega/maimaidx_jp.py
index 90530f0..720a618 100644
--- a/sega/maimaidx_jp.py
+++ b/sega/maimaidx_jp.py
@@ -1,46 +1,59 @@
 from bs4 import BeautifulSoup
 from datetime import datetime, timezone, timedelta
 from urllib.parse import urljoin
-import re
+from enum import Enum
 
-def parse_maimaidx_jp_prism_plus_news_site(html: str):
-    soup = BeautifulSoup(html, "html.parser")
-    base_url = "https://info-maimai.sega.jp/"
-    news_items = []
+class ParserVersion(Enum):
+    ALPHA=1
 
-    news_boxes = soup.select(".maiPager-content .newsBox")
-    for box in news_boxes:
-        a_tag = box.select_one("a")
-        url = urljoin(base_url, a_tag["href"]) if a_tag and a_tag.get("href") else None
+def make_maimaidx_jpn_parser(identifier: str, parser: ParserVersion):
+    def alpha_parser(html: str):
+        """
+        Confirmed on:
+        PRISM PLUS
+        """
+        soup = BeautifulSoup(html, "html.parser")
+        base_url = "https://info-maimai.sega.jp/"
+        news_items = []
 
-        img_tag = box.select_one("img")
-        image_url = urljoin(base_url, img_tag["src"]) if img_tag else None
+        news_boxes = soup.select(".maiPager-content .newsBox")
+        for box in news_boxes:
+            a_tag = box.select_one("a")
+            url = urljoin(base_url, a_tag["href"]) if a_tag and a_tag.get("href") else None
 
-        date_tag = box.select_one(".newsDate")
-        raw_date = date_tag.get_text(strip=True) if date_tag else None
+            img_tag = box.select_one("img")
+            image_url = urljoin(base_url, img_tag["src"]) if img_tag else None
 
-        jst = timezone(timedelta(hours=9))
-        try:
-            dt = datetime.strptime(raw_date.split(" ")[0], "%Y.%m.%d").replace(tzinfo=jst)
-            timestamp = int(dt.timestamp())
-        except:
-            dt = None
-            timestamp = 0
+            date_tag = box.select_one(".newsDate")
+            raw_date = date_tag.get_text(strip=True) if date_tag else None
 
-        content_tag = box.select_one(".newsLink")
-        content = content_tag.get_text(strip=True) if content_tag else None
-        news_items.append({
-            "date": raw_date,
-            "identifier": "MAIMAIDX_JPN_PRISM_PLUS",
-            "type": None,
-            "timestamp": timestamp,
-            "headline": None,
-            "content": content,
-            "url": url,
-            "images": [{
-                "image": image_url,
-                "link": url
-            }] if image_url else []
-        })
+            jst = timezone(timedelta(hours=9))
+            try:
+                dt = datetime.strptime(raw_date.split(" ")[0], "%Y.%m.%d").replace(tzinfo=jst)
+                timestamp = int(dt.timestamp())
+            except:
+                dt = None
+                timestamp = 0
 
-    return news_items
+            content_tag = box.select_one(".newsLink")
+            content = content_tag.get_text(strip=True) if content_tag else None
+
+            news_items.append({
+                "date": raw_date,
+                "identifier": identifier,
+                "type": None,
+                "timestamp": timestamp,
+                "headline": None,
+                "content": content,
+                "url": url,
+                "images": [{
+                    "image": image_url,
+                    "link": url
+                }] if image_url else []
+            })
+
+        return news_items
+    if parser == ParserVersion.ALPHA:
+        return alpha_parser
+
+parse_maimaidx_jp_prism_plus_news_site = make_maimaidx_jpn_parser("MAIMAIDX_JPN_PRISM_PLUS", ParserVersion.ALPHA)