mirror of
https://github.com/TheFunny/ArisuAutoSweeper
synced 2025-12-16 22:05:12 +00:00
52 lines
1.7 KiB
Python
52 lines
1.7 KiB
Python
from module.base.timer import Timer
|
|
from module.logger import logger
|
|
from module.ocr.ocr import Digit, DigitCounter
|
|
from tasks.base.page import page_work
|
|
from tasks.base.ui import UI
|
|
from tasks.item.assets.assets_item_data import *
|
|
|
|
|
|
class DataUpdate(UI):
|
|
def _get_data(self):
|
|
"""
|
|
Page:
|
|
in: page_work
|
|
"""
|
|
ap, _, ap_total = DigitCounter(OCR_AP).ocr_single_line(self.device.image)
|
|
if ap_total == 0:
|
|
logger.warning('Invalid AP')
|
|
return False
|
|
# Data for Credit and Pyroxene
|
|
ocr = Digit(OCR_DATA)
|
|
timeout = Timer(2, count=6).start()
|
|
while 1:
|
|
data = ocr.detect_and_ocr(self.device.image)
|
|
if len(data) != 2:
|
|
data = [data[0], data[-1]]
|
|
logger.attr('Data', data)
|
|
credit, pyroxene = [int(''.join(filter(lambda x: x.isdigit(), d.ocr_text))) for d in data]
|
|
if credit > 0 or pyroxene > 0:
|
|
break
|
|
|
|
logger.warning(f'Invalid credit and pyroxene: {data}')
|
|
if timeout.reached():
|
|
logger.warning('Get data timeout')
|
|
return False
|
|
|
|
logger.attr('Credit', credit)
|
|
logger.attr('Pyroxene', pyroxene)
|
|
with self.config.multi_set():
|
|
self.config.stored.AP.set(ap, ap_total)
|
|
self.config.stored.Credit.value = credit
|
|
self.config.stored.Pyroxene.value = pyroxene
|
|
|
|
return True
|
|
|
|
def run(self):
|
|
self.ui_ensure(page_work, acquire_lang_checked=False)
|
|
|
|
if self._get_data():
|
|
self.config.task_delay(server_update=True)
|
|
else:
|
|
self.config.task_delay(minute=1)
|