Example #1
0
class _PrebattleDispatcher(ListenersCollection):
    def __init__(self):
        super(_PrebattleDispatcher, self).__init__()
        self.__requestCtx = None
        self.__collection = FunctionalCollection()
        self.__factories = ControlFactoryDecorator()
        self.__nextPrbFunctional = None
        self._setListenerClass(IGlobalListener)
        return

    def __del__(self):
        LOG_DEBUG('_PrebattleDispatcher deleted')

    def start(self):
        g_eventDispatcher.init(self)
        result = self.__setFunctional(CreateFunctionalCtx())
        self.__requestCtx = PrbCtrlRequestCtx()
        self.__startListening()
        functional.initDevFunctional()
        if result & FUNCTIONAL_FLAG.LOAD_PAGE == 0:
            BigWorld.callback(0.001, lambda: g_eventDispatcher.loadHangar())
        g_eventDispatcher.updateUI()
        if GUI_SETTINGS.specPrebatlesVisible and not prb_getters.areSpecBattlesHidden(
        ):
            g_eventDispatcher.addSpecBattlesToCarousel()

    def stop(self):
        self.__nextPrbFunctional = None
        self.__stopListening()
        functional.finiDevFunctional()
        self.__clear(woEvents=True)
        g_eventDispatcher.fini()
        return

    def getPrbFunctional(self):
        return self.__collection.getItem(_CTRL_TYPE.PREBATTLE)

    def getUnitFunctional(self):
        return self.__collection.getItem(_CTRL_TYPE.UNIT)

    def getPreQueueFunctional(self):
        return self.__collection.getItem(_CTRL_TYPE.PREQUEUE)

    def getFunctional(self, ctrlType):
        return self.__collection.getItem(ctrlType)

    def getFunctionalCollection(self):
        return self.__collection

    def getControlFactories(self):
        return self.__factories

    def hasModalEntity(self):
        return self.__collection.hasModalEntity()

    def getFunctionalState(self):
        return self.__collection.getState(self.__factories)

    @async
    @process
    def create(self, ctx, callback=None):
        if ctx.getRequestType() == _RQ_TYPE.CREATE:
            if not self.__requestCtx.isProcessing():
                result = yield self.unlock(ctx)
                if result:
                    entry = self.__factories.createEntry(ctx)
                    if entry:
                        LOG_DEBUG('Request to create', ctx)
                        self.__requestCtx = ctx
                        entry.create(ctx, callback=callback)
                    else:
                        LOG_ERROR('Entry not found', ctx)
                        if callback is not None:
                            callback(False)
                elif callback is not None:
                    callback(False)
            else:
                LOG_ERROR('Request is processing', self.__requestCtx)
                if callback is not None:
                    callback(False)
                yield lambda callback=None: callback
        else:
            LOG_ERROR('Invalid context to create', ctx)
            if callback is not None:
                callback(False)
            yield lambda callback=None: callback
        return

    @async
    @process
    def join(self, ctx, callback=None):
        if self.__validateJoinOp(ctx):
            result = yield self.unlock(ctx)
            ctx.setForced(result)
            if result:
                entry = self.__factories.createEntry(ctx)
                if entry:
                    LOG_DEBUG('Request to join', ctx)
                    self.__requestCtx = ctx
                    entry.join(ctx, callback=callback)
                else:
                    LOG_ERROR('Entry not found', ctx)
                    if callback is not None:
                        callback(False)
        else:
            if callback is not None:
                callback(False)
            yield lambda callback=None: callback
        return

    @async
    def leave(self, ctx, callback=None):
        if ctx.getRequestType() != _RQ_TYPE.LEAVE:
            LOG_ERROR('Invalid context to leave prebattle/unit', ctx)
            if callback is not None:
                callback(False)
            return
        elif self.__requestCtx.isProcessing():
            LOG_ERROR('Request is processing', self.__requestCtx)
            if callback is not None:
                callback(False)
            return
        else:
            ctrlType = ctx.getCtrlType()
            formation = self.__collection.getItem(ctx.getCtrlType())
            if formation is not None:
                if formation.hasLockedState():
                    entityType = formation.getEntityType()
                    SystemMessages.pushI18nMessage(
                        messages.getLeaveDisabledMessage(ctrlType, entityType),
                        type=SystemMessages.SM_TYPE.Warning)
                    if callback is not None:
                        callback(False)
                    return
                LOG_DEBUG('Request to leave formation', ctx)
                self.__requestCtx = ctx
                formation.leave(ctx, callback=callback)
            else:
                LOG_ERROR('Functional not found', ctx)
                if callback is not None:
                    callback(False)
            return

    @async
    @process
    def unlock(self, unlockCtx, callback=None):
        state = self.getFunctionalState()
        result = True
        if not state.isIntroMode:
            canDoLeave = True
        elif unlockCtx.hasFlags(FUNCTIONAL_FLAG.SWITCH):
            if state.ctrlTypeID == unlockCtx.getCtrlType(
            ) and state.entityTypeID != unlockCtx.getEntityType():
                canDoLeave = True
                unlockCtx.removeFlags(FUNCTIONAL_FLAG.SWITCH)
            else:
                canDoLeave = False
        else:
            canDoLeave = True
        if canDoLeave:
            factory = self.__factories.get(state.ctrlTypeID)
            result = False
            if factory:
                ctx = factory.createLeaveCtx(unlockCtx.getFlags())
                if ctx:
                    meta = self.__collection.getConfirmDialogMeta(
                        state.ctrlTypeID, unlockCtx)
                    if meta:
                        result = yield DialogsInterface.showDialog(meta)
                    else:
                        result = True
                    if result:
                        result = yield self.leave(ctx)
                else:
                    LOG_ERROR('Can not create leave ctx', state)
            else:
                LOG_ERROR('Factory is not found', state)
        if callback is not None:
            callback(result)
        yield lambda callback=None: callback
        return

    @async
    @process
    def select(self, entry, callback=None):
        ctx = entry.makeDefCtx()
        if ctx and self.__validateJoinOp(ctx):
            result = yield self.unlock(ctx)
            ctx.setForced(result)
            if result:
                LOG_DEBUG('Request to select', ctx)
                self.__requestCtx = ctx
                entry.select(ctx, callback=callback)
            elif callback is not None:
                callback(False)
        else:
            if callback is not None:
                callback(False)
            yield lambda callback=None: callback
        return

    @async
    def sendPrbRequest(self, ctx, callback=None):
        prbFunctional = self.getFunctional(_CTRL_TYPE.PREBATTLE)
        if prbFunctional:
            prbFunctional.request(ctx, callback=callback)
        else:
            LOG_ERROR('prbFunctional is not found', ctx)
            if callback is not None:
                callback(False)
        return

    @async
    def sendUnitRequest(self, ctx, callback=None):
        unitFunctional = self.getFunctional(_CTRL_TYPE.UNIT)
        if unitFunctional:
            unitFunctional.request(ctx, callback=callback)
        else:
            LOG_ERROR('unitFunctional is not found', ctx)
            if callback is not None:
                callback(False)
        return

    @async
    def sendPreQueueRequest(self, ctx, callback=None):
        preQueueFunctional = self.getFunctional(_CTRL_TYPE.PREQUEUE)
        if preQueueFunctional is not None:
            preQueueFunctional.request(ctx, callback=callback)
        else:
            LOG_ERROR('preQueueFunctional is not found', ctx)
            if callback is not None:
                callback(False)
        return

    def exitFromQueue(self):
        self.__collection.exitFromQueue()

    def canPlayerDoAction(self):
        canDo, restriction = self.__collection.canPlayerDoAction(False)
        if canDo:
            if g_currentPreviewVehicle.isPresent():
                canDo = False
                restriction = PREBATTLE_RESTRICTION.PREVIEW_VEHICLE_IS_PRESENT
            elif not g_currentVehicle.isReadyToFight():
                if not g_currentVehicle.isPresent():
                    canDo = False
                    restriction = PREBATTLE_RESTRICTION.VEHICLE_NOT_PRESENT
                elif g_currentVehicle.isInBattle():
                    canDo = False
                    restriction = PREBATTLE_RESTRICTION.VEHICLE_IN_BATTLE
                elif not g_currentVehicle.isCrewFull():
                    canDo = False
                    restriction = PREBATTLE_RESTRICTION.CREW_NOT_FULL
                elif g_currentVehicle.isBroken():
                    canDo = False
                    restriction = PREBATTLE_RESTRICTION.VEHICLE_BROKEN
                elif g_currentVehicle.isFalloutOnly(
                ) and not game_control.getFalloutCtrl().isSelected():
                    canDo = False
                    restriction = PREBATTLE_RESTRICTION.VEHICLE_FALLOUT_ONLY
                elif g_currentVehicle.isDisabledInRoaming():
                    canDo = False
                    restriction = PREBATTLE_RESTRICTION.VEHICLE_ROAMING
                elif g_currentVehicle.isDisabledInPremIGR():
                    canDo = False
                    restriction = PREBATTLE_RESTRICTION.VEHICLE_IN_PREMIUM_IGR_ONLY
                elif g_currentVehicle.isDisabledInRent():
                    unit = self.getUnitFunctional()
                    if unit is not None and unit.getFlags().isInPreArena():
                        canDo = True
                    else:
                        canDo = False
                        if g_currentVehicle.isPremiumIGR():
                            restriction = PREBATTLE_RESTRICTION.VEHICLE_IGR_RENTALS_IS_OVER
                        else:
                            restriction = PREBATTLE_RESTRICTION.VEHICLE_RENTALS_IS_OVER
            if canDo:
                canDo, restriction = self.__collection.canPlayerDoAction(True)
        return (canDo, restriction)

    def getPlayerInfo(self):
        return self.__collection.getPlayerInfo(self.__factories)

    def doAction(self, action=None):
        if not g_currentVehicle.isPresent():
            SystemMessages.pushMessage(messages.getInvalidVehicleMessage(
                PREBATTLE_RESTRICTION.VEHICLE_NOT_PRESENT),
                                       type=SystemMessages.SM_TYPE.Error)
            return False
        LOG_DEBUG('Do GUI action', action)
        return self.__collection.doAction(self.__factories, action)

    def doSelectAction(self, action):
        result = self.__collection.doSelectAction(action)
        if result.isProcessed:
            if result.newEntry is not None:
                self.__doSelect(result.newEntry)
            return True
        else:
            entry = self.__factories.createEntryByAction(action)
            if entry is not None:
                self.__doSelect(entry)
                return True
            return False

    @process
    def doLeaveAction(self, ctx):
        meta = self.__collection.getConfirmDialogMeta(ctx.getCtrlType(), ctx)
        if meta is not None:
            isConfirmed = yield DialogsInterface.showDialog(meta)
        else:
            isConfirmed = yield lambda callback: callback(True)
        if isConfirmed:
            yield self.leave(ctx)
        return

    def getGUIPermissions(self):
        return self.__collection.getGUIPermissions()

    def onCompanyStateChanged(self, _):
        if not self.getPrbFunctional().hasLockedState():
            g_eventDispatcher.updateUI()

    def setRequestCtx(self, ctx):
        result = True
        if self.__requestCtx.isProcessing():
            LOG_ERROR('Request is processing', self.__requestCtx)
            result = False
        else:
            self.__requestCtx = ctx
        return result

    def pe_onArenaJoinFailure(self, errorCode, _):
        self.__collection.reset()
        SystemMessages.pushMessage(messages.getJoinFailureMessage(errorCode),
                                   type=SystemMessages.SM_TYPE.Error)

    def pe_onKickedFromArena(self, reasonCode):
        self.__collection.reset()
        SystemMessages.pushMessage(messages.getKickReasonMessage(reasonCode),
                                   type=SystemMessages.SM_TYPE.Error)

    def pe_onPrebattleAutoInvitesChanged(self):
        if GUI_SETTINGS.specPrebatlesVisible:
            isHidden = prb_getters.areSpecBattlesHidden()
            if isHidden:
                g_eventDispatcher.removeSpecBattlesFromCarousel()
            else:
                g_eventDispatcher.addSpecBattlesToCarousel()
        g_eventDispatcher.updateUI()

    def pe_onPrebattleJoined(self):
        if prb_getters.getClientPrebattle() is not None:
            flags = self.__requestCtx.getFlags()
            self.__setFunctional(
                CreateFunctionalCtx(_CTRL_TYPE.PREBATTLE, flags=flags))
        else:
            LOG_ERROR('ClientPrebattle is not defined')
            self.__requestCtx.stopProcessing(result=False)
        return

    def pe_onPrebattleJoinFailure(self, errorCode):
        SystemMessages.pushMessage(messages.getJoinFailureMessage(errorCode),
                                   type=SystemMessages.SM_TYPE.Error)
        self.__requestCtx.stopProcessing(result=False)
        self.__requestCtx.clearFlags()
        g_eventDispatcher.updateUI()

    def pe_onPrebattleLeft(self):
        if self.__nextPrbFunctional is not None:
            self.__nextPrbFunctional()
            self.__nextPrbFunctional = None
            return
        else:
            flags = self.__requestCtx.getFlags()
            flags |= FUNCTIONAL_FLAG.LEAVE_ENTITY
            prbType = 0
            if flags & FUNCTIONAL_FLAG.SWITCH > 0:
                prbFunctional = self.getFunctional(_CTRL_TYPE.PREBATTLE)
                if prbFunctional is not None:
                    prbType = prbFunctional.getEntityType()
            self.__changePrbFunctional(flags=flags,
                                       prbType=prbType,
                                       stop=False)
            return

    def pe_onKickedFromPrebattle(self, _):
        self.__changePrbFunctional(stop=True)

    def gs_onTillBanNotification(self, isPlayTimeBan, timeTillBlock):
        if prb_getters.isParentControlActivated():
            self.__collection.reset()
            key = '#system_messages:gameSessionControl/korea/{0:>s}'
            if isPlayTimeBan:
                SystemMessages.g_instance.pushI18nMessage(
                    key.format('playTimeNotification'),
                    timeTillBlock,
                    type=SystemMessages.SM_TYPE.Warning)
            else:
                gameSession = game_control.g_instance.gameSession
                notifyStartTime, blockTime = gameSession.getCurfewBlockTime()
                formatter = lambda t: time.strftime('%H:%M', time.localtime(t))
                SystemMessages.g_instance.pushI18nMessage(
                    key.format('midnightNotification'),
                    type=SystemMessages.SM_TYPE.Warning,
                    preBlockTime=formatter(notifyStartTime),
                    blockTime=formatter(blockTime))

    def rc_onRentChange(self, vehicles):
        if g_currentVehicle.isPresent(
        ) and g_currentVehicle.item.intCD in vehicles and g_currentVehicle.isDisabledInRent(
        ) and g_currentVehicle.isInPrebattle():
            self.__collection.reset()

    def igr_onRoomChange(self, roomType, _):
        if roomType != IGR_TYPE.PREMIUM:
            if g_currentVehicle.isPremiumIGR(
            ) and g_currentVehicle.isInPrebattle():
                self.__collection.reset()

    def ctrl_onPrebattleIntroModeJoined(self, prbType, isLeaving):
        flags = self.__requestCtx.getFlags()
        if not isLeaving:
            self.__changePrbFunctional(flags=flags, prbType=prbType)
        else:
            self.__nextPrbFunctional = partial(self.__changePrbFunctional,
                                               flags=flags,
                                               prbType=prbType)

    def ctrl_onPrebattleIntroModeLeft(self):
        flags = self.__requestCtx.getFlags()
        flags |= FUNCTIONAL_FLAG.LEAVE_INTRO
        self.__changePrbFunctional(flags=flags)

    def ctrl_onPrebattleInited(self):
        self.__requestCtx.stopProcessing(result=True)
        self.__setFunctional(
            CreateFunctionalCtx(_CTRL_TYPE.PREBATTLE,
                                flags=self.__requestCtx.getFlags(),
                                initCtx=self.__requestCtx))
        g_eventDispatcher.updateUI()

    def ctrl_onUnitIntroModeJoined(self, prbType, flags):
        self.__changeUnitFunctional(flags=flags, prbType=prbType)

    def ctrl_onUnitIntroModeLeft(self):
        flags = self.__requestCtx.getFlags()
        flags |= FUNCTIONAL_FLAG.LEAVE_INTRO
        self.__changeUnitFunctional(flags=flags)

    def ctrl_onPreQueueFunctionalCreated(self, queueType):
        self.__changePreQueueFunctional(flags=self.__requestCtx.getFlags(),
                                        queueType=queueType)
        g_eventDispatcher.updateUI()

    def ctrl_onPreQueueFunctionalDestroyed(self):
        flags = self.__requestCtx.getFlags()
        flags |= FUNCTIONAL_FLAG.LEAVE_PRE_QUEUE
        self.__changePreQueueFunctional(flags=flags)

    def unitMgr_onUnitJoined(self, unitMgrID, unitIdx):
        unitFunctional = self.getFunctional(_CTRL_TYPE.UNIT)
        if unitFunctional is not None and unitFunctional.getID(
        ) == unitMgrID and unitFunctional.getUnitIdx() == unitIdx:
            unitFunctional.rejoin()
        else:
            if unitFunctional is not None:
                prbType = unitFunctional.getEntityType()
            else:
                prbType = 0
            self.__changeUnitFunctional(flags=self.__requestCtx.getFlags(),
                                        prbType=prbType)
        return

    def unitMgr_onUnitLeft(self, unitMgrID, unitIdx):
        flags = self.__requestCtx.getFlags()
        flags |= FUNCTIONAL_FLAG.LEAVE_ENTITY
        prbType = 0
        if flags & FUNCTIONAL_FLAG.SWITCH > 0:
            unitFunctional = self.getFunctional(_CTRL_TYPE.UNIT)
            if unitFunctional is not None:
                prbType = unitFunctional.getEntityType()
        self.__changeUnitFunctional(flags=flags, prbType=prbType)
        g_eventDispatcher.updateUI()
        return

    def unitMgr_onUnitRestored(self, unitMgrID, unitIdx):
        unitFunctional = self.getFunctional(_CTRL_TYPE.UNIT)
        flags = unitFunctional.getFlags()
        pInfo = unitFunctional.getPlayerInfo()
        if flags.isInPreArena() and pInfo.isInSlot:
            g_eventDispatcher.loadHangar()

    def unitMgr_onUnitErrorReceived(self, requestID, unitMgrID, unitIdx,
                                    errorCode, errorString):
        unitFunctional = self.getFunctional(_CTRL_TYPE.UNIT)
        if unitFunctional is not None:
            unitFunctional.setLastError(errorCode)
            if errorCode in RETURN_INTRO_UNIT_MGR_ERRORS and unitFunctional.canSwitchToIntro(
            ):
                self.__requestCtx.addFlags(FUNCTIONAL_FLAG.SWITCH)
            if errorCode == UNIT_ERROR.CANT_PICK_LEADER:
                self.__requestCtx.removeFlags(FUNCTIONAL_FLAG.SWITCH)
            elif errorCode == UNIT_ERROR.REMOVED_PLAYER:
                if self.__requestCtx.getCtrlType(
                ) == _CTRL_TYPE.UNIT and unitFunctional.getEntityType(
                ) != self.__requestCtx.getEntityType():
                    self.__requestCtx.removeFlags(FUNCTIONAL_FLAG.SWITCH)
        else:
            LOG_ERROR('Unit functional is not found')
        if errorCode not in IGNORED_UNIT_MGR_ERRORS:
            msgType, msgBody = messages.getUnitMessage(errorCode, errorString)
            SystemMessages.pushMessage(msgBody, type=msgType)
            self.__requestCtx.stopProcessing(result=False)
            g_eventDispatcher.updateUI()
        return

    def unitBrowser_onErrorReceived(self, errorCode, errorString):
        if errorCode not in IGNORED_UNIT_BROWSER_ERRORS:
            msgType, msgBody = messages.getUnitBrowserMessage(
                errorCode, errorString)
            SystemMessages.pushMessage(msgBody, type=msgType)

    def forMgr_onFortStateChanged(self):
        g_eventDispatcher.updateUI()

    def fortMgr_onFortResponseReceived(self, requestID, resultCode, _):
        self.__requestCtx.stopProcessing(
            result=resultCode in (FORT_ERROR.OK, ))
        g_eventDispatcher.updateUI()

    def __startListening(self):
        """
        Subscribes to player events.
        """
        g_playerEvents.onPrebattleJoined += self.pe_onPrebattleJoined
        g_playerEvents.onPrebattleJoinFailure += self.pe_onPrebattleJoinFailure
        g_playerEvents.onPrebattleLeft += self.pe_onPrebattleLeft
        g_playerEvents.onKickedFromPrebattle += self.pe_onKickedFromPrebattle
        g_playerEvents.onArenaJoinFailure += self.pe_onArenaJoinFailure
        g_playerEvents.onKickedFromArena += self.pe_onKickedFromArena
        g_playerEvents.onPrebattleAutoInvitesChanged += self.pe_onPrebattleAutoInvitesChanged
        gameSession = game_control.g_instance.gameSession
        rentCtr = game_control.g_instance.rentals
        igrCtr = game_control.g_instance.igr
        if gameSession.lastBanMsg is not None:
            self.gs_onTillBanNotification(*gameSession.lastBanMsg)
        gameSession.onTimeTillBan += self.gs_onTillBanNotification
        rentCtr.onRentChangeNotify += self.rc_onRentChange
        igrCtr.onIgrTypeChanged += self.igr_onRoomChange
        unitMgr = prb_getters.getClientUnitMgr()
        if unitMgr:
            unitMgr.onUnitJoined += self.unitMgr_onUnitJoined
            unitMgr.onUnitLeft += self.unitMgr_onUnitLeft
            unitMgr.onUnitRestored += self.unitMgr_onUnitRestored
            unitMgr.onUnitErrorReceived += self.unitMgr_onUnitErrorReceived
        else:
            LOG_ERROR('Unit manager is not defined')
        unitBrowser = prb_getters.getClientUnitBrowser()
        if unitBrowser:
            unitBrowser.onErrorReceived += self.unitBrowser_onErrorReceived
        else:
            LOG_ERROR('Unit browser is not defined')
        fortMgr = getClientFortMgr()
        if fortMgr:
            fortMgr.onFortStateChanged += self.forMgr_onFortStateChanged
            fortMgr.onFortResponseReceived += self.fortMgr_onFortResponseReceived
        else:
            LOG_ERROR('Fort manager is not defined')
        g_prbCtrlEvents.onPrebattleIntroModeJoined += self.ctrl_onPrebattleIntroModeJoined
        g_prbCtrlEvents.onPrebattleIntroModeLeft += self.ctrl_onPrebattleIntroModeLeft
        g_prbCtrlEvents.onPrebattleInited += self.ctrl_onPrebattleInited
        g_prbCtrlEvents.onUnitIntroModeJoined += self.ctrl_onUnitIntroModeJoined
        g_prbCtrlEvents.onUnitIntroModeLeft += self.ctrl_onUnitIntroModeLeft
        g_prbCtrlEvents.onPreQueueFunctionalCreated += self.ctrl_onPreQueueFunctionalCreated
        g_prbCtrlEvents.onPreQueueFunctionalDestroyed += self.ctrl_onPreQueueFunctionalDestroyed
        g_eventsCache.companies.onCompanyStateChanged += self.onCompanyStateChanged
        return

    def __stopListening(self):
        """
        Unsubscribe from player events.
        """
        g_eventsCache.companies.onCompanyStateChanged -= self.onCompanyStateChanged
        g_playerEvents.onPrebattleJoined -= self.pe_onPrebattleJoined
        g_playerEvents.onPrebattleJoinFailure -= self.pe_onPrebattleJoinFailure
        g_playerEvents.onPrebattleLeft -= self.pe_onPrebattleLeft
        g_playerEvents.onKickedFromPrebattle -= self.pe_onKickedFromPrebattle
        g_playerEvents.onArenaJoinFailure -= self.pe_onArenaJoinFailure
        g_playerEvents.onKickedFromArena -= self.pe_onKickedFromArena
        g_playerEvents.onPrebattleAutoInvitesChanged -= self.pe_onPrebattleAutoInvitesChanged
        game_control.g_instance.gameSession.onTimeTillBan -= self.gs_onTillBanNotification
        game_control.g_instance.rentals.onRentChangeNotify -= self.rc_onRentChange
        game_control.g_instance.igr.onIgrTypeChanged -= self.igr_onRoomChange
        unitMgr = prb_getters.getClientUnitMgr()
        if unitMgr:
            unitMgr.onUnitJoined -= self.unitMgr_onUnitJoined
            unitMgr.onUnitLeft -= self.unitMgr_onUnitLeft
            unitMgr.onUnitRestored -= self.unitMgr_onUnitRestored
            unitMgr.onUnitErrorReceived -= self.unitMgr_onUnitErrorReceived
        unitBrowser = prb_getters.getClientUnitBrowser()
        if unitBrowser:
            unitBrowser.onErrorReceived -= self.unitBrowser_onErrorReceived
        fortMgr = getClientFortMgr()
        if fortMgr:
            fortMgr.onFortResponseReceived -= self.fortMgr_onFortResponseReceived
            fortMgr.onFortStateChanged -= self.forMgr_onFortStateChanged
        g_prbCtrlEvents.clear()

    def __clear(self, woEvents=False):
        if self.__requestCtx:
            self.__requestCtx.clear()
            self.__requestCtx = None
        if self.__collection is not None:
            self.__collection.clear(woEvents=woEvents)
            self.__collection = None
        if self.__factories is not None:
            self.__factories.clear()
            self.__factories = None
        g_eventDispatcher.removeSpecBattlesFromCarousel()
        self.clear()
        return

    def __changePrbFunctional(self,
                              flags=FUNCTIONAL_FLAG.UNDEFINED,
                              prbType=0,
                              stop=True):
        self.__setFunctional(
            CreateFunctionalCtx(_CTRL_TYPE.PREBATTLE, prbType, flags=flags))
        if stop:
            self.__requestCtx.stopProcessing(result=True)

    def __changeUnitFunctional(self,
                               flags=FUNCTIONAL_FLAG.UNDEFINED,
                               prbType=0):
        self.__setFunctional(
            CreateFunctionalCtx(_CTRL_TYPE.UNIT,
                                prbType,
                                flags=flags,
                                initCtx=self.__requestCtx))
        self.__requestCtx.stopProcessing(result=True)

    def __changePreQueueFunctional(self,
                                   flags=FUNCTIONAL_FLAG.UNDEFINED,
                                   queueType=0):
        self.__setFunctional(
            CreateFunctionalCtx(_CTRL_TYPE.PREQUEUE, queueType, flags=flags))
        self.__requestCtx.stopProcessing(result=True)

    def __validateJoinOp(self, ctx):
        if self.__requestCtx.isProcessing():
            LOG_ERROR('Request is processing', self.__requestCtx)
            return False
        for func in self.__collection.getIterator():
            if func.isPlayerJoined(ctx):
                LOG_DEBUG('Player already joined', ctx)
                func.showGUI(ctx)
                return False
            if func.hasLockedState():
                SystemMessages.pushI18nMessage(
                    '#system_messages:prebattle/hasLockedState',
                    type=SystemMessages.SM_TYPE.Warning)
                return False

        return True

    @process
    def __doSelect(self, entry):
        yield self.select(entry)

    def __setFunctional(self, ctx):
        ctx.addFlags(self.__collection.getFunctionalFlags())
        items = []
        for created in self.__factories.createFunctional(ctx):
            if ctx.hasFlags(FUNCTIONAL_FLAG.SET_GLOBAL_LISTENERS):
                created.addMutualListeners(self)
            items.append(created)

        for item in items:
            ctrlType = item.getCtrlType()
            self.__collection.setFunctionalFlags(ctrlType, ctx.getFlags())
            flag = self.__collection.setItem(item.getCtrlType(), item,
                                             ctx.getInitCtx())
            ctx.addFlags(flag)

        LOG_DEBUG('Functionals have been updated', ctx.getFlagsToStrings())
        ctx.clear()
        return ctx.getFlags()
Example #2
0
class _PrebattleDispatcher(ListenersCollection):
    def __init__(self):
        super(_PrebattleDispatcher, self).__init__()
        self.__requestCtx = None
        self.__collection = FunctionalCollection()
        self.__factories = ControlFactoryDecorator()
        self.__nextPrbFunctional = None
        self._setListenerClass(IGlobalListener)
        return

    def __del__(self):
        LOG_DEBUG("_PrebattleDispatcher deleted")

    def start(self):
        g_eventDispatcher.init(self)
        result = self.__setFunctional(CreateFunctionalCtx())
        self.__requestCtx = PrbCtrlRequestCtx()
        self.__startListening()
        functional.initDevFunctional()
        if result & FUNCTIONAL_FLAG.LOAD_PAGE == 0:
            BigWorld.callback(0.001, lambda: g_eventDispatcher.loadHangar())
        g_eventDispatcher.updateUI()
        if GUI_SETTINGS.specPrebatlesVisible and not prb_getters.areSpecBattlesHidden():
            g_eventDispatcher.addSpecBattlesToCarousel()

    def stop(self):
        self.__nextPrbFunctional = None
        self.__stopListening()
        functional.finiDevFunctional()
        self.__clear(woEvents=True)
        g_eventDispatcher.fini()
        return

    def getPrbFunctional(self):
        return self.__collection.getItem(_CTRL_TYPE.PREBATTLE)

    def getUnitFunctional(self):
        return self.__collection.getItem(_CTRL_TYPE.UNIT)

    def getPreQueueFunctional(self):
        return self.__collection.getItem(_CTRL_TYPE.PREQUEUE)

    def getFunctional(self, ctrlType):
        return self.__collection.getItem(ctrlType)

    def getFunctionalCollection(self):
        return self.__collection

    def getControlFactories(self):
        return self.__factories

    def hasModalEntity(self):
        return self.__collection.hasModalEntity()

    def getFunctionalState(self):
        return self.__collection.getState(self.__factories)

    @async
    @process
    def create(self, ctx, callback=None):
        if ctx.getRequestType() == _RQ_TYPE.CREATE:
            if not self.__requestCtx.isProcessing():
                result = yield self.unlock(ctx)
                if result:
                    entry = self.__factories.createEntry(ctx)
                    if entry:
                        LOG_DEBUG("Request to create", ctx)
                        self.__requestCtx = ctx
                        entry.create(ctx, callback=callback)
                    else:
                        LOG_ERROR("Entry not found", ctx)
                        if callback is not None:
                            callback(False)
                elif callback is not None:
                    callback(False)
            else:
                LOG_ERROR("Request is processing", self.__requestCtx)
                if callback is not None:
                    callback(False)
                yield lambda callback=None: callback
        else:
            LOG_ERROR("Invalid context to create", ctx)
            if callback is not None:
                callback(False)
            yield lambda callback=None: callback
        return

    @async
    @process
    def join(self, ctx, callback=None):
        if self.__validateJoinOp(ctx):
            result = yield self.unlock(ctx)
            ctx.setForced(result)
            if result:
                entry = self.__factories.createEntry(ctx)
                if entry:
                    LOG_DEBUG("Request to join", ctx)
                    self.__requestCtx = ctx
                    entry.join(ctx, callback=callback)
                else:
                    LOG_ERROR("Entry not found", ctx)
                    if callback is not None:
                        callback(False)
        else:
            if callback is not None:
                callback(False)
            yield lambda callback=None: callback
        return

    @async
    def leave(self, ctx, callback=None):
        if ctx.getRequestType() != _RQ_TYPE.LEAVE:
            LOG_ERROR("Invalid context to leave prebattle/unit", ctx)
            if callback is not None:
                callback(False)
            return
        elif self.__requestCtx.isProcessing():
            LOG_ERROR("Request is processing", self.__requestCtx)
            if callback is not None:
                callback(False)
            return
        else:
            ctrlType = ctx.getCtrlType()
            formation = self.__collection.getItem(ctx.getCtrlType())
            if formation is not None:
                if formation.hasLockedState():
                    entityType = formation.getEntityType()
                    SystemMessages.pushI18nMessage(
                        messages.getLeaveDisabledMessage(ctrlType, entityType), type=SystemMessages.SM_TYPE.Warning
                    )
                    if callback is not None:
                        callback(False)
                    return
                LOG_DEBUG("Request to leave formation", ctx)
                self.__requestCtx = ctx
                formation.leave(ctx, callback=callback)
            else:
                LOG_ERROR("Functional not found", ctx)
                if callback is not None:
                    callback(False)
            return

    @async
    @process
    def unlock(self, unlockCtx, callback=None):
        state = self.getFunctionalState()
        result = True
        if not state.isIntroMode:
            canDoLeave = True
        elif unlockCtx.hasFlags(FUNCTIONAL_FLAG.SWITCH):
            if state.ctrlTypeID == unlockCtx.getCtrlType() and state.entityTypeID != unlockCtx.getEntityType():
                canDoLeave = True
                unlockCtx.removeFlags(FUNCTIONAL_FLAG.SWITCH)
            else:
                canDoLeave = False
        else:
            canDoLeave = True
        if canDoLeave:
            factory = self.__factories.get(state.ctrlTypeID)
            result = False
            if factory:
                ctx = factory.createLeaveCtx(unlockCtx.getFlags())
                if ctx:
                    meta = self.__collection.getConfirmDialogMeta(state.ctrlTypeID, unlockCtx)
                    if meta:
                        result = yield DialogsInterface.showDialog(meta)
                    else:
                        result = True
                    if result:
                        result = yield self.leave(ctx)
                else:
                    LOG_ERROR("Can not create leave ctx", state)
            else:
                LOG_ERROR("Factory is not found", state)
        if callback is not None:
            callback(result)
        yield lambda callback=None: callback
        return

    @async
    @process
    def select(self, entry, callback=None):
        ctx = entry.makeDefCtx()
        if ctx and self.__validateJoinOp(ctx):
            result = yield self.unlock(ctx)
            ctx.setForced(result)
            if result:
                LOG_DEBUG("Request to select", ctx)
                self.__requestCtx = ctx
                entry.select(ctx, callback=callback)
            elif callback is not None:
                callback(False)
        else:
            if callback is not None:
                callback(False)
            yield lambda callback=None: callback
        return

    @async
    def sendPrbRequest(self, ctx, callback=None):
        prbFunctional = self.getFunctional(_CTRL_TYPE.PREBATTLE)
        if prbFunctional:
            prbFunctional.request(ctx, callback=callback)
        else:
            LOG_ERROR("prbFunctional is not found", ctx)
            if callback is not None:
                callback(False)
        return

    @async
    def sendUnitRequest(self, ctx, callback=None):
        unitFunctional = self.getFunctional(_CTRL_TYPE.UNIT)
        if unitFunctional:
            unitFunctional.request(ctx, callback=callback)
        else:
            LOG_ERROR("unitFunctional is not found", ctx)
            if callback is not None:
                callback(False)
        return

    @async
    def sendPreQueueRequest(self, ctx, callback=None):
        preQueueFunctional = self.getFunctional(_CTRL_TYPE.PREQUEUE)
        if preQueueFunctional is not None:
            preQueueFunctional.request(ctx, callback=callback)
        else:
            LOG_ERROR("preQueueFunctional is not found", ctx)
            if callback is not None:
                callback(False)
        return

    def exitFromQueue(self):
        self.__collection.exitFromQueue()

    def canPlayerDoAction(self):
        canDo, restriction = self.__collection.canPlayerDoAction(False)
        if canDo:
            if g_currentPreviewVehicle.isPresent():
                canDo = False
                restriction = PREBATTLE_RESTRICTION.PREVIEW_VEHICLE_IS_PRESENT
            elif not g_currentVehicle.isReadyToFight():
                if not g_currentVehicle.isPresent():
                    canDo = False
                    restriction = PREBATTLE_RESTRICTION.VEHICLE_NOT_PRESENT
                elif g_currentVehicle.isInBattle():
                    canDo = False
                    restriction = PREBATTLE_RESTRICTION.VEHICLE_IN_BATTLE
                elif not g_currentVehicle.isCrewFull():
                    canDo = False
                    restriction = PREBATTLE_RESTRICTION.CREW_NOT_FULL
                elif g_currentVehicle.isBroken():
                    canDo = False
                    restriction = PREBATTLE_RESTRICTION.VEHICLE_BROKEN
                elif g_currentVehicle.isFalloutOnly() and not game_control.getFalloutCtrl().isSelected():
                    canDo = False
                    restriction = PREBATTLE_RESTRICTION.VEHICLE_FALLOUT_ONLY
                elif g_currentVehicle.isDisabledInRoaming():
                    canDo = False
                    restriction = PREBATTLE_RESTRICTION.VEHICLE_ROAMING
                elif g_currentVehicle.isDisabledInPremIGR():
                    canDo = False
                    restriction = PREBATTLE_RESTRICTION.VEHICLE_IN_PREMIUM_IGR_ONLY
                elif g_currentVehicle.isDisabledInRent():
                    unit = self.getUnitFunctional()
                    if unit is not None and unit.getFlags().isInPreArena():
                        canDo = True
                    else:
                        canDo = False
                        if g_currentVehicle.isPremiumIGR():
                            restriction = PREBATTLE_RESTRICTION.VEHICLE_IGR_RENTALS_IS_OVER
                        else:
                            restriction = PREBATTLE_RESTRICTION.VEHICLE_RENTALS_IS_OVER
            if canDo:
                canDo, restriction = self.__collection.canPlayerDoAction(True)
        return (canDo, restriction)

    def getPlayerInfo(self):
        return self.__collection.getPlayerInfo(self.__factories)

    def doAction(self, action=None):
        if not g_currentVehicle.isPresent():
            SystemMessages.pushMessage(
                messages.getInvalidVehicleMessage(PREBATTLE_RESTRICTION.VEHICLE_NOT_PRESENT),
                type=SystemMessages.SM_TYPE.Error,
            )
            return False
        LOG_DEBUG("Do GUI action", action)
        return self.__collection.doAction(self.__factories, action)

    def doSelectAction(self, action):
        result = self.__collection.doSelectAction(action)
        if result.isProcessed:
            if result.newEntry is not None:
                self.__doSelect(result.newEntry)
            return True
        else:
            entry = self.__factories.createEntryByAction(action)
            if entry is not None:
                self.__doSelect(entry)
                return True
            return False

    @process
    def doLeaveAction(self, ctx):
        meta = self.__collection.getConfirmDialogMeta(ctx.getCtrlType(), ctx)
        if meta is not None:
            isConfirmed = yield DialogsInterface.showDialog(meta)
        else:
            isConfirmed = yield lambda callback: callback(True)
        if isConfirmed:
            yield self.leave(ctx)
        return

    def getGUIPermissions(self):
        return self.__collection.getGUIPermissions()

    def onCompanyStateChanged(self, _):
        if not self.getPrbFunctional().hasLockedState():
            g_eventDispatcher.updateUI()

    def setRequestCtx(self, ctx):
        result = True
        if self.__requestCtx.isProcessing():
            LOG_ERROR("Request is processing", self.__requestCtx)
            result = False
        else:
            self.__requestCtx = ctx
        return result

    def pe_onArenaJoinFailure(self, errorCode, _):
        self.__collection.reset()
        SystemMessages.pushMessage(messages.getJoinFailureMessage(errorCode), type=SystemMessages.SM_TYPE.Error)

    def pe_onKickedFromArena(self, reasonCode):
        self.__collection.reset()
        SystemMessages.pushMessage(messages.getKickReasonMessage(reasonCode), type=SystemMessages.SM_TYPE.Error)

    def pe_onPrebattleAutoInvitesChanged(self):
        if GUI_SETTINGS.specPrebatlesVisible:
            isHidden = prb_getters.areSpecBattlesHidden()
            if isHidden:
                g_eventDispatcher.removeSpecBattlesFromCarousel()
            else:
                g_eventDispatcher.addSpecBattlesToCarousel()
        g_eventDispatcher.updateUI()

    def pe_onPrebattleJoined(self):
        if prb_getters.getClientPrebattle() is not None:
            flags = self.__requestCtx.getFlags()
            self.__setFunctional(CreateFunctionalCtx(_CTRL_TYPE.PREBATTLE, flags=flags))
        else:
            LOG_ERROR("ClientPrebattle is not defined")
            self.__requestCtx.stopProcessing(result=False)
        return

    def pe_onPrebattleJoinFailure(self, errorCode):
        SystemMessages.pushMessage(messages.getJoinFailureMessage(errorCode), type=SystemMessages.SM_TYPE.Error)
        self.__requestCtx.stopProcessing(result=False)
        self.__requestCtx.clearFlags()
        g_eventDispatcher.updateUI()

    def pe_onPrebattleLeft(self):
        if self.__nextPrbFunctional is not None:
            self.__nextPrbFunctional()
            self.__nextPrbFunctional = None
            return
        else:
            flags = self.__requestCtx.getFlags()
            flags |= FUNCTIONAL_FLAG.LEAVE_ENTITY
            prbType = 0
            if flags & FUNCTIONAL_FLAG.SWITCH > 0:
                prbFunctional = self.getFunctional(_CTRL_TYPE.PREBATTLE)
                if prbFunctional is not None:
                    prbType = prbFunctional.getEntityType()
            self.__changePrbFunctional(flags=flags, prbType=prbType, stop=False)
            return

    def pe_onKickedFromPrebattle(self, _):
        self.__changePrbFunctional(stop=True)

    def gs_onTillBanNotification(self, isPlayTimeBan, timeTillBlock):
        if prb_getters.isParentControlActivated():
            self.__collection.reset()
            key = "#system_messages:gameSessionControl/korea/{0:>s}"
            if isPlayTimeBan:
                SystemMessages.g_instance.pushI18nMessage(
                    key.format("playTimeNotification"), timeTillBlock, type=SystemMessages.SM_TYPE.Warning
                )
            else:
                gameSession = game_control.g_instance.gameSession
                notifyStartTime, blockTime = gameSession.getCurfewBlockTime()
                formatter = lambda t: time.strftime("%H:%M", time.localtime(t))
                SystemMessages.g_instance.pushI18nMessage(
                    key.format("midnightNotification"),
                    type=SystemMessages.SM_TYPE.Warning,
                    preBlockTime=formatter(notifyStartTime),
                    blockTime=formatter(blockTime),
                )

    def rc_onRentChange(self, vehicles):
        if (
            g_currentVehicle.isPresent()
            and g_currentVehicle.item.intCD in vehicles
            and g_currentVehicle.isDisabledInRent()
            and g_currentVehicle.isInPrebattle()
        ):
            self.__collection.reset()

    def igr_onRoomChange(self, roomType, _):
        if roomType != IGR_TYPE.PREMIUM:
            if g_currentVehicle.isPremiumIGR() and g_currentVehicle.isInPrebattle():
                self.__collection.reset()

    def ctrl_onPrebattleIntroModeJoined(self, prbType, isLeaving):
        flags = self.__requestCtx.getFlags()
        if not isLeaving:
            self.__changePrbFunctional(flags=flags, prbType=prbType)
        else:
            self.__nextPrbFunctional = partial(self.__changePrbFunctional, flags=flags, prbType=prbType)

    def ctrl_onPrebattleIntroModeLeft(self):
        flags = self.__requestCtx.getFlags()
        flags |= FUNCTIONAL_FLAG.LEAVE_INTRO
        self.__changePrbFunctional(flags=flags)

    def ctrl_onPrebattleInited(self):
        self.__requestCtx.stopProcessing(result=True)
        self.__setFunctional(
            CreateFunctionalCtx(_CTRL_TYPE.PREBATTLE, flags=self.__requestCtx.getFlags(), initCtx=self.__requestCtx)
        )
        g_eventDispatcher.updateUI()

    def ctrl_onUnitIntroModeJoined(self, prbType, flags):
        self.__changeUnitFunctional(flags=flags, prbType=prbType)

    def ctrl_onUnitIntroModeLeft(self):
        flags = self.__requestCtx.getFlags()
        flags |= FUNCTIONAL_FLAG.LEAVE_INTRO
        self.__changeUnitFunctional(flags=flags)

    def ctrl_onPreQueueFunctionalCreated(self, queueType):
        self.__changePreQueueFunctional(flags=self.__requestCtx.getFlags(), queueType=queueType)
        g_eventDispatcher.updateUI()

    def ctrl_onPreQueueFunctionalDestroyed(self):
        flags = self.__requestCtx.getFlags()
        flags |= FUNCTIONAL_FLAG.LEAVE_PRE_QUEUE
        self.__changePreQueueFunctional(flags=flags)

    def unitMgr_onUnitJoined(self, unitMgrID, unitIdx):
        unitFunctional = self.getFunctional(_CTRL_TYPE.UNIT)
        if (
            unitFunctional is not None
            and unitFunctional.getID() == unitMgrID
            and unitFunctional.getUnitIdx() == unitIdx
        ):
            unitFunctional.rejoin()
        else:
            if unitFunctional is not None:
                prbType = unitFunctional.getEntityType()
            else:
                prbType = 0
            self.__changeUnitFunctional(flags=self.__requestCtx.getFlags(), prbType=prbType)
        return

    def unitMgr_onUnitLeft(self, unitMgrID, unitIdx):
        flags = self.__requestCtx.getFlags()
        flags |= FUNCTIONAL_FLAG.LEAVE_ENTITY
        prbType = 0
        if flags & FUNCTIONAL_FLAG.SWITCH > 0:
            unitFunctional = self.getFunctional(_CTRL_TYPE.UNIT)
            if unitFunctional is not None:
                prbType = unitFunctional.getEntityType()
        self.__changeUnitFunctional(flags=flags, prbType=prbType)
        g_eventDispatcher.updateUI()
        return

    def unitMgr_onUnitRestored(self, unitMgrID, unitIdx):
        unitFunctional = self.getFunctional(_CTRL_TYPE.UNIT)
        flags = unitFunctional.getFlags()
        pInfo = unitFunctional.getPlayerInfo()
        if flags.isInPreArena() and pInfo.isInSlot:
            g_eventDispatcher.loadHangar()

    def unitMgr_onUnitErrorReceived(self, requestID, unitMgrID, unitIdx, errorCode, errorString):
        unitFunctional = self.getFunctional(_CTRL_TYPE.UNIT)
        if unitFunctional is not None:
            unitFunctional.setLastError(errorCode)
            if errorCode in RETURN_INTRO_UNIT_MGR_ERRORS and unitFunctional.canSwitchToIntro():
                self.__requestCtx.addFlags(FUNCTIONAL_FLAG.SWITCH)
            if errorCode == UNIT_ERROR.CANT_PICK_LEADER:
                self.__requestCtx.removeFlags(FUNCTIONAL_FLAG.SWITCH)
            elif errorCode == UNIT_ERROR.REMOVED_PLAYER:
                if (
                    self.__requestCtx.getCtrlType() == _CTRL_TYPE.UNIT
                    and unitFunctional.getEntityType() != self.__requestCtx.getEntityType()
                ):
                    self.__requestCtx.removeFlags(FUNCTIONAL_FLAG.SWITCH)
        else:
            LOG_ERROR("Unit functional is not found")
        if errorCode not in IGNORED_UNIT_MGR_ERRORS:
            msgType, msgBody = messages.getUnitMessage(errorCode, errorString)
            SystemMessages.pushMessage(msgBody, type=msgType)
            self.__requestCtx.stopProcessing(result=False)
            g_eventDispatcher.updateUI()
        return

    def unitBrowser_onErrorReceived(self, errorCode, errorString):
        if errorCode not in IGNORED_UNIT_BROWSER_ERRORS:
            msgType, msgBody = messages.getUnitBrowserMessage(errorCode, errorString)
            SystemMessages.pushMessage(msgBody, type=msgType)

    def forMgr_onFortStateChanged(self):
        g_eventDispatcher.updateUI()

    def fortMgr_onFortResponseReceived(self, requestID, resultCode, _):
        self.__requestCtx.stopProcessing(result=resultCode in (FORT_ERROR.OK,))
        g_eventDispatcher.updateUI()

    def __startListening(self):
        """
        Subscribes to player events.
        """
        g_playerEvents.onPrebattleJoined += self.pe_onPrebattleJoined
        g_playerEvents.onPrebattleJoinFailure += self.pe_onPrebattleJoinFailure
        g_playerEvents.onPrebattleLeft += self.pe_onPrebattleLeft
        g_playerEvents.onKickedFromPrebattle += self.pe_onKickedFromPrebattle
        g_playerEvents.onArenaJoinFailure += self.pe_onArenaJoinFailure
        g_playerEvents.onKickedFromArena += self.pe_onKickedFromArena
        g_playerEvents.onPrebattleAutoInvitesChanged += self.pe_onPrebattleAutoInvitesChanged
        gameSession = game_control.g_instance.gameSession
        rentCtr = game_control.g_instance.rentals
        igrCtr = game_control.g_instance.igr
        if gameSession.lastBanMsg is not None:
            self.gs_onTillBanNotification(*gameSession.lastBanMsg)
        gameSession.onTimeTillBan += self.gs_onTillBanNotification
        rentCtr.onRentChangeNotify += self.rc_onRentChange
        igrCtr.onIgrTypeChanged += self.igr_onRoomChange
        unitMgr = prb_getters.getClientUnitMgr()
        if unitMgr:
            unitMgr.onUnitJoined += self.unitMgr_onUnitJoined
            unitMgr.onUnitLeft += self.unitMgr_onUnitLeft
            unitMgr.onUnitRestored += self.unitMgr_onUnitRestored
            unitMgr.onUnitErrorReceived += self.unitMgr_onUnitErrorReceived
        else:
            LOG_ERROR("Unit manager is not defined")
        unitBrowser = prb_getters.getClientUnitBrowser()
        if unitBrowser:
            unitBrowser.onErrorReceived += self.unitBrowser_onErrorReceived
        else:
            LOG_ERROR("Unit browser is not defined")
        fortMgr = getClientFortMgr()
        if fortMgr:
            fortMgr.onFortStateChanged += self.forMgr_onFortStateChanged
            fortMgr.onFortResponseReceived += self.fortMgr_onFortResponseReceived
        else:
            LOG_ERROR("Fort manager is not defined")
        g_prbCtrlEvents.onPrebattleIntroModeJoined += self.ctrl_onPrebattleIntroModeJoined
        g_prbCtrlEvents.onPrebattleIntroModeLeft += self.ctrl_onPrebattleIntroModeLeft
        g_prbCtrlEvents.onPrebattleInited += self.ctrl_onPrebattleInited
        g_prbCtrlEvents.onUnitIntroModeJoined += self.ctrl_onUnitIntroModeJoined
        g_prbCtrlEvents.onUnitIntroModeLeft += self.ctrl_onUnitIntroModeLeft
        g_prbCtrlEvents.onPreQueueFunctionalCreated += self.ctrl_onPreQueueFunctionalCreated
        g_prbCtrlEvents.onPreQueueFunctionalDestroyed += self.ctrl_onPreQueueFunctionalDestroyed
        g_eventsCache.companies.onCompanyStateChanged += self.onCompanyStateChanged
        return

    def __stopListening(self):
        """
        Unsubscribe from player events.
        """
        g_eventsCache.companies.onCompanyStateChanged -= self.onCompanyStateChanged
        g_playerEvents.onPrebattleJoined -= self.pe_onPrebattleJoined
        g_playerEvents.onPrebattleJoinFailure -= self.pe_onPrebattleJoinFailure
        g_playerEvents.onPrebattleLeft -= self.pe_onPrebattleLeft
        g_playerEvents.onKickedFromPrebattle -= self.pe_onKickedFromPrebattle
        g_playerEvents.onArenaJoinFailure -= self.pe_onArenaJoinFailure
        g_playerEvents.onKickedFromArena -= self.pe_onKickedFromArena
        g_playerEvents.onPrebattleAutoInvitesChanged -= self.pe_onPrebattleAutoInvitesChanged
        game_control.g_instance.gameSession.onTimeTillBan -= self.gs_onTillBanNotification
        game_control.g_instance.rentals.onRentChangeNotify -= self.rc_onRentChange
        game_control.g_instance.igr.onIgrTypeChanged -= self.igr_onRoomChange
        unitMgr = prb_getters.getClientUnitMgr()
        if unitMgr:
            unitMgr.onUnitJoined -= self.unitMgr_onUnitJoined
            unitMgr.onUnitLeft -= self.unitMgr_onUnitLeft
            unitMgr.onUnitRestored -= self.unitMgr_onUnitRestored
            unitMgr.onUnitErrorReceived -= self.unitMgr_onUnitErrorReceived
        unitBrowser = prb_getters.getClientUnitBrowser()
        if unitBrowser:
            unitBrowser.onErrorReceived -= self.unitBrowser_onErrorReceived
        fortMgr = getClientFortMgr()
        if fortMgr:
            fortMgr.onFortResponseReceived -= self.fortMgr_onFortResponseReceived
            fortMgr.onFortStateChanged -= self.forMgr_onFortStateChanged
        g_prbCtrlEvents.clear()

    def __clear(self, woEvents=False):
        if self.__requestCtx:
            self.__requestCtx.clear()
            self.__requestCtx = None
        if self.__collection is not None:
            self.__collection.clear(woEvents=woEvents)
            self.__collection = None
        if self.__factories is not None:
            self.__factories.clear()
            self.__factories = None
        g_eventDispatcher.removeSpecBattlesFromCarousel()
        self.clear()
        return

    def __changePrbFunctional(self, flags=FUNCTIONAL_FLAG.UNDEFINED, prbType=0, stop=True):
        self.__setFunctional(CreateFunctionalCtx(_CTRL_TYPE.PREBATTLE, prbType, flags=flags))
        if stop:
            self.__requestCtx.stopProcessing(result=True)

    def __changeUnitFunctional(self, flags=FUNCTIONAL_FLAG.UNDEFINED, prbType=0):
        self.__setFunctional(CreateFunctionalCtx(_CTRL_TYPE.UNIT, prbType, flags=flags, initCtx=self.__requestCtx))
        self.__requestCtx.stopProcessing(result=True)

    def __changePreQueueFunctional(self, flags=FUNCTIONAL_FLAG.UNDEFINED, queueType=0):
        self.__setFunctional(CreateFunctionalCtx(_CTRL_TYPE.PREQUEUE, queueType, flags=flags))
        self.__requestCtx.stopProcessing(result=True)

    def __validateJoinOp(self, ctx):
        if self.__requestCtx.isProcessing():
            LOG_ERROR("Request is processing", self.__requestCtx)
            return False
        for func in self.__collection.getIterator():
            if func.isPlayerJoined(ctx):
                LOG_DEBUG("Player already joined", ctx)
                func.showGUI(ctx)
                return False
            if func.hasLockedState():
                SystemMessages.pushI18nMessage(
                    "#system_messages:prebattle/hasLockedState", type=SystemMessages.SM_TYPE.Warning
                )
                return False

        return True

    @process
    def __doSelect(self, entry):
        yield self.select(entry)

    def __setFunctional(self, ctx):
        ctx.addFlags(self.__collection.getFunctionalFlags())
        items = []
        for created in self.__factories.createFunctional(ctx):
            if ctx.hasFlags(FUNCTIONAL_FLAG.SET_GLOBAL_LISTENERS):
                created.addMutualListeners(self)
            items.append(created)

        for item in items:
            ctrlType = item.getCtrlType()
            self.__collection.setFunctionalFlags(ctrlType, ctx.getFlags())
            flag = self.__collection.setItem(item.getCtrlType(), item, ctx.getInitCtx())
            ctx.addFlags(flag)

        LOG_DEBUG("Functionals have been updated", ctx.getFlagsToStrings())
        ctx.clear()
        return ctx.getFlags()