mirror of
https://github.com/LmeSzinc/StarRailCopilot.git
synced 2024-12-04 20:47:33 +00:00
83 lines
2.7 KiB
Python
83 lines
2.7 KiB
Python
import typing as t
|
|
|
|
from dev_tools.keywords.base import GenerateKeyword
|
|
from module.base.decorator import cached_property
|
|
from module.config.utils import deep_get
|
|
|
|
|
|
class GenerateMapPlane(GenerateKeyword):
|
|
output_file = './tasks/map/keywords/plane.py'
|
|
|
|
@cached_property
|
|
def data(self):
|
|
return self.read_file('./ExcelOutput/AreaMapConfig.json')
|
|
|
|
def iter_planes(self) -> t.Iterable[dict]:
|
|
for data in self.data:
|
|
plane_id = int(deep_get(data, 'ID', 0))
|
|
world_id = int(str(plane_id)[-5])
|
|
sort_id = int(deep_get(data, 'MenuSortID', 0))
|
|
text_id = deep_get(data, 'Name.Hash')
|
|
yield dict(
|
|
text_id=text_id,
|
|
world_id=world_id,
|
|
plane_id=plane_id,
|
|
sort_id=sort_id,
|
|
)
|
|
|
|
def iter_keywords(self) -> t.Iterable[dict]:
|
|
"""
|
|
1010201
|
|
^^ floor
|
|
^^ plane
|
|
^ world
|
|
"""
|
|
def to_id(name):
|
|
return self.find_keyword(name, lang='cn')[0]
|
|
|
|
domains = ['黑塔的办公室', '锋芒崭露']
|
|
for index, domain in enumerate(domains):
|
|
yield dict(
|
|
text_id=to_id(domain),
|
|
world_id=-1,
|
|
plane_id=index + 1,
|
|
)
|
|
domains = ['区域-战斗', '区域-事件', '区域-遭遇', '区域-休整', '区域-精英', '区域-首领', '区域-交易']
|
|
for index, domain in enumerate(domains):
|
|
yield dict(
|
|
text_id=to_id(domain),
|
|
world_id=-2,
|
|
plane_id=index + 1,
|
|
)
|
|
domains = ['晖长石号', '开拓之尾号', '塔塔洛夫号', '飞翔时针号']
|
|
for index, domain in enumerate(domains):
|
|
yield dict(
|
|
text_id=to_id(domain),
|
|
world_id=3,
|
|
plane_id=index + 1,
|
|
)
|
|
|
|
keywords = sorted(self.iter_planes(), key=lambda x: (x['world_id'], x['sort_id']))
|
|
for keyword in keywords:
|
|
keyword.pop('sort_id')
|
|
yield keyword
|
|
|
|
def convert_name(self, text: str, keyword: dict) -> str:
|
|
text = super().convert_name(text, keyword=keyword)
|
|
text = text.replace('_', '')
|
|
if not text:
|
|
return ""
|
|
from tasks.map.keywords import MapWorld
|
|
world = MapWorld.find_world_id(keyword['world_id'])
|
|
if world is None:
|
|
if text.startswith('Domain'):
|
|
return f'Rogue_{text}'
|
|
else:
|
|
return f'Special_{text}'
|
|
else:
|
|
return f'{world.short_name}_{text}'
|
|
|
|
def convert_keyword(self, text: str, lang: str) -> str:
|
|
text = text.replace('™', '')
|
|
return super().convert_keyword(text, lang=lang)
|