AzurLaneAutoScript/module/shop/shop_reward.py

from module.logger import logger
from module.shop.shop_core import CoreShop
from module.shop.shop_general import GeneralShop
from module.shop.shop_guild import GuildShop
from module.shop.shop_medal import MedalShop2
from module.shop.shop_merit import MeritShop
from module.shop.ui import ShopUI


class RewardShop(ShopUI):
    def run_frequent(self):
        # Munitions shops
        self.ui_goto_shop()

        GeneralShop(self.config, self.device).run()

        self.config.task_delay(server_update=True)

    def run_once(self):
        # Munitions shops
        self.ui_goto_shop()

        current = self._shop_bottom_navbar.get_active(main=self)
        logger.info(f'Shop nav actives at left={current}')

        if current is not None:
            current += 1
            # prototype, core, merit, general, event
            # core, merit, general, event, event (2 event shops sometimes)
            if self.shop_bottom_navbar_ensure(left=current - 1):
                MeritShop(self.config, self.device).run()

            if self.shop_bottom_navbar_ensure(left=current - 2):
                CoreShop(self.config, self.device).run()
        else:
            logger.error('Cannot get the position of general shop, skip merit shop and core shop')

        self.shop_swipe()
        # meta, medal, guild, prototype, core
        if self.shop_bottom_navbar_ensure(left=3):
            GuildShop(self.config, self.device).run()

        if self.shop_bottom_navbar_ensure(left=2):
            MedalShop2(self.config, self.device).run()

        # Cannot go back to general shop so don't stay in page_munitions
        self.ui_goto_main()
        self.config.task_delay(server_update=True)