add back games, conform them to new title dispatch

2023-02-17 01:02:21 -05:00
parent f18e939dd0
commit 7e3396a7ff
214 changed files with 19412 additions and 23 deletions
--- a/titles/cxb/read.py
+++ b/titles/cxb/read.py
@ -0,0 +1,62 @@
+from typing import Optional, Dict, List
+from os import walk, path
+import urllib
+import csv
+
+from read import BaseReader
+from core.config import CoreConfig
+from titles.cxb.database import CxbData
+from titles.cxb.const import CxbConstants
+
+class CxbReader(BaseReader):
+    def __init__(self, config: CoreConfig, version: int, bin_arg: Optional[str], opt_arg: Optional[str], extra: Optional[str]) -> None:
+        super().__init__(config, version, bin_arg, opt_arg, extra)
+        self.data = CxbData(config)
+
+        try:
+            self.logger.info(f"Start importer for {CxbConstants.game_ver_to_string(version)}")
+        except IndexError:
+            self.logger.error(f"Invalid project cxb version {version}")
+            exit(1)
+
+    def read(self) -> None:
+        pull_bin_ram = True
+
+        if not path.exists(f"{self.bin_dir}"):
+            self.logger.warn(f"Couldn't find csv file in {self.bin_dir}, skipping")
+            pull_bin_ram = False
+
+        if pull_bin_ram:
+            self.read_csv(f"{self.bin_dir}")
+        
+    def read_csv(self, bin_dir: str) -> None:
+        self.logger.info(f"Read csv from {bin_dir}")
+
+        try:
+            fullPath = bin_dir + "/export.csv"
+            with open(fullPath, encoding="UTF-8") as fp:
+                reader = csv.DictReader(fp)
+                for row in reader:
+                    song_id = row["mcode"]
+                    index = row["index"]
+                    title = row["name"]
+                    artist = row["artist"]
+                    genre = row["category"]
+
+                    if not "N/A" in row["standard"]:
+                        self.logger.info(f"Added song {song_id} chart 0")
+                        self.data.static.put_music(self.version, song_id, index, 0, title, artist, genre, int(row["standard"].replace("Standard ","").replace("N/A","0")))
+                    if not "N/A" in row["hard"]:
+                        self.logger.info(f"Added song {song_id} chart 1")
+                        self.data.static.put_music(self.version, song_id, index, 1, title, artist, genre, int(row["hard"].replace("Hard ","").replace("N/A","0")))
+                    if not "N/A" in row["master"]:
+                        self.logger.info(f"Added song {song_id} chart 2")
+                        self.data.static.put_music(self.version, song_id, index, 2, title, artist, genre, int(row["master"].replace("Master ","").replace("N/A","0")))
+                    if not "N/A" in row["unlimited"]:
+                        self.logger.info(f"Added song {song_id} chart 3")
+                        self.data.static.put_music(self.version, song_id, index, 3, title, artist, genre, int(row["unlimited"].replace("Unlimited ","").replace("N/A","0")))
+                    if not "N/A" in row["easy"]:
+                        self.logger.info(f"Added song {song_id} chart 4")
+                        self.data.static.put_music(self.version, song_id, index, 4, title, artist, genre, int(row["easy"].replace("Easy ","").replace("N/A","0")))
+        except:
+            self.logger.warn(f"Couldn't read csv file in {self.bin_dir}, skipping")