Fix reading WORLD'S END charts for SDBT

2023-09-10 12:29:19 +10:00
parent 6c98f5f0a7
commit 4a916cb4d1
1 changed files with 4 additions and 2 deletions
@@ -169,8 +169,10 @@ class ChuniReader(BaseReader):
                            fumen_path = MusicFumenData.find("file").find("path")

                            if fumen_path is not None:
-                                chart_id = MusicFumenData.find("type").find("id").text
-                                if chart_id == "5":
+                                chart_type = MusicFumenData.find("type")
+                                chart_id = chart_type.find("id").text
+                                chart_diff = chart_type.find("str").text
+                                if chart_diff == "WorldsEnd" and (chart_id == "4" or chart_id == "5"): # 4 in SDBT, 5 in SDHD
                                    level = float(xml_root.find("starDifType").text)
                                    we_chara = (
                                        xml_root.find("worldsEndTagName")