1
0
mirror of https://gitee.com/sui-feng-cb/AzurLaneAutoScript1 synced 2026-03-12 05:08:19 +08:00

Fix: Ignore suffix in research names (#153)

This commit is contained in:
LmeSzinc
2020-08-12 02:47:42 +08:00
parent ffb08bbbdc
commit 9dbbfc5938

View File

@@ -96,7 +96,7 @@ class ResearchProject:
matched = False
for data in LIST_RESEARCH_PROJECT:
if data['name'] == self.name and data['series'] == series:
if self.name_equal(data['name']) and data['series'] == series:
matched = True
self.data = data
self.genre = data['name'][0]
@@ -140,6 +140,20 @@ class ResearchProject:
return '-'.join([prefix, number, suffix])
return name
def name_equal(self, name):
"""
Ignore suffix like `-MI`, `-RF`, `-UL`, only match names like `D-022`
Args:
name (str):
Returns:
bool:
"""
name1 = self.name.rstrip('MIRFUL-')
name2 = name.rstrip('MIRFUL-')
return name1 == name2
class ResearchSelector(UI):
projects: list