2023-08-11 09:35:57 +00:00
|
|
|
|
import re
|
|
|
|
|
|
|
2023-08-14 07:12:02 +00:00
|
|
|
|
from module.base.base import ModuleBase
|
|
|
|
|
|
from module.base.utils import area_offset
|
|
|
|
|
|
from module.ocr.ocr import OcrResultButton
|
|
|
|
|
|
|
2023-08-11 09:35:57 +00:00
|
|
|
|
REGEX_PUNCTUATION = re.compile(r'[ ,.\'"“”，。:：!！?？·•—/()（）「」『』【】《》]')
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def parse_name(n):
|
|
|
|
|
|
n = REGEX_PUNCTUATION.sub('', str(n)).lower()
|
|
|
|
|
|
return n
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def get_regex_from_keyword_name(keyword, attr_name):
|
|
|
|
|
|
string = ""
|
|
|
|
|
|
for instance in keyword.instances.values():
|
|
|
|
|
|
if hasattr(instance, attr_name):
|
|
|
|
|
|
for name in instance.__getattribute__(attr_name):
|
|
|
|
|
|
string += f"{name}|"
|
|
|
|
|
|
# some pattern contain each other, make sure each pattern end with "-" or the end of string
|
|
|
|
|
|
return f"(?:({string[:-1]})(?:-|$))?"
|
2023-08-14 07:12:02 +00:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def is_card_selected(main: ModuleBase, target: OcrResultButton, confirm_button):
|
|
|
|
|
|
"""
|
|
|
|
|
|
There is a white border if a blessing is selected.
|
|
|
|
|
|
For the enforce case, just check the confirm button turning to white
|
|
|
|
|
|
"""
|
|
|
|
|
|
if not target:
|
|
|
|
|
|
return main.image_color_count(confirm_button, (230, 230, 230))
|
|
|
|
|
|
top_border = area_offset(target.area, (0, -180))
|
|
|
|
|
|
return main.image_color_count(top_border, (255, 255, 255))
|
