mirror of
https://gitee.com/sui-feng-cb/AzurLaneAutoScript1
synced 2026-03-19 00:13:35 +08:00
Add: Claim mail rewards
This commit is contained in:
LmeSzinc
221
module/freebies/mail_white.py
Normal file
221
module/freebies/mail_white.py
Normal file
@@ -0,0 +1,221 @@
|
||||
from module.base.decorator import cached_property
|
||||
from module.combat.assets import GET_ITEMS_1, GET_ITEMS_2
|
||||
from module.freebies.assets import *
|
||||
from module.logger import logger
|
||||
from module.ui.page import GOTO_MAIN_WHITE, page_mail, page_main
|
||||
from module.ui.setting import Setting
|
||||
from module.ui.ui import UI
|
||||
|
||||
|
||||
class MailSelectSetting(Setting):
|
||||
def is_option_active(self, option: Button) -> bool:
|
||||
return self.main.image_color_count(option, color=(57, 56, 57), threshold=221, count=50)
|
||||
|
||||
|
||||
class MailWhite(UI):
|
||||
@cached_property
|
||||
def mail_select_setting(self):
|
||||
setting = MailSelectSetting('Mail', main=self)
|
||||
setting.reset_first = False
|
||||
setting.need_deselect = True
|
||||
setting.add_setting(
|
||||
setting='contains',
|
||||
option_buttons=[MAIL_SELECT_CUBE, MAIL_SELECT_COINS, MAIL_SELECT_OIL, MAIL_SELECT_MERIT, MAIL_SELECT_GEMS],
|
||||
option_names=['cube', 'coins', 'oil', 'merit', 'gems'],
|
||||
option_default='merit'
|
||||
)
|
||||
return setting
|
||||
|
||||
def _mail_enter(self, skip_first_screenshot=True):
|
||||
"""
|
||||
Page:
|
||||
in: page_main_white or MAIL_MANAGE
|
||||
out: MAIL_BATCH_CLAIM
|
||||
"""
|
||||
logger.info('Mail enter')
|
||||
self.interval_clear([
|
||||
MAIL_MANAGE
|
||||
])
|
||||
while 1:
|
||||
if skip_first_screenshot:
|
||||
skip_first_screenshot = False
|
||||
else:
|
||||
self.device.screenshot()
|
||||
|
||||
# End
|
||||
if self.appear(MAIL_BATCH_CLAIM, offset=(20, 20)):
|
||||
logger.info('Mail entered')
|
||||
break
|
||||
|
||||
# Click
|
||||
if self.appear_then_click(MAIL_MANAGE, offset=(30, 30), interval=3):
|
||||
continue
|
||||
if self.ui_main_appear_then_click(page_mail, offset=(30, 30), interval=3):
|
||||
continue
|
||||
|
||||
def _mail_quit(self, skip_first_screenshot=True):
|
||||
"""
|
||||
Page:
|
||||
in: Any page in page_mail
|
||||
out: page_main_white
|
||||
"""
|
||||
logger.info('Mail quit')
|
||||
self.interval_clear([
|
||||
MAIL_BATCH_CLAIM,
|
||||
GOTO_MAIN_WHITE,
|
||||
GET_ITEMS_1,
|
||||
GET_ITEMS_2,
|
||||
])
|
||||
self.popup_interval_clear()
|
||||
while 1:
|
||||
if skip_first_screenshot:
|
||||
skip_first_screenshot = False
|
||||
else:
|
||||
self.device.screenshot()
|
||||
|
||||
# End
|
||||
if self.ui_page_appear(page_main):
|
||||
logger.info('Mail quit to page_main')
|
||||
break
|
||||
|
||||
# Click
|
||||
if self.handle_popup_confirm('MAIL_QUIT'):
|
||||
continue
|
||||
if self.appear(MAIL_BATCH_CLAIM, offset=(30, 30), interval=3):
|
||||
logger.info(f'{MAIL_BATCH_CLAIM} -> {MAIL_MANAGE}')
|
||||
self.device.click(MAIL_MANAGE)
|
||||
continue
|
||||
if self.appear_then_click(GOTO_MAIN_WHITE, offset=(30, 30), interval=3):
|
||||
continue
|
||||
if self._handle_mail_reward():
|
||||
continue
|
||||
|
||||
def _handle_mail_reward(self):
|
||||
if self.appear(GET_ITEMS_1, offset=(30, 30), interval=3):
|
||||
logger.info(f'{GET_ITEMS_1} -> {MAIL_BATCH_CLAIM}')
|
||||
self.device.click(MAIL_BATCH_CLAIM)
|
||||
return True
|
||||
if self.appear(GET_ITEMS_2, offset=(30, 30), interval=3):
|
||||
logger.info(f'{GET_ITEMS_2} -> {MAIL_BATCH_CLAIM}')
|
||||
self.device.click(MAIL_BATCH_CLAIM)
|
||||
return True
|
||||
return False
|
||||
|
||||
def _mail_claim_execute(self, skip_first_screenshot=True):
|
||||
"""
|
||||
Page:
|
||||
in: MAIL_BATCH_CLAIM
|
||||
out: page_main_white, may have info_bar
|
||||
|
||||
Returns:
|
||||
int: If success to claim
|
||||
"""
|
||||
self.handle_info_bar()
|
||||
self.interval_clear([
|
||||
MAIL_BATCH_CLAIM,
|
||||
GET_ITEMS_1,
|
||||
GET_ITEMS_2,
|
||||
])
|
||||
self.popup_interval_clear()
|
||||
|
||||
claimed = False
|
||||
while 1:
|
||||
if skip_first_screenshot:
|
||||
skip_first_screenshot = False
|
||||
else:
|
||||
self.device.screenshot()
|
||||
|
||||
# End
|
||||
if claimed and self.appear(MAIL_BATCH_CLAIM, offset=(30, 30)):
|
||||
break
|
||||
# Click
|
||||
if not claimed and self.appear_then_click(MAIL_BATCH_CLAIM, offset=(30, 30), interval=3):
|
||||
continue
|
||||
if self.handle_popup_confirm('MAIL_CLAIM'):
|
||||
claimed = True
|
||||
continue
|
||||
if self._handle_mail_reward():
|
||||
claimed = True
|
||||
continue
|
||||
|
||||
success = self.info_bar_count() > 0
|
||||
logger.info(f'Mail claim success: {success}')
|
||||
return success
|
||||
|
||||
def _mail_delete(self, skip_first_screenshot=True):
|
||||
"""
|
||||
Pages:
|
||||
in: MAIL_BATCH_DELETE
|
||||
out: MAIL_BATCH_DELETE
|
||||
"""
|
||||
self.handle_info_bar()
|
||||
self.interval_clear([
|
||||
MAIL_BATCH_DELETE
|
||||
])
|
||||
self.popup_interval_clear()
|
||||
|
||||
deleted = False
|
||||
while 1:
|
||||
if skip_first_screenshot:
|
||||
skip_first_screenshot = False
|
||||
else:
|
||||
self.device.screenshot()
|
||||
|
||||
# End
|
||||
if deleted and self.appear(MAIL_BATCH_DELETE, offset=(30, 30)):
|
||||
break
|
||||
# Click
|
||||
if not deleted and self.appear_then_click(MAIL_BATCH_DELETE, offset=(30, 30), interval=3):
|
||||
continue
|
||||
if self.handle_popup_confirm('MAIL_CLAIM'):
|
||||
deleted = True
|
||||
continue
|
||||
|
||||
# info_bar appears if mail success to delete and no mail deleted
|
||||
return True
|
||||
|
||||
def mail_claim(
|
||||
self,
|
||||
merit=True,
|
||||
maintenance=True,
|
||||
trade_license=True,
|
||||
delete=True,
|
||||
):
|
||||
"""
|
||||
Pages:
|
||||
in: page_main_white or MAIL_MANAGE
|
||||
out: MAIL_BATCH_CLAIM
|
||||
"""
|
||||
if merit:
|
||||
logger.hr('Mail merit', level=2)
|
||||
self._mail_enter()
|
||||
self.mail_select_setting.set(contains=['merit'])
|
||||
self._mail_claim_execute()
|
||||
if maintenance:
|
||||
logger.hr('Mail maintenance', level=2)
|
||||
self._mail_enter()
|
||||
self.mail_select_setting.set(contains=['coins', 'oil'])
|
||||
self._mail_claim_execute()
|
||||
self._mail_enter()
|
||||
self.mail_select_setting.set(contains=['coins', 'oil', 'gems'])
|
||||
self._mail_claim_execute()
|
||||
if trade_license:
|
||||
logger.hr('Mail trade license', level=2)
|
||||
self._mail_enter()
|
||||
self.mail_select_setting.set(contains=['coins', 'oil', 'cube'])
|
||||
self._mail_claim_execute()
|
||||
if delete:
|
||||
logger.hr('Mail delete', level=2)
|
||||
self._mail_enter()
|
||||
self._mail_delete()
|
||||
|
||||
self._mail_quit()
|
||||
|
||||
def run(self):
|
||||
pass
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
self = MailWhite('alas')
|
||||
self.device.screenshot()
|
||||
self.mail_claim()
|
||||
Reference in New Issue
Block a user