def __updateCrew(self): from gui.shared.utils.requesters import Requester self.__crew.clear() barracks = yield Requester('tankman').getFromInventory() for tankman in barracks: for crewman in self.item.crew: if crewman[1] is not None and crewman[ 1].invID == tankman.inventoryId: factor = tankman.descriptor.efficiencyOnVehicle( g_itemsCache.items.getVehicle( self.__vehInvID).descriptor) crew_member = { "level": tankman.descriptor.roleLevel * factor[0], "skill": {} } skills = [] for skill_name in tankman.descriptor.skills: skills.append({"name": skill_name, "level": 100}) if len(skills) != 0: skills[-1]["level"] = tankman.descriptor.lastSkillLevel for skill in skills: crew_member["skill"][skill["name"]] = skill["level"] self.__crew[tankman.descriptor.role] = crew_member
def setRefillSettings(self, vehicleCompact, repair, shells, equipment): vcls = yield Requester('vehicle').getFromInventory() vehicle = getItemByCompact(vehicleCompact) for v in vcls: if v.inventoryId == vehicle.inventoryId: vehicle = v if vehicle.isAutoRepair != repair: yield vehicle.setAutoRepair(repair) if vehicle.isAutoLoad != shells: yield vehicle.setAutoLoad(shells) if vehicle.isAutoEquip != equipment: yield vehicle.setAutoEquip(equipment)
def repair(self): myVehicles = yield Requester('vehicle').getFromInventory() oldStyleVehicle = None for v in myVehicles: if v.inventoryId == g_currentVehicle.invID: oldStyleVehicle = v break if oldStyleVehicle.repairCost > 0: success, message = yield oldStyleVehicle.repair() SystemMessages.g_instance.pushMessage( message, SystemMessages.SM_TYPE.Repair if success else SystemMessages.SM_TYPE.Error) return
def updateVehicleTypeDropdown(self, nationID, vclass): Waiting.show('updating') unlocks = yield StatsRequester().getUnlocks() modulesAll = yield Requester('vehicle').getFromShop() data = [{'id': None, 'label': DIALOGS.RECRUITWINDOW_MENUEMPTYROW}] modulesAll.sort() for module in modulesAll: compdecs = module.descriptor.type.compactDescr if compdecs in unlocks and module.descriptor.type.id[ 0] == nationID and module.type == vclass: data.append({ 'id': module.descriptor.type.id[1], 'label': module.descriptor.type.shortUserString }) self.flashObject.as_setVehicleTypeDropdown(data) Waiting.hide('updating') return
def tryLoadShells(self): ammo = yield Requester('shell').getFromInventory() message = '#system_messages:charge/inventory_error' success = False rootCD = self._data.getRootCD() item = self._data.getInvItem(rootCD) raise item is not None or AssertionError for shell in item.shells: count = ammo[ammo.index(shell)].count if shell in ammo else 0 if shell.default > count: break else: success, message = yield item.loadShells(None) SystemMessages.pushI18nMessage( message, type=SystemMessages.SM_TYPE.Information if success else SystemMessages.SM_TYPE.Warning) return
def updateAllDropdowns(self, nationID, tankType, typeID, roleType): nationsDP = [{ 'id': None, 'label': DIALOGS.RECRUITWINDOW_MENUEMPTYROW }, { 'id': nationID, 'label': MENU.nations(nations.NAMES[int(nationID)]) }] classesDP = [{ 'id': None, 'label': DIALOGS.RECRUITWINDOW_MENUEMPTYROW }, { 'id': tankType, 'label': DIALOGS.recruitwindow_vehicleclassdropdown(tankType) }] typesDP = [{'id': None, 'label': DIALOGS.RECRUITWINDOW_MENUEMPTYROW}] rolesDP = [{'id': None, 'label': DIALOGS.RECRUITWINDOW_MENUEMPTYROW}] unlocks = yield StatsRequester().getUnlocks() modulesAll = yield Requester('vehicle').getFromShop() for module in modulesAll: compdecs = module.descriptor.type.compactDescr if compdecs in unlocks and module.descriptor.type.id[ 0] == nationID and module.descriptor.type.id[1] == typeID: typesDP.append({ 'id': module.descriptor.type.id[1], 'label': module.descriptor.type.shortUserString }) for role in module.descriptor.type.crewRoles: if role[0] == roleType: rolesDP.append({ 'id': role[0], 'label': convert(getSkillsConfig()[role[0]]['userString']) }) break self.flashObject.as_setAllDropdowns(nationsDP, classesDP, typesDP, rolesDP) return
def __updateTanksList(self): data = list() modulesAll = yield Requester('vehicle').getFromInventory() modulesAll.sort() for module in modulesAll: if self.filter['nation'] != -1 and self.filter[ 'nation'] != module.descriptor.type.id[0] or self.filter[ 'tankType'] != 'None' and self.filter[ 'tankType'] != -1 and self.filter[ 'tankType'] != module.type: continue data.append({ 'data': { 'type': module.type, 'nationID': module.descriptor.type.id[0], 'typeID': module.descriptor.type.id[1] }, 'label': module.descriptor.type.shortUserString }) self.as_updateTanksListS(data)
def updateVehicleClassDropdown(self, nationID): Waiting.show('updating') unlocks = yield StatsRequester().getUnlocks() modulesAll = yield Requester('vehicle').getFromShop() classes = [] data = [{'id': None, 'label': DIALOGS.RECRUITWINDOW_MENUEMPTYROW}] modulesAll.sort() for module in modulesAll: compdecs = module.descriptor.type.compactDescr if compdecs in unlocks and module.descriptor.type.id[ 0] == nationID and module.type not in classes: classes.append(module.type) data.append({ 'id': module.type, 'label': DIALOGS.recruitwindow_vehicleclassdropdown(module.type) }) self.flashObject.as_setVehicleClassDropdown(data) Waiting.hide('updating') return
def onShowRecruitWindow(self, callbackID): credits, gold = g_itemsCache.items.stats.money upgradeParams = yield StatsRequester().getTankmanCost() data = [ credits, gold, round(upgradeParams[1]['credits']), upgradeParams[2]['gold'], len(ROLES) ] for role in ROLES: data.append(role) data.append(convert(getSkillsConfig()[role]['userString'])) unlocks = yield StatsRequester().getUnlocks() modulesAll = yield Requester('vehicle').getFromShop() modulesAll.sort() for module in modulesAll: compdecs = module.descriptor.type.compactDescr if compdecs in unlocks: data.append(module.type) data.append(module.descriptor.type.id[0]) data.append(module.descriptor.type.id[1]) data.append(module.descriptor.type.shortUserString)
def updateRoleDropdown(self, nationID, vclass, typeID): Waiting.show('updating') unlocks = yield StatsRequester().getUnlocks() modulesAll = yield Requester('vehicle').getFromShop() roles = [] data = [{'id': None, 'label': DIALOGS.RECRUITWINDOW_MENUEMPTYROW}] modulesAll.sort() for module in modulesAll: compdecs = module.descriptor.type.compactDescr if compdecs in unlocks and module.descriptor.type.id[ 0] == nationID and module.descriptor.type.id[1] == typeID: for role in module.descriptor.type.crewRoles: if role[0] not in roles: roles.append(role[0]) data.append({ 'id': role[0], 'label': convert(getSkillsConfig()[role[0]]['userString']) }) self.flashObject.as_setRoleDropdown(data) Waiting.hide('updating') return
def __requestAvailableItems(self, type): myVehicles = yield Requester('vehicle').getFromInventory() modulesAllVehicle = VehicleItemsRequester(myVehicles).getItems([type]) oldStyleVehicle = None for v in myVehicles: if v.inventoryId == g_currentVehicle.invID: oldStyleVehicle = v break newStyleVehicle = g_currentVehicle.item modulesAllInventory = yield Requester(type).getFromInventory() data = yield AvailableItemsRequester(oldStyleVehicle, type).request() if type in AmmunitionPanel.__ARTEFACTS_SLOTS: unlocks = [ m for m in data if m.isCurrent ] else: unlocks = yield StatsRequester().getUnlocks() data.sort(reverse=True) if type in AmmunitionPanel.__ARTEFACTS_SLOTS: dataProvider = [[], [], []] else: dataProvider = [] credits, gold = g_itemsCache.items.stats.money for module in data: price = yield module.getPrice() inventoryCount, vehicleCount = self.__getModuleInventoryAndVehicleCounts(modulesAllVehicle, modulesAllInventory, module, type) moduleData = {'id': gui_items.compactItem(module), 'type': type, 'name': module.name if type in AmmunitionPanel.__ARTEFACTS_SLOTS else module.longName, 'desc': module.getTableName(oldStyleVehicle), 'target': 2 if type == ITEM_TYPE_NAMES[3] and not oldStyleVehicle.hasTurrets else module.target, 'price': price[0] if price[1] == 0 else price[1], 'currency': 'credits' if price[1] == 0 else 'gold', 'icon': module.icon if type in AmmunitionPanel.__ARTEFACTS_SLOTS else module.level, 'inventoryCount': inventoryCount, 'vehicleCount': vehicleCount} if type == ITEM_TYPE_NAMES[4]: if module.isClipGun(oldStyleVehicle.descriptor): moduleData[EXTRA_MODULE_INFO] = CLIP_ICON_PATH if type in AmmunitionPanel.__ARTEFACTS_SLOTS: moduleData['removable'] = module.isRemovable moduleData['slotIndex'] = module.index for i in xrange(3): md = moduleData.copy() fits = isModuleFitVehicle(module, newStyleVehicle, price, (credits, gold), unlocks, i) if md.get('target') == 1: md['status'] = MENU.MODULEFITS_WRONG_SLOT if i != md.get('slotIndex') else fits[1] md['isSelected'] = i == md.get('slotIndex') else: md['status'] = fits[1] md['isSelected'] = False md['slotIndex'] = i dataProvider[i].append(md) else: fits = isModuleFitVehicle(module, newStyleVehicle, price, (credits, gold), unlocks) moduleData['removable'] = True moduleData['isSelected'] = moduleData.get('target') == 1 moduleData['status'] = fits[1] dataProvider.append(moduleData) self.as_setDataS(dataProvider, type) return
def updateTankmen(self): Waiting.show('updateTankmen') tankmen = yield Requester('tankman').getFromInventory() vehicles = yield Requester('vehicle').getFromInventory() vehicle = None if g_currentVehicle.isPresent(): for veh in vehicles: if veh.inventoryId == g_currentVehicle.invID: vehicle = veh break if vehicle is None: LOG_ERROR('Vehicle not found', g_currentVehicle.invID) Waiting.hide('updateTankmen') return crew = [] for tId in vehicle.crew: if tId is None: crew.append(None) continue for tankman in tankmen: if tankman.inventoryId == tId: crew.append(tankman) vehicle = g_currentVehicle.item commander_bonus = vehicle.bonuses['commander'] roles = [] brotherhood_bonus = getSkillsConfig( )['brotherhood']['crewLevelIncrease'] for t in crew: if t is None or 'brotherhood' not in t.skills or t.skills.index( 'brotherhood') == len( t.skills) - 1 and t.lastSkillLevel != 100: brotherhood_bonus = 0 break lessMastered = 0 tankmenDescrs = dict(vehicle.crew) for slotIdx, tman in vehicle.crew: if slotIdx > 0 and tman is not None and ( tankmenDescrs[lessMastered] is None or compareMastery(tankmenDescrs[lessMastered].descriptor, tman.descriptor) > 0): lessMastered = slotIdx role = vehicle.descriptor.type.crewRoles[slotIdx][0] roles.append({ 'tankmanID': tman.invID if tman is not None else None, 'roleType': role, 'role': convert(getSkillsConfig()[role]['userString']), 'roleIcon': '%s/%s' % (Tankman.ROLE_ICON_PATH_BIG, getSkillsConfig()[role]['icon']), 'nationID': vehicle.nationID, 'typeID': vehicle.innationID, 'slot': slotIdx, 'vehicleType': vehicle.shortUserName, 'tankType': vehicle.type, 'vehicleElite': vehicle.isPremium, 'roles': list(vehicle.descriptor.type.crewRoles[slotIdx]) }) tankmenData = [] crewIDs = vehicle.crewIndices.keys() for tankman in tankmen: if not tankman.isInTank or tankman.inventoryId in crewIDs: bonus_role_level = commander_bonus if tankman.descriptor.role != 'commander' else 0.0 skills_count = len(list(ACTIVE_SKILLS)) skillsList = [] for skill in tankman.skills: skillLevel = tankman.lastSkillLevel if tankman.skills.index( skill) == len(tankman.skills) - 1 else 100 skillsList.append({ 'tankmanID': tankman.inventoryId, 'id': skill, 'name': getSkillsConfig()[skill]['userString'], 'desc': getSkillsConfig()[skill]['description'], 'icon': getSkillsConfig()[skill]['icon'], 'level': tankman.lastSkillLevel, 'active': skill not in PERKS or skill == 'brotherhood' and brotherhood_bonus != 0 or skill != 'brotherhood' and skillLevel == 100 }) newSkillsCount, lastNewSkillLvl = tankman.newSkillCount if newSkillsCount > 0: skillsList.append({ 'buy': True, 'tankmanID': tankman.inventoryId, 'level': lastNewSkillLvl }) tankmanData = { 'firstname': tankman.firstname, 'lastname': tankman.lastname, 'rank': tankman.rank, 'specializationLevel': tankman.roleLevel, 'role': tankman.role, 'vehicleType': tankman.vehicle.type.shortUserString, 'iconFile': tankman.icon, 'rankIconFile': tankman.iconRank, 'roleIconFile': '%s/%s' % (Tankman.ROLE_ICON_PATH_BIG, tankman.iconRole), 'contourIconFile': tankman.vehicleIconContour, 'tankmanID': tankman.inventoryId, 'nationID': tankman.nation, 'typeID': tankman.vehicle.type.id[1], 'inTank': tankman.isInTank, 'roleType': tankman.descriptor.role, 'tankType': tankman.vehicleType, 'efficiencyLevel': tankman.efficiencyRoleLevel(vehicle.descriptor), 'bonus': bonus_role_level, 'lastSkillLevel': tankman.lastSkillLevel, 'isLessMastered': vehicle.crewIndices.get(tankman.inventoryId) == lessMastered and vehicle.isXPToTman, 'compact': tankman.pack(), 'availableSkillsCount': skills_count, 'skills': skillsList } tankmenData.append(tankmanData) self.as_tankmenResponseS(roles, tankmenData) Waiting.hide('updateTankmen') return
def populateTechnicalMaintenanceEquipment(self, eId1=None, currency1=None, eId2=None, currency2=None, eId3=None, currency3=None, slotIndex=None): shopRqs = yield ShopRequester().request() goldEqsForCredits = shopRqs.isEnabledBuyingGoldEqsForCredits credits, gold = g_itemsCache.items.stats.money myVehicles = yield Requester('vehicle').getFromInventory() modulesAllVehicle = VehicleItemsRequester(myVehicles).getItems( ['equipment']) oldStyleVehicle = None for v in myVehicles: if v.inventoryId == g_currentVehicle.invID: oldStyleVehicle = v break newStyleVehicle = g_currentVehicle.item availableData = yield AvailableItemsRequester(oldStyleVehicle, 'equipment').request() shopEqs = yield Requester('equipment').getFromShop() invEqs = yield Requester('equipment').getFromInventory() eqs = oldStyleVehicle.equipmentsLayout def getShopModule(module): for eq in shopEqs: if eq == module: return eq return None def getInventoryModule(module): for eq in invEqs: if eq == module: return eq return None def getModuleByCD(compactDescr): for eq in availableData: if eq.compactDescr == compactDescr: return eq return None installed = [m for m in availableData if m.isCurrent] currencies = [None, None, None] if eId1 is not None or eId2 is not None or eId3 is not None or slotIndex is not None: installed = [ getModuleByCD(id) for id in (eId1, eId2, eId3) if id is not None ] currencies = [currency1, currency2, currency3] data = [] for item in availableData: if item in installed: invEq = getInventoryModule(item) shopModule = getShopModule(item) i = InventoryItem( itemTypeName='equipment', compactDescr=item.compactDescr, count=invEq.count if invEq is not None else 0, priceOrder=shopModule.priceOrder if shopModule is not None else (0, 0)) if item == getModuleByCD(eId1): i.index = 0 elif item == getModuleByCD(eId2): i.index = 1 elif item == getModuleByCD(eId3): i.index = 2 else: i.index = item.index i.isCurrent = True elif isinstance(item, InventoryItem): i = InventoryItem(itemTypeName='equipment', compactDescr=item.compactDescr, count=item.count, priceOrder=item.priceOrder) else: i = item data.append(i) unlocks = [m for m in data if m.isCurrent] data.sort(reverse=True) installed = [0, 0, 0] for m in availableData: if m.isCurrent: installed[m.index] = m.compactDescr setup = [0, 0, 0] modules = [] if len(eqs): setup = eqs for module in data: vehCount = 0 try: vehCount = modulesAllVehicle[modulesAllVehicle.index( module)].count except Exception: pass invCount = 0 try: invCount = invEqs[invEqs.index(module)].count except Exception: pass shopModule = getShopModule(module) price = (yield shopModule.getPrice()) if shopModule is not None else (0, 0) if goldEqsForCredits: price = (price[0] + price[1] * shopRqs.exchangeRateForShellsAndEqs, price[1]) priceCurrency = 'gold' if not price[1]: priceCurrency = 'credits' elif goldEqsForCredits and module.index is not None: if module.index < len(eqs) and eqs[module.index] < 0: priceCurrency = 'credits' elif currencies[module.index] is not None: priceCurrency = currencies[module.index] fits = [] for i in xrange(3): fits.append( isModuleFitVehicle(module, newStyleVehicle, price, (credits, gold), unlocks, i)[1]) modules.append({ 'id': compactItem(module), 'name': module.name, 'desc': getShortDescr(module.getTableName(oldStyleVehicle)), 'target': module.target, 'compactDescr': module.compactDescr, 'prices': list(price)[:2], 'currency': priceCurrency, 'icon': module.icon, 'index': module.index, 'inventoryCount': invCount, 'vehicleCount': vehCount, 'count': module.count if isinstance(module, InventoryItem) else 0, 'fits': fits, 'goldEqsForCredits': goldEqsForCredits }) self.as_setEquipmentS(installed, setup, modules) return
def setVehicleModule(self, newId, slotIdx, oldId, isRemove): if isRemove: isUseGold = oldId is not None newComponent = gui_items.getItemByCompact(newId) newComponentItem = g_itemsCache.items.getItemByCD(newComponent.compactDescr) return newComponentItem is None and None else: oldComponentItem = None if oldId: oldComponent = gui_items.getItemByCompact(oldId) oldComponentItem = g_itemsCache.items.getItemByCD(oldComponent.compactDescr) if not isRemove and oldComponentItem and oldComponentItem.itemTypeID == GUI_ITEM_TYPE.OPTIONALDEVICE: result = yield getInstallerProcessor(g_currentVehicle.item, oldComponentItem, slotIdx, False, True).request() if result and result.auxData: for m in result.auxData: SystemMessages.g_instance.pushI18nMessage(m.userMsg, type=m.sysMsgType) if result and len(result.userMsg): SystemMessages.g_instance.pushI18nMessage(result.userMsg, type=result.sysMsgType) if not result.success: return iVehicles = yield Requester(ITEM_TYPE_NAMES[1]).getFromInventory() oldStyleVehicle = None for v in iVehicles: if v.inventoryId == g_currentVehicle.invID: oldStyleVehicle = v break conflictedEqs = findConflictedEquipmentForModule(newComponent, oldStyleVehicle) if isinstance(newComponent, gui_items.ShopItem): Waiting.show('buyItem') buyResult = yield ModuleBuyer(newComponentItem, count=1, buyForCredits=True, conflictedEqs=conflictedEqs, install=True).request() if len(buyResult.userMsg): SystemMessages.g_instance.pushI18nMessage(buyResult.userMsg, type=buyResult.sysMsgType) if buyResult.success: newComponentItem = g_itemsCache.items.getItemByCD(newComponent.compactDescr) Waiting.hide('buyItem') if not buyResult.success: return Waiting.show('applyModule') result = yield getInstallerProcessor(g_currentVehicle.item, newComponentItem, slotIdx, not isRemove, isUseGold, conflictedEqs).request() if result and result.auxData: for m in result.auxData: SystemMessages.g_instance.pushI18nMessage(m.userMsg, type=m.sysMsgType) if result and len(result.userMsg): SystemMessages.g_instance.pushI18nMessage(result.userMsg, type=result.sysMsgType) if result and result.success and newComponentItem.itemTypeID in (GUI_ITEM_TYPE.TURRET, GUI_ITEM_TYPE.GUN): iAmmo = yield Requester(ITEM_TYPE_NAMES[10]).getFromInventory() iVehicles = yield Requester(ITEM_TYPE_NAMES[1]).getFromInventory() for iVehicle in iVehicles: if iVehicle.inventoryId == g_currentVehicle.invID: installAmmoVehicle = iVehicle for shell in installAmmoVehicle.shells: iCount = iAmmo[iAmmo.index(shell)].count if shell in iAmmo else 0 if shell.default > iCount: success, message = False, '#system_messages:charge/inventory_error' break else: success, message = yield installAmmoVehicle.loadShells(None) SystemMessages.g_instance.pushI18nMessage(message, type=SystemMessages.SM_TYPE.Information if success else SystemMessages.SM_TYPE.Warning) Waiting.hide('applyModule') return
def populateTechnicalMaintenance(self): shopRqs = yield ShopRequester().request() statsRqs = yield StatsRequesterr().request() goldShellsForCredits = shopRqs.isEnabledBuyingGoldShellsForCredits data = {'gold': statsRqs.gold, 'credits': statsRqs.credits} if g_currentVehicle.isPresent(): iVehicles = yield Requester('vehicle').getFromInventory() for v in iVehicles: if v.inventoryId == g_currentVehicle.invID: vehicle = v break gun = VehicleItem(vehicle.descriptor.gun) iAmmo = yield Requester('shell').getFromInventory() sAmmo = yield Requester('shell').getFromShop() casseteCount = vehicle.descriptor.gun['clip'][0] data.update({ 'vehicleId': vehicle.pack(), 'repairCost': vehicle.repairCost, 'maxRepairCost': vehicle.descriptor.getMaxRepairCost(), 'autoRepair': vehicle.isAutoRepair, 'maxAmmo': gun.descriptor['maxAmmo'], 'casseteFieldText': '' if casseteCount == 1 else makeString('#menu:technicalMaintenance/ammoTitleEx') % casseteCount, 'shells': [] }) shells = data.get('shells') for shell in vehicle.shells: shopShell = sAmmo[sAmmo.index( shell)] if shell in sAmmo else None if shopShell: iCount = iAmmo[iAmmo.index( shell)].count if shell in iAmmo else 0 sPrice = (yield shopShell.getPrice() ) if shell is not shopShell else (0, 0) if goldShellsForCredits: sPrice = ( sPrice[0] + sPrice[1] * shopRqs.exchangeRateForShellsAndEqs, sPrice[1]) priceCurrency = 'gold' if sPrice[ 1] == 0 or goldShellsForCredits and shell.boughtForCredits: priceCurrency = 'credits' buyCount = max(shell.default - iCount - shell.count, 0) shells.append({ 'id': compactItem(shopShell), 'compactDescr': shopShell.compactDescr, 'type': shell.type, 'icon': '../maps/icons/ammopanel/ammo/%s' % shell.descriptor['icon'][0], 'count': shell.count, 'userCount': shell.default, 'step': casseteCount, 'inventoryCount': iCount, 'goldShellsForCredits': goldShellsForCredits, 'prices': list(sPrice)[:2], 'currency': priceCurrency, 'ammoName': shell.longName, 'tableName': shell.tableName, 'maxAmmo': gun.descriptor['maxAmmo'] }) data.update({ 'autoShells': vehicle.isAutoLoad, 'autoEqip': vehicle.isAutoEquip }) self.as_setDataS(data) return
def __updateAmmo(self): Waiting.show('updateAmmo') credits = g_itemsCache.items.stats.credits shopRqs = yield ShopRequester().request() ammo = {'gunName': '', 'maxAmmo': 0, 'reserved1': False, 'reserved2': False, 'defaultAmmoCount': 0, 'reserved3': 0, 'vehicleLocked': True, 'stateMsg': '', 'stateLevel': 'info', 'shells': [], 'stateWarning': 0} if g_currentVehicle.isPresent(): vehicle = g_currentVehicle.item default_ammo_count = 0 default_ammo = dict(((s.intCD, s.defaultCount) for s in vehicle.shells)) stateWarning = vehicle.isBroken or not vehicle.isCrewFull or not vehicle.isAmmoFull or not g_currentVehicle.isAutoLoadFull() or not g_currentVehicle.isAutoEquipFull() for compactDescr, count in default_ammo.iteritems(): default_ammo_count += count msg, msgLvl = g_currentVehicle.getHangarMessage() ammo.update({'gunName': vehicle.gun.longUserName, 'maxAmmo': vehicle.ammoMaxSize, 'reserved1': not g_currentVehicle.isLocked(), 'reserved2': not g_currentVehicle.isBroken(), 'defaultAmmoCount': default_ammo_count, 'reserved3': 0, 'vehicleLocked': g_currentVehicle.isLocked(), 'stateMsg': msg, 'stateLevel': msgLvl, 'stateWarning': int(stateWarning)}) iAmmo = yield Requester('shell').getFromInventory() sAmmo = yield Requester('shell').getFromShop() iVehicles = yield Requester('vehicle').getFromInventory() oldStyleVehicle = None for v in iVehicles: if v.inventoryId == vehicle.invID: oldStyleVehicle = v break shells = ammo.get('shells') for shell in oldStyleVehicle.shells: shopShell = sAmmo[sAmmo.index(shell)] if shell in sAmmo else None goldAmmoForCredits = shopRqs.isEnabledBuyingGoldShellsForCredits if shopShell: iCount = iAmmo[iAmmo.index(shell)].count if shell in iAmmo else 0 sPrice = (yield shopShell.getPrice()) if shell is not shopShell else (0, 0) if goldAmmoForCredits: shopShell.priceOrder = (sPrice[0] + sPrice[1] * shopRqs.exchangeRateForShellsAndEqs, sPrice[1]) shells.append({'id': gui_items.compactItem(shopShell), 'type': shell.type, 'label': ITEM_TYPES.shell_kindsabbreviation(shell.type), 'icon': '../maps/icons/ammopanel/ammo/%s' % shell.descriptor['icon'][0], 'count': shell.count, 'defaultCount': shell.default, 'inventoryCount': iCount, 'price': sPrice[0 if not sPrice[1] else 1], 'currentcy': 'credits' if not sPrice[1] else 'gold', 'ammoName': shell.longName, 'tableName': shell.tableName}) self.as_setAmmoS(ammo) Waiting.hide('updateAmmo') return
def requestTableData(self, nation, type, filter): Waiting.show('updateShop') AccountSettings.setFilter('shop_current', (nation, type)) AccountSettings.setFilter('shop_' + type, filter) nation = int(nation) if nation >= 0 else None if nation is not None: nation = getNationIndex(nation) filter = list(filter) requestType = [type] checkFits = None checkFitsArtefacts = None checkExtra = False modulesFits = {} vehicleFits = [] extra = [] modulesAllVehicle = [] if type == self._MODULE: typeSize = int(filter.pop(0)) requestType = filter[0:typeSize] filter = filter[typeSize:] fitsType = filter.pop(0) compact = filter.pop(0) if compact == '0': LOG_ERROR('compact value has invalid value: ', compact) Waiting.hide('updateShop') return fitsVehicle = getItemByCompact(compact) checkExtra = True extra = filter[:] checkFits = True if fitsType != 'otherVehicles' else None myVehicles = yield Requester(self._VEHICLE).getFromInventory() modulesAllVehicle = VehicleItemsRequester(myVehicles).getItems( requestType) if fitsType == 'myVehicle': for rType in requestType: modulesFits.update( _getComponentsByType(fitsVehicle, ITEM_TYPE_INDICES[rType])) elif fitsType != 'otherVehicles': for vehicle in myVehicles: for rType in requestType: modulesFits.update( _getComponentsByType(vehicle, ITEM_TYPE_INDICES[rType])) filter = requestType elif type == self._SHELL: filterSize = int(filter.pop(0)) fitsType = filter.pop(filterSize) compact = filter.pop(filterSize) if compact == '0': LOG_ERROR('compact value has invalid value: ', compact) Waiting.hide('updateShop') return fitsVehicle = getItemByCompact(compact) checkFits = True if fitsType != 'otherGuns' else None if fitsType == 'myVehicleGun': for shoot in fitsVehicle.descriptor.gun['shots']: modulesFits[shoot[self._SHELL]['compactDescr']] = True elif fitsType == 'myInventoryGuns': myGuns = yield Requester('vehicleGun').getFromInventory() for gun in myGuns: for shoot in gun.descriptor['shots']: modulesFits[shoot[self._SHELL]['compactDescr']] = True elif fitsType != 'otherGuns': myGuns = yield Requester('vehicleGun').getFromInventory() for gun in myGuns: for shoot in gun.descriptor['shots']: modulesFits[shoot[self._SHELL]['compactDescr']] = True myVehicles = yield Requester(self._VEHICLE).getFromInventory() for vehicle in myVehicles: for shoot in vehicle.descriptor.gun['shots']: modulesFits[shoot[self._SHELL]['compactDescr']] = True elif type == self._VEHICLE: filterSize = int(filter.pop(0)) extra = filter[filterSize:] checkExtra = True filter = filter[0:filterSize] else: fitsType = filter.pop(0) compact = filter.pop(0) if compact == '0': LOG_ERROR('compact value has invalid value: ', compact) Waiting.hide('updateShop') return fitsVehicle = getItemByCompact(compact) extra = filter checkExtra = type in (self._OPTIONAL_DEVICE, self._EQUIPMENT) checkFitsArtefacts = True if fitsType != 'otherVehicles' else None myVehicles = yield Requester(self._VEHICLE).getFromInventory() modulesAllVehicle = VehicleItemsRequester(myVehicles).getItems( requestType) if fitsType == 'myVehicle': vehicleFits = [fitsVehicle] elif fitsType != 'otherVehicles': vehicleFits = [v for v in myVehicles if v.nation == nation ] if nation != None else myVehicles filter = requestType filter = map(lambda w: w.lower(), filter) modulesAll = list() modulesAllInventory = list() for rType in requestType: inv = yield Requester(rType).getFromInventory() modulesAllInventory.extend(inv) shp = yield Requester(rType).getFromShop(nation=nation) modulesAll.extend(shp) unlocks = yield StatsRequester().getUnlocks() shopRqs = yield ShopRequester().request() self.__clearTableData() self.__tableData = [type] modulesAll.sort() for module in modulesAll: extraModuleInfo = None if module.hidden: continue if module.type.lower() not in filter: continue if checkFits is not None: if (module.compactDescr in modulesFits.keys()) != checkFits: continue if checkFitsArtefacts is not None: for veh in vehicleFits: if module.descriptor.checkCompatibilityWithVehicle( veh.descriptor)[0] == checkFitsArtefacts: break else: continue if module.isClipGun(): extraModuleInfo = CLIP_ICON_PATH inventoryCount = 0 vehicleCount = 0 installedIn = '' if module in modulesAllInventory: inventoryCount = 1 if type != self._VEHICLE: inventoryModule = modulesAllInventory[ modulesAllInventory.index(module)] inventoryCount = inventoryModule.count if type in (self._MODULE, self._OPTIONAL_DEVICE, self._EQUIPMENT) and module in modulesAllVehicle: vehModule = modulesAllVehicle[modulesAllVehicle.index(module)] vehicleCount = vehModule.count installedIn = ', '.join( [v.shortName for v in vehModule.vehicles]) if checkExtra: if 'locked' not in extra: if type == self._VEHICLE: compdecs = module.descriptor.type.compactDescr if compdecs not in unlocks: continue elif type not in (self._SHELL, self._OPTIONAL_DEVICE, self._EQUIPMENT ) and module.compactDescr not in unlocks: continue if 'inHangar' not in extra and type not in ( self._OPTIONAL_DEVICE, self._EQUIPMENT): if inventoryCount > 0: continue if 'onVehicle' not in extra: if vehicleCount > 0: continue disabled = '' if type == self._VEHICLE: if BigWorld.player().isLongDisconnectedFromCenter: disabled = MENU.SHOP_ERRORS_CENTERISDOWN if inventoryCount > 0: disabled = MENU.SHOP_ERRORS_INHANGAR else: compdecs = module.descriptor.type.compactDescr if compdecs not in unlocks: disabled = MENU.SHOP_ERRORS_UNLOCKNEEDED elif type not in ( self._SHELL, self._OPTIONAL_DEVICE, self._EQUIPMENT) and module.compactDescr not in unlocks: disabled = MENU.SHOP_ERRORS_UNLOCKNEEDED if not (shopRqs.isEnabledBuyingGoldShellsForCredits and module.itemTypeName == 'shell'): goldAmmoForCredits = shopRqs.isEnabledBuyingGoldEqsForCredits and module.itemTypeName == 'equipment' module.priceOrder = goldAmmoForCredits and ( module.priceOrder[0] + module.priceOrder[1] * shopRqs.exchangeRateForShellsAndEqs, module.priceOrder[1]) valueElement = { 'id': compactItem(module), 'name': module.name if type in (self._OPTIONAL_DEVICE, self._EQUIPMENT) else module.longName, 'desc': getShortDescr(module.tableName), 'inventoryId': None, 'inventoryCount': inventoryCount, 'vehicleCount': vehicleCount, 'credits': module.priceOrder[0], 'gold': module.priceOrder[1], 'price': module.priceOrder, 'currency': 'credits' if module.priceOrder[1] == 0 else 'gold', 'level': module.level, 'nation': module.nation, 'type': module.itemTypeName if type not in (self._VEHICLE, self._OPTIONAL_DEVICE, self._SHELL, self._EQUIPMENT) else module.icon, 'disabled': disabled, 'statusLevel': InventoryVehicle.STATE_LEVEL.WARNING, 'removable': module.descriptor['removable'] if type == self._OPTIONAL_DEVICE else True, 'tankType': module.type if type == self._VEHICLE else type, 'isPremium': module.isPremium if type == self._VEHICLE else False, 'isElite': self.app.tooltipManager.isVehicleElite(module) if type == self._VEHICLE else False, 'itemTypeName': module.itemTypeName, 'goldShellsForCredits': shopRqs.isEnabledBuyingGoldShellsForCredits, 'goldEqsForCredits': shopRqs.isEnabledBuyingGoldEqsForCredits, EXTRA_MODULE_INFO: extraModuleInfo } self.__tableData.append(valueElement) requester = yield StatsRequesterr().request() self._table.as_setGoldS(requester.gold) self._table.as_setCreditsS(requester.credits) self._table.as_setTableS(self.__tableData) Waiting.hide('updateShop') return
def requestTableData(self, nation, type, filter): Waiting.show('updateInventory') AccountSettings.setFilter('inventory_current', (nation, type)) AccountSettings.setFilter('inventory_' + type, filter) nation = int(nation) if nation >= 0 else None if nation is not None: nation = getNationIndex(nation) filter = list(filter) requestType = [type] checkFits = None checkFitsArtefacts = None checkExtra = False modulesFits = {} vehicleFits = [] extra = [] modulesAllVehicle = [] if type == self._MODULE: typeSize = int(filter.pop(0)) requestType = filter[0:typeSize] filter = filter[typeSize:] fitsType = filter.pop(0) fitsVehicle = getItemByCompact(filter.pop(0)) checkExtra = True extra = filter[:] checkFits = True if fitsType != 'otherVehicles' else False myVehicles = yield Requester(self._VEHICLE).getFromInventory() modulesAllVehicle = VehicleItemsRequester(myVehicles).getItems(requestType) if fitsType == 'myVehicle': if fitsVehicle: for rType in requestType: modulesFits.update(_getComponentsByType(fitsVehicle, ITEM_TYPE_INDICES[rType])) else: for vehicle in myVehicles: for rType in requestType: modulesFits.update(_getComponentsByType(vehicle, ITEM_TYPE_INDICES[rType])) filter = requestType elif type == self._SHELL: filterSize = int(filter.pop(0)) fitsType = filter.pop(filterSize) fitsVehicle = getItemByCompact(filter.pop(filterSize)) checkFits = True if fitsType != 'otherGuns' else False if fitsType == 'myVehicleGun': for shoot in fitsVehicle.descriptor.gun['shots']: modulesFits[shoot[self._SHELL]['compactDescr']] = True elif fitsType == 'myInventoryGuns': myGuns = yield Requester('vehicleGun').getFromInventory() for gun in myGuns: for shoot in gun.descriptor['shots']: modulesFits[shoot[self._SHELL]['compactDescr']] = True else: myGuns = yield Requester('vehicleGun').getFromInventory() for gun in myGuns: for shoot in gun.descriptor['shots']: modulesFits[shoot[self._SHELL]['compactDescr']] = True myVehicles = yield Requester(self._VEHICLE).getFromInventory() for vehicle in myVehicles: for shoot in vehicle.descriptor.gun['shots']: modulesFits[shoot[self._SHELL]['compactDescr']] = True elif type == self._VEHICLE: filterSize = int(filter.pop(0)) extra = filter[filterSize:] checkExtra = True filter = filter[0:filterSize] else: fitsType = filter.pop(0) compact = filter.pop(0) fitsVehicle = getItemByCompact(compact) extra = filter checkExtra = type in (self._OPTIONAL_DEVICE, self._EQUIPMENT) checkFitsArtefacts = True if fitsType != 'otherVehicles' else False myVehicles = yield Requester(self._VEHICLE).getFromInventory() modulesAllVehicle = VehicleItemsRequester(myVehicles).getItems(requestType) if fitsType == 'myVehicle': vehicleFits = [fitsVehicle] else: vehicleFits = myVehicles filter = requestType filter = map(lambda w: w.lower(), filter) modulesAll = list() modulesShop = list() for rType in requestType: inv = yield Requester(rType).getFromInventory() shp = yield Requester(rType).getFromShop() modulesShop.extend(shp) modulesAll.extend(inv) vehPrices = {} if type == self._VEHICLE: compactDescrs = [ v.compactDescr for v in modulesAll ] vehPrices = yield StatsRequester().getVehiclesPrices(compactDescrs) vehPrices = dict(zip(compactDescrs, vehPrices)) if type in (self._MODULE, self._OPTIONAL_DEVICE, self._EQUIPMENT): for vehModule in modulesAllVehicle: if vehModule not in modulesAll: if modulesShop.count(vehModule) != 0: modulesAll.append(vehModule) self.__clearTableData() self.__tableData = [type] excludeModules = [] for module in modulesAll: if modulesShop.count(module) != 0: module.priceOrder = modulesShop[modulesShop.index(module)].priceOrder elif constants.IS_DEVELOPMENT: excludeModules.append(module) LOG_ERROR("Not found module %s '%s' (%r) in shop." % (module.type, module.unicName, module.compactDescr)) modulesAll.sort() shopRqs = yield ShopRequester().request() for module in modulesAll: extraModuleInfo = None if module in excludeModules: continue if nation is not None: if module.nation != nation and module.nation != nations.NONE_INDEX: continue if module.type.lower() not in filter: continue if checkFits is not None: if (module.compactDescr in modulesFits.keys()) != checkFits: continue if module.isClipGun(): extraModuleInfo = CLIP_ICON_PATH if checkFitsArtefacts is not None: compatible = False for veh in vehicleFits: if nation is not None and veh.nation != nation: continue compatible |= module.descriptor.checkCompatibilityWithVehicle(veh.descriptor)[0] if compatible != checkFitsArtefacts: continue inventoryCount = 0 vehicleCount = 0 if isinstance(module, VehicleItem): vehicleCount = module.count else: inventoryCount = module.count if type in (self._MODULE, self._OPTIONAL_DEVICE, self._EQUIPMENT) and module in modulesAllVehicle: vehModule = modulesAllVehicle[modulesAllVehicle.index(module)] vehicleCount = vehModule.count if checkExtra: if type == self._VEHICLE and 'brocken' not in extra: if module.repairCost > 0: continue if type == self._VEHICLE and 'locked' not in extra: if module.lock != 0: continue if 'onVehicle' not in extra: if vehicleCount > 0 and inventoryCount == 0: continue disable = '' if type == self._VEHICLE and not module.canSell: disable = makeString(MENU.tankcarousel_vehiclestates(module.getState())) elif type in (self._MODULE, self._OPTIONAL_DEVICE, self._EQUIPMENT) and isinstance(module, VehicleItem): if type == self._OPTIONAL_DEVICE: if not module.descriptor['removable']: disable = makeString(MENU.INVENTORY_DEVICE_ERRORS_NOT_REMOVABLE) else: disable = makeString(MENU.INVENTORY_DEVICE_ERRORS_RESERVED) else: disable = makeString(MENU.INVENTORY_ERRORS_RESERVED) sellPrice = isinstance(module, InventoryVehicle) and vehPrices.get(module.compactDescr, (0, 0)) else: sellPrice = (0, 0) item = g_itemsCache.items.getItemByCD(module.compactDescr) if item is not None: sellPrice = item.sellPrice valueElement = {'id': compactItem(module), 'name': module.name if type in (self._OPTIONAL_DEVICE, self._EQUIPMENT) else module.longName, 'desc': getShortDescr(module.tableName), 'inventoryId': module.inventoryId if isinstance(module, InventoryVehicle) else None, 'inventoryCount': inventoryCount, 'vehicleCount': vehicleCount, 'credits': sellPrice[0], 'gold': sellPrice[1], 'price': sellPrice, 'currency': 'credits' if sellPrice[1] == 0 else 'gold', 'level': module.level, 'nation': module.nation, 'type': module.itemTypeName if type not in (self._VEHICLE, self._OPTIONAL_DEVICE, self._SHELL, self._EQUIPMENT) else module.icon, 'disabled': disable, 'statusLevel': module.getStateLevel() if isinstance(module, InventoryVehicle) else InventoryVehicle.STATE_LEVEL.INFO, 'removable': module.descriptor['removable'] if type == self._OPTIONAL_DEVICE else True, 'tankType': module.type if type == self._VEHICLE else type, 'isPremium': module.isPremium if type == self._VEHICLE else False, 'isElite': self.app.tooltipManager.isVehicleElite(module) if type == self._VEHICLE else False, 'itemTypeName': module.itemTypeName, 'goldShellsForCredits': shopRqs.isEnabledBuyingGoldShellsForCredits, 'goldEqsForCredits': shopRqs.isEnabledBuyingGoldEqsForCredits, EXTRA_MODULE_INFO: extraModuleInfo} self.__tableData.append(valueElement) requester = yield StatsRequesterr().request() self._table.as_setGoldS(requester.gold) self._table.as_setCreditsS(requester.credits) self._table.as_setTableS(self.__tableData) Waiting.hide('updateInventory') return
def __updateTankmen(self, *args): tankmen = yield Requester('tankman').getFromInventory() vcls = yield Requester('vehicle').getFromInventory() slots = yield StatsRequester().getTankmenBerthsCount() berths = yield StatsRequester().getTankmenBerthsCount() berthsPrices = yield StatsRequester().getBerthsPrices() berthPrice = BigWorld.player().shop.getNextBerthPackPrice( berths, berthsPrices) tankmenList = list() tankmenInBarracks = 0 TANKMEN_ROLES_ORDER = { 'commander': 0, 'gunner': 1, 'driver': 2, 'radioman': 3, 'loader': 4 } def tankmenSortFunc(first, second): if first is None or second is None: return 1 res = nationCompareByIndex(first.nation, second.nation) if res: return res elif first.isInTank and not second.isInTank: return -1 elif not first.isInTank and second.isInTank: return 1 if first.isInTank and second.isInTank: tman1vehicle, tman2vehicle = (None, None) for vcl in vcls: if vcl.inventoryId == first.vehicleID: tman1vehicle = vcl if vcl.inventoryId == second.vehicleID: tman2vehicle = vcl if tman1vehicle is not None and tman2vehicle is not None: break if tman1vehicle is not None and tman2vehicle is not None: res = tman1vehicle.__cmp__(tman2vehicle) if res: return res if TANKMEN_ROLES_ORDER[ first.descriptor.role] < TANKMEN_ROLES_ORDER[ second.descriptor.role]: return -1 if TANKMEN_ROLES_ORDER[ first.descriptor.role] > TANKMEN_ROLES_ORDER[ second.descriptor.role]: return 1 if first.lastname < second.lastname: return -1 elif first.lastname > second.lastname: return 1 else: return 1 tankmen.sort(tankmenSortFunc) for tankman in tankmen: if not tankman.isInTank: tankmenInBarracks += 1 if self.filter['nation'] != -1 and tankman.nation != self.filter[ 'nation'] or self.filter[ 'role'] != 'None' and tankman.descriptor.role != self.filter[ 'role'] or self.filter[ 'tankType'] != 'None' and tankman.vehicleType != self.filter[ 'tankType'] or self.filter[ 'location'] == 'tanks' and tankman.isInTank != True or self.filter[ 'location'] == 'barracks' and tankman.isInTank == True or self.filter[ 'nationID'] is not None and ( self.filter['location'] != str(tankman.vehicle.type. id[1]) or self.filter['nationID'] != str(tankman.nation)): continue slot, vehicleID, vehicle = (None, None, None) if tankman.isInTank: for vcl in vcls: if vcl.inventoryId == tankman.vehicleID: vehicle = vcl vehicleID = vehicle.inventoryId break if vehicle is None: LOG_ERROR('Cannot find vehicle for tankman: ', tankman, tankman.descriptor.role, tankman.vehicle.name, tankman.firstname, tankman.lastname) continue for i in range(len(vehicle.crew)): if vehicle.crew[i] == tankman.inventoryId: slot = i break isLocked, msg = self.getTankmanLockMessage( vehicle) if tankman.isInTank else (False, '') isInCurrentTank = tankman.vehicleID == g_currentVehicle.invID if tankman.isInTank and g_currentVehicle.isPresent( ) else False tankmenList.append({ 'firstname': tankman.firstname, 'lastname': tankman.lastname, 'rank': tankman.rank, 'specializationLevel': tankman.roleLevel, 'role': tankman.role, 'vehicleType': tankman.vehicle.type.shortUserString, 'iconFile': tankman.icon, 'rankIconFile': tankman.iconRank, 'roleIconFile': '%s/%s' % (Tankman.ROLE_ICON_PATH_BIG, tankman.iconRole), 'contourIconFile': tankman.vehicleIconContour, 'tankmanID': tankman.inventoryId, 'nationID': tankman.nation, 'typeID': tankman.vehicle.type.id[1], 'slot': slot, 'roleType': tankman.descriptor.role, 'tankType': tankman.vehicleType, 'inTank': tankman.isInTank, 'inCurrentTank': isInCurrentTank, 'vehicleID': vehicleID, 'compact': tankman.pack(), 'locked': isLocked, 'lockMessage': msg, 'vehicleBroken': vehicle.repairCost > 0 if tankman.isInTank else None, 'isInSelfVehicleClass': vehicle.type == tankman.vehicleType if tankman.isInTank else True, 'isInSelfVehicleType': vehicle.shortName == tankman.vehicle.type.shortUserString if tankman.isInTank else True }) self.as_setTankmenS(len(tankmen), slots, tankmenInBarracks, BigWorld.wg_getGoldFormat(berthPrice), berthsPrices[1], tankmenList) return