mirror of
https://github.com/LmeSzinc/StarRailCopilot.git
synced 2024-11-16 06:25:24 +00:00
130 lines
4.6 KiB
Python
130 lines
4.6 KiB
Python
import re
|
||
import typing as t
|
||
|
||
from dev_tools.keywords.base import GenerateKeyword, SHARE_DATA, text_to_variable
|
||
from module.base.decorator import cached_property
|
||
from module.config.utils import deep_get
|
||
|
||
|
||
def dungeon_name(name: str) -> str:
|
||
name = text_to_variable(name)
|
||
name = re.sub('Bud_of_(Memories|Aether|Treasures)', r'Calyx_Golden_\1', name)
|
||
name = re.sub('Bud_of_(.*)', r'Calyx_Crimson_\1', name).replace('Calyx_Crimson_Calyx_Crimson_', 'Calyx_Crimson_')
|
||
name = re.sub('Shape_of_(.*)', r'Stagnant_Shadow_\1', name)
|
||
name = re.sub('Path_of_(.*)', r'Cavern_of_Corrosion_Path_of_\1', name)
|
||
if name in ['Destruction_Beginning', 'End_of_the_Eternal_Freeze', 'Divine_Seed', 'Borehole_Planet_Old_Crater']:
|
||
name = f'Echo_of_War_{name}'
|
||
if name in ['The_Swarm_Disaster', 'Gold_and_Gears']:
|
||
name = f'Simulated_Universe_{name}'
|
||
name = name.replace('Stagnant_Shadow_Stagnant_Shadow', 'Stagnant_Shadow')
|
||
name = name.replace('Cavern_of_Corrosion_Cavern_of_Corrosion', 'Cavern_of_Corrosion')
|
||
return name
|
||
|
||
|
||
class GenerateDungeonList(GenerateKeyword):
|
||
output_file = './tasks/dungeon/keywords/dungeon.py'
|
||
|
||
@cached_property
|
||
def data(self):
|
||
return SHARE_DATA.GameplayGuideData
|
||
|
||
def iter_keywords(self) -> t.Iterable[dict]:
|
||
for keyword in self.iter_dungeon():
|
||
if isinstance(keyword, str):
|
||
yield dict(
|
||
text_id=self.find_keyword(keyword, lang='cn')[0],
|
||
dungeon_id=-1,
|
||
plane_id=-1,
|
||
)
|
||
else:
|
||
yield keyword
|
||
|
||
def iter_dungeon(self):
|
||
temp_save = ""
|
||
for data in self.data:
|
||
dungeon_id = data.get('ID', 0)
|
||
text_id = deep_get(data, keys='Name.Hash')
|
||
plane_id = deep_get(data, 'MapEntranceID', 0)
|
||
_, name = self.find_keyword(text_id, lang='cn')
|
||
if '永屹之城遗秘' in name: # load after all forgotten hall to make sure the same order in Game UI
|
||
temp_save = text_id
|
||
continue
|
||
if '忘却之庭' in name:
|
||
continue
|
||
yield dict(
|
||
text_id=text_id,
|
||
dungeon_id=dungeon_id,
|
||
plane_id=plane_id,
|
||
)
|
||
if temp_save:
|
||
yield temp_save
|
||
# Consider rogue DLC as a dungeon
|
||
yield '寰宇蝗灾'
|
||
yield '黄金与机械'
|
||
# 'Memory of Chaos' is not a real dungeon, but represents a group
|
||
yield '混沌回忆'
|
||
yield '天艟求仙迷航录'
|
||
yield '永屹之城遗秘'
|
||
|
||
def convert_name(self, text: str, keyword: dict) -> str:
|
||
text = super().convert_name(text, keyword=keyword)
|
||
text = dungeon_name(text)
|
||
|
||
# Add plane suffix
|
||
from tasks.map.keywords import MapPlane
|
||
|
||
if text.startswith('Calyx_Crimson'):
|
||
plane = MapPlane.find_plane_id(keyword['plane_id'])
|
||
if plane is not None:
|
||
text = f'{text}_{plane.name}'
|
||
else:
|
||
text = f'{text}_unknown_plane'
|
||
return text
|
||
|
||
def convert_keyword(self, text: str, lang: str) -> str:
|
||
text = super().convert_keyword(text, lang=lang)
|
||
# Bud of Memories (Jarilo-Ⅵ)
|
||
# Use roman numbers instead
|
||
text = re.sub(r'-[VⅤ][IⅠ]', '-Ⅵ', text)
|
||
|
||
return text
|
||
|
||
def iter_rows(self) -> t.Iterable[dict]:
|
||
dungeons = list(super().iter_rows())
|
||
|
||
# Sort by path
|
||
calyx = []
|
||
order = [
|
||
'Calyx_Golden',
|
||
'Calyx_Crimson_Destruction',
|
||
'Calyx_Crimson_Preservation',
|
||
'Calyx_Crimson_The_Hunt',
|
||
'Calyx_Crimson_Abundance',
|
||
'Calyx_Crimson_Erudition',
|
||
'Calyx_Crimson_Harmony',
|
||
'Calyx_Crimson_Nihility',
|
||
]
|
||
for keyword in order:
|
||
condition = lambda x: x['name'].startswith(keyword)
|
||
calyx += [d for d in dungeons if condition(d)]
|
||
dungeons = [d for d in dungeons if not condition(d)]
|
||
dungeons = calyx + dungeons
|
||
|
||
# Reverse Divergent_Universe
|
||
start = 0
|
||
end = 0
|
||
for index, dungeon in enumerate(dungeons):
|
||
if dungeon['name'].startswith('Divergent_Universe'):
|
||
if start == 0:
|
||
start = index
|
||
end = index + 1
|
||
if start > 0 and end > 0:
|
||
dungeons = dungeons[:start] + dungeons[start:end][::-1] + dungeons[end:]
|
||
|
||
# Re-sort ID
|
||
self.keyword_index = 0
|
||
for row in dungeons:
|
||
self.keyword_index += 1
|
||
row['id'] = self.keyword_index
|
||
yield row
|