mirror of
https://github.com/LmeSzinc/StarRailCopilot.git
synced 2024-11-27 02:27:12 +00:00
86 lines
3.4 KiB
Python
86 lines
3.4 KiB
Python
|
import re
|
||
|
|
|
||
|
|
from module.exception import ScriptError
|
||
|
|
from module.logger import logger
|
||
|
|
from tasks.base.ui import UI
|
||
|
|
from tasks.combat.assets.assets_combat_interact import DUNGEON_COMBAT_INTERACT, DUNGEON_COMBAT_INTERACT_TEXT
|
||
|
|
from tasks.dungeon.keywords import (
|
||
|
|
DungeonList,
|
||
|
|
DungeonNav,
|
||
|
|
KEYWORDS_DUNGEON_LIST,
|
||
|
|
KEYWORDS_DUNGEON_NAV
|
||
|
|
)
|
||
|
|
from tasks.dungeon.ui.llist import OcrDungeonName
|
||
|
|
|
||
|
|
|
||
|
|
class DungeonUIInteract(UI):
|
||
|
|
def get_dungeon_interact(self) -> DungeonList | None:
|
||
|
|
"""
|
||
|
|
Pages:
|
||
|
|
in: page_main
|
||
|
|
"""
|
||
|
|
if not self.appear(DUNGEON_COMBAT_INTERACT):
|
||
|
|
logger.info('No dungeon interact')
|
||
|
|
return None
|
||
|
|
|
||
|
|
self.acquire_lang_checked()
|
||
|
|
|
||
|
|
ocr = OcrDungeonName(DUNGEON_COMBAT_INTERACT_TEXT)
|
||
|
|
result = ocr.detect_and_ocr(self.device.image)
|
||
|
|
|
||
|
|
dungeon = None
|
||
|
|
# Special match names in English
|
||
|
|
# Second row must have at least 3 characters which is the shortest name "Ire"
|
||
|
|
# Stangnant Shadow: Shape of
|
||
|
|
# Quanta
|
||
|
|
if len(result) == 2 and len(result[1].ocr_text) >= 3:
|
||
|
|
first, second = result[0].ocr_text, result[1].ocr_text
|
||
|
|
if re.search(r'Stagnant\s*Shadow', first):
|
||
|
|
dungeon = DungeonList.find_dungeon_by_string(en=second, is_Stagnant_Shadow=True)
|
||
|
|
elif re.search(r'Cavern\s*of\s*Corrosion', first):
|
||
|
|
dungeon = DungeonList.find_dungeon_by_string(en=second, is_Cavern_of_Corrosion=True)
|
||
|
|
elif re.search(r'Echo\s*of\s*War', first):
|
||
|
|
dungeon = DungeonList.find_dungeon_by_string(en=second, is_Echo_of_War=True)
|
||
|
|
elif re.search(r'Calyx[\s(]+Golden', first):
|
||
|
|
dungeon = DungeonList.find_dungeon_by_string(en=second, is_Calyx_Golden=True, world=self.plane.world)
|
||
|
|
elif re.search(r'Calyx[\s(]+Crimson', first):
|
||
|
|
dungeon = DungeonList.find_dungeon_by_string(en=second, is_Calyx_Crimson=True, plane=self.plane)
|
||
|
|
if dungeon is not None:
|
||
|
|
logger.attr('DungeonInteract', dungeon)
|
||
|
|
return dungeon
|
||
|
|
|
||
|
|
# Join
|
||
|
|
result = ' '.join([row.ocr_text for row in result])
|
||
|
|
|
||
|
|
# Special match names in Chinese
|
||
|
|
# Only calyxes need spacial match
|
||
|
|
if res := re.search(r'(^.+之蕾)', result):
|
||
|
|
dungeon = DungeonList.find_dungeon_by_string(cn=res.group(1), is_Calyx_Crimson=True, plane=self.plane)
|
||
|
|
if dungeon is not None:
|
||
|
|
logger.attr('DungeonInteract', dungeon)
|
||
|
|
return dungeon
|
||
|
|
dungeon = DungeonList.find_dungeon_by_string(cn=res.group(1), is_Calyx_Golden=True, world=self.plane.world)
|
||
|
|
if dungeon is not None:
|
||
|
|
logger.attr('DungeonInteract', dungeon)
|
||
|
|
return dungeon
|
||
|
|
|
||
|
|
# Dungeons
|
||
|
|
try:
|
||
|
|
dungeon = DungeonList.find(result)
|
||
|
|
logger.attr('DungeonInteract', dungeon)
|
||
|
|
return dungeon
|
||
|
|
except ScriptError:
|
||
|
|
pass
|
||
|
|
# Simulated Universe returns Simulated_Universe_World_1
|
||
|
|
try:
|
||
|
|
dungeon = DungeonNav.find(result)
|
||
|
|
if dungeon == KEYWORDS_DUNGEON_NAV.Simulated_Universe:
|
||
|
|
dungeon = KEYWORDS_DUNGEON_LIST.Simulated_Universe_World_1
|
||
|
|
logger.attr('DungeonInteract', dungeon)
|
||
|
|
return dungeon
|
||
|
|
except ScriptError:
|
||
|
|
pass
|
||
|
|
# Unknown
|
||
|
|
logger.attr('DungeonInteract', None)
|
||
|
|
return None
|