YoursFunny/ArisuAutoSweeper
1
0
Fork 0
mirror of https://github.com/TheFunny/ArisuAutoSweeper synced 2025-12-16 22:05:12 +00:00
Code Releases Wiki Activity

fix(ocr): force substitute % with / in DigitCounter

Browse Source
This commit is contained in:
YoursFunny 2024-05-05 16:08:35 +08:00
parent bd0749c058
commit b2ae30b45c
Signed by: YoursFunny
GPG Key ID: 207EDC3CD5B40F9C
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
module/ocr/ocr.py
View File

@ -349,6 +349,11 @@ class DigitCounter(Ocr):
def __init__(self, button: ButtonWrapper, lang='en', name=None): def __init__(self, button: ButtonWrapper, lang='en', name=None):
super().__init__(button, lang=lang, name=name) super().__init__(button, lang=lang, name=name)
def after_process(self, result):
result = super().after_process(result)
result = result.replace('%', '/')
return result
def format_result(self, result) -> tuple[int, int, int]: def format_result(self, result) -> tuple[int, int, int]:
""" """
Do OCR on a counter, such as `14/15`, and returns 14, 1, 15 Do OCR on a counter, such as `14/15`, and returns 14, 1, 15
@ -356,7 +361,7 @@ class DigitCounter(Ocr):
Returns: Returns:
int: int:
""" """
result = super().after_process(result) result = self.after_process(result)
logger.attr(name=self.name, text=str(result)) logger.attr(name=self.name, text=str(result))
res = re.search(r'(\d+)/(\d+)', result) res = re.search(r'(\d+)/(\d+)', result)