mirror of
https://github.com/TheFunny/ArisuAutoSweeper
synced 2025-12-17 00:15:13 +00:00
55 lines
1.6 KiB
Python
55 lines
1.6 KiB
Python
from module.logger import logger
|
|
from module.ocr.ocr import DigitCounter
|
|
from module.ui.switch import Switch
|
|
from tasks.base.ui import UI
|
|
from tasks.tactical_challenge.assets.assets_tactical_challenge import *
|
|
|
|
SWITCH_SKIP = Switch('Skip_switch')
|
|
SWITCH_SKIP.add_state('on', SKIP_ON)
|
|
SWITCH_SKIP.add_state('off', SKIP_OFF)
|
|
|
|
|
|
class TacticalChallengeUI(UI):
|
|
def get_ticket(self):
|
|
"""
|
|
Page:
|
|
in: page_tactical_challenge
|
|
"""
|
|
ocr = DigitCounter(OCR_TICKET).ocr_single_line(self.device.image)
|
|
# number of tickets remaining
|
|
ticket, _, total = ocr
|
|
if total == 0:
|
|
logger.warning('Invalid ticket')
|
|
return False, 5
|
|
logger.attr('Ticket', ticket)
|
|
|
|
return True, ticket
|
|
|
|
def get_reward(self):
|
|
if self.match_color(GET_REWARD_DAILY):
|
|
self.device.click(GET_REWARD_DAILY)
|
|
logger.info('Get tc daily reward')
|
|
return True
|
|
if self.match_color(GET_REWARD_CREDIT):
|
|
self.device.click(GET_REWARD_CREDIT)
|
|
logger.info('Get tc credit reward')
|
|
return True
|
|
if self.match_color(GOT_REWARD_DAILY) and self.match_color(GOT_REWARD_CREDIT):
|
|
logger.info('Both tc reward got')
|
|
return True
|
|
|
|
return False
|
|
|
|
def set_skip(self):
|
|
"""
|
|
Set skip switch to on
|
|
Returns:
|
|
True if switch is set, False if switch not found
|
|
"""
|
|
if not SWITCH_SKIP.appear(main=self):
|
|
logger.info('Skip switch not found')
|
|
return False
|
|
SWITCH_SKIP.set('on', main=self)
|
|
|
|
return True
|