def __init__(self, avatar=None):
super(ArenaDataProvider, self).__init__()
self.__playerTeam = avatar_getter.getPlayerTeam(avatar)
self.__teamsOnArena = range(
1,
avatar_getter.getMaxTeamsOnArena(avatar) + 1)
self.__playerVehicleID = avatar_getter.getPlayerVehicleID(avatar)
self.__vInfoVOs = {}
self.__vStatsVOs = VehicleArenaStatsDict()
self.__prbStats = {}
self.__playersVIDs = {}
self.__viStatsVOs = VehicleArenaInteractiveStatsDict()
self.__teamsVIStats = {}
fo_precache.add(settings.UNKNOWN_CONTOUR_ICON_RES_PATH)
def __init__(self):
super(ArenaDataProvider, self).__init__()
self.__playerTeam = avatar_getter.getPlayerTeam()
self.__playerVehicleID = avatar_getter.getPlayerVehicleID()
self.__vInfoVOs = {}
self.__vStatsVOs = VehicleArenaStatsDict()
self.__prbStats = {}
self.__playersVIDs = {}
self.__viStatsVOs = VehicleArenaInteractiveStatsDict()
def __init__(self, avatar = None):
super(ArenaDataProvider, self).__init__()
self.__playerTeam = avatar_getter.getPlayerTeam(avatar)
self.__teamsOnArena = range(1, avatar_getter.getMaxTeamsOnArena(avatar) + 1)
self.__playerVehicleID = avatar_getter.getPlayerVehicleID(avatar)
self.__vInfoVOs = {}
self.__vStatsVOs = VehicleArenaStatsDict()
self.__prbStats = {}
self.__playersVIDs = {}
self.__viStatsVOs = VehicleArenaInteractiveStatsDict()
self.__teamsVIStats = {}
fo_precache.add(settings.UNKNOWN_CONTOUR_ICON_RES_PATH)
class ArenaDataProvider(object):
__slots__ = ('__playerTeam', '__playerVehicleID', '__vInfoVOs', '__vStatsVOs', '__viStatsVOs', '__prbStats', '__playersVIDs', '__weakref__', '__teamsOnArena', '__teamsVIStats')
def __init__(self, avatar = None):
super(ArenaDataProvider, self).__init__()
self.__playerTeam = avatar_getter.getPlayerTeam(avatar)
self.__teamsOnArena = range(1, avatar_getter.getMaxTeamsOnArena(avatar) + 1)
self.__playerVehicleID = avatar_getter.getPlayerVehicleID(avatar)
self.__vInfoVOs = {}
self.__vStatsVOs = VehicleArenaStatsDict()
self.__prbStats = {}
self.__playersVIDs = {}
self.__viStatsVOs = VehicleArenaInteractiveStatsDict()
self.__teamsVIStats = {}
fo_precache.add(settings.UNKNOWN_CONTOUR_ICON_RES_PATH)
def __del__(self):
LOG_DEBUG('Deleted:', self)
def clearInfo(self):
self.__vInfoVOs.clear()
self.__prbStats.clear()
self.__playersVIDs.clear()
def clearStats(self):
self.__vStatsVOs.clear()
self.__viStatsVOs.clear()
self.__teamsVIStats.clear()
def clear(self):
fo_precache.clear()
self.clearInfo()
self.clearStats()
def defaultInfo(self):
self.clearInfo()
for team in self.getTeamsOnArena():
self.__prbStats[team] = defaultdict(list)
def buildVehiclesData(self, vehicles, arenaGuiType):
self.defaultInfo()
for vID, vData in vehicles.iteritems():
self.__addVehicleInfoVO(vID, VehicleArenaInfoVO(vID, **vData))
self.__findSquads(arenaGuiType)
def buildStatsData(self, stats):
self.clearStats()
for vID, vStats in stats.iteritems():
self.__vStatsVOs[vID] = VehicleArenaStatsVO(vID, **vStats)
def addVehicleInfo(self, vID, vInfo, arenaGuiType):
vInfoVO = None
if vID not in self.__vInfoVOs:
vInfoVO = VehicleArenaInfoVO(vID, **vInfo)
if self.__addVehicleInfoVO(vID, vInfoVO):
self.__findSquads(arenaGuiType)
else:
LOG_WARNING('Vehicle already exists', self.__vInfoVOs[vID])
return vInfoVO
def updateVehicleInfo(self, vID, vInfo, arenaGuiType):
vInfoVO = self.__vInfoVOs[vID]
flags = vInfoVO.update(**vInfo)
prebattleId = vInfoVO.prebattleID
if flags & settings.INVALIDATE_OP.PREBATTLE_CHANGED:
self.__updateStats(vInfoVO.team, prebattleId, vID)
self.__findSquads(arenaGuiType)
dbID = vInfoVO.player.accountDBID
if dbID:
self.__playersVIDs[dbID] = vID
return (flags, vInfoVO)
def updateVehicleStatus(self, vID, vInfo):
vInfoVO = self.__vInfoVOs[vID]
flags = vInfoVO.updateVehicleStatus(**vInfo)
return (flags, vInfoVO)
def updatePlayerStatus(self, vID, vInfo):
vInfoVO = self.__vInfoVOs[vID]
flags = vInfoVO.updatePlayerStatus(**vInfo)
return (flags, vInfoVO)
def updateVehicleStats(self, vID, vStats):
vStatsVO = self.__vStatsVOs[vID]
flags = vStatsVO.update(**vStats)
return (flags, vStatsVO)
def updateVehicleInteractiveStats(self, iStats):
self.__viStatsVOs.clear()
self.__teamsVIStats.clear()
for (vID, _), iStat in iStats.iteritems():
vStatsVO = self.__viStatsVOs[vID]
vStatsVO.update(*iStat)
vInfo = self.__vInfoVOs[vID]
team = vInfo.team
if team in self.__teamsVIStats:
self.__teamsVIStats[team][vID] = vStatsVO
else:
self.__teamsVIStats[team] = {vID: vStatsVO}
def isRequiredDataExists(self):
return self.__checkRequiredData()
def getTeamsOnArena(self):
return self.__teamsOnArena
def getAllyTeams(self):
return (self.__playerTeam,)
def getEnemyTeams(self):
allyTeams = self.getAllyTeams()
return filter(lambda t: t not in allyTeams, self.__teamsOnArena)
def isEnemyTeam(self, team):
return team not in self.getAllyTeams()
def isAllyTeam(self, team):
return team in self.getAllyTeams()
def isMultipleTeams(self):
return len(self.__teamsOnArena) > TEAMS_IN_ARENA.MIN_TEAMS
def getMultiTeamsType(self):
if self.isMultipleTeams():
squadTeamNumber = 0
for team in self.__prbStats.itervalues():
squads = filter(lambda item: len(item[1]) in settings.SQUAD_RANGE_TO_SHOW, team.iteritems())
if len(squads):
squadTeamNumber += 1
if squadTeamNumber == 0:
return MULTIPLE_TEAMS_TYPE.FFA
if squadTeamNumber == len(self.__teamsOnArena):
return MULTIPLE_TEAMS_TYPE.TDM
return MULTIPLE_TEAMS_TYPE.MIXED
else:
return None
def getMultiTeamsIndexes(self):
lastTeamIdx = 0
result = {}
for team in self.__teamsOnArena:
teamIdx = 0
vInfoVO, _, _ = next(self._getVehiclesIterator(lambda v: v.team == team), (None, None, None))
if vInfoVO is not None and vInfoVO.prebattleID > 0:
lastTeamIdx += 1
teamIdx = lastTeamIdx
result[team] = teamIdx
return result
def getTeamIDsIterator(self):
allyTeams = self.getAllyTeams()
for teamIdx in self.__teamsOnArena:
yield (teamIdx not in allyTeams, teamIdx)
def getNumberOfTeam(self, enemy = False):
if enemy:
return first(self.getEnemyTeams())
else:
return self.__playerTeam
def getPlayerVehicleID(self):
return self.__playerVehicleID
def getVehicleInfo(self, vID = None):
if vID is None:
vID = self.getPlayerVehicleID()
try:
result = self.__vInfoVOs[vID]
except KeyError:
result = VehicleArenaInfoVO(vID)
return result
def getVehicleStats(self, vID = None):
if vID is None:
vID = self.getPlayerVehicleID()
return self.__vStatsVOs[vID]
def getTeamStats(self, team = None):
if team is None:
team = self.__playerTeam
return self.__teamsVIStats.get(team, {})
def getVehIDsByPrebattleID(self, team, prebattleID):
vehIDs = None
if team in self.__prbStats and prebattleID in self.__prbStats[team]:
vehIDs = list(self.__prbStats[team][prebattleID])
return vehIDs
def getPrbVehCount(self, team, prebattleID):
vehIDs = self.getVehIDsByPrebattleID(team, prebattleID)
if vehIDs:
return len(vehIDs)
return 0
def getVehicleInteractiveStats(self, vID = None):
if vID is None:
vID = self.getPlayerVehicleID()
return self.__viStatsVOs[vID]
def getPlayerGuiProps(self, vID, team):
if team in self.getAllyTeams():
if self.isSquadMan(vID=vID):
return PLAYER_GUI_PROPS.squadman
if self.isTeamKiller(vID=vID):
return PLAYER_GUI_PROPS.teamKiller
return PLAYER_GUI_PROPS.ally
return PLAYER_GUI_PROPS.enemy
def isSquadMan(self, vID, prebattleID = None):
if prebattleID is None:
if not self.__playerVehicleID:
self.__playerVehicleID = avatar_getter.getPlayerVehicleID()
if self.__playerVehicleID in self.__vInfoVOs:
prebattleID = self.__vInfoVOs[self.__playerVehicleID].prebattleID
return self.__getStateFlag(vID, 'isSquadMan', playerTeam=self.__playerTeam, prebattleID=prebattleID)
def isTeamKiller(self, vID):
return self.__getStateFlag(vID, 'isTeamKiller', playerTeam=self.__playerTeam)
def isObserver(self, vID):
return self.__getStateFlag(vID, 'isObserver')
def getVehIDByAccDBID(self, dbID):
try:
vID = self.__playersVIDs[dbID]
except KeyError:
vID = 0
return vID
def getTeamIterator(self, teamIdx):
return self._getVehiclesIterator(lambda v: v.team == teamIdx)
def getVehiclesIterator(self, enemy = False):
if enemy:
teams = self.getEnemyTeams()
else:
teams = self.getAllyTeams()
return self._getVehiclesIterator(lambda v: v.team in teams)
def getAllVehiclesIterator(self):
return self._getVehiclesIterator()
def getAllVehiclesIteratorByTeamScore(self):
def sortByScore(vInfoVOX, vInfoVOY):
teamStatsX = self.getTeamStats(vInfoVOX.team)
teamStatsY = self.getTeamStats(vInfoVOY.team)
teamScoreX = sum(map(operator.attrgetter('winPoints'), teamStatsX.itervalues()))
teamScoreY = sum(map(operator.attrgetter('winPoints'), teamStatsY.itervalues()))
res = cmp(teamScoreY, teamScoreX)
if res:
return res
return cmp(vInfoVOX, vInfoVOY)
return self._getVehiclesIterator(sortFunction=sortByScore)
def getAllVehiclesIDsIterator(self):
for vInfoVO, _, _ in self.getAllVehiclesIterator():
yield vInfoVO.vehicleID
def getAllVehiclesIDs(self):
return list(self.getAllVehiclesIDsIterator())
def getVehiclesIDsIterator(self, enemy = False):
for vInfoVO, _, _ in self.getVehiclesIterator(enemy):
yield vInfoVO.vehicleID
def getVehiclesIDs(self, enemy = False):
return list(self.getVehiclesIDsIterator(enemy))
def _getVehiclesIterator(self, filterPredicate = None, sortFunction = None):
if filterPredicate is None:
data = self.__vInfoVOs.itervalues()
else:
data = filter(filterPredicate, self.__vInfoVOs.itervalues())
if sortFunction is None:
sortedData = sorted(data)
else:
sortedData = sorted(data, cmp=sortFunction)
for vInfoVO in sortedData:
yield (vInfoVO, self.__vStatsVOs[vInfoVO.vehicleID], self.__viStatsVOs[vInfoVO.vehicleID])
return
def __findSquads(self, arenaGuiType):
if arenaGuiType not in (ARENA_GUI_TYPE.RANDOM, ARENA_GUI_TYPE.EVENT_BATTLES):
return
for team in self.__prbStats.itervalues():
squads = filter(lambda item: len(item[1]) in settings.SQUAD_RANGE_TO_SHOW, team.iteritems())
if len(squads):
squads = sorted(squads, key=lambda item: item[0])
for index, (prbID, vIDs) in enumerate(squads):
squadsIndex = index + 1
for vID in vIDs:
vInfoVO = self.__vInfoVOs[vID]
vInfoVO.squadIndex = squadsIndex
vInfoVO.updatePlayerStatus(isSquadMan=True)
def __addVehicleInfoVO(self, vID, vInfoVO):
dbID = vInfoVO.player.accountDBID
hasPrbID = False
self.__vInfoVOs[vID] = vInfoVO
if dbID:
self.__playersVIDs[dbID] = vID
prebattleID = vInfoVO.prebattleID
if prebattleID > 0:
self.__updateStats(vInfoVO.team, prebattleID, vID)
hasPrbID = True
return hasPrbID
def __getStateFlag(self, vID, flagName, **kwargs):
result = False
if vID in self.__vInfoVOs:
result = operator.methodcaller(flagName, **kwargs)(self.__vInfoVOs[vID])
return result
def __checkRequiredData(self):
result = self.__playerTeam > 0 and self.__playerVehicleID > 0
if not result:
requestToFind = False
self.__playerTeam = avatar_getter.getPlayerTeam()
if not self.__playerTeam:
requestToFind = True
LOG_NOTE("Player's team not found.")
self.__playerVehicleID = avatar_getter.getPlayerVehicleID()
if not self.__playerVehicleID:
requestToFind = True
LOG_NOTE("Player's vehicle ID not found.")
if not requestToFind:
return
playerName = avatar_getter.getPlayerName()
LOG_NOTE('Uses slow player search by name')
for vo in self.__vInfoVOs.itervalues():
if vo.player.name == playerName:
self.__playerTeam = vo.team
self.__playerVehicleID = vo.vehicleID
result = True
break
return result
def __updateStats(self, team, prebattleID, vID):
if prebattleID not in self.__prbStats[team]:
self.__prbStats[team][prebattleID] = set()
self.__prbStats[team][prebattleID].add(vID)
class ArenaDataProvider(object):
__slots__ = ('__playerTeam', '__playerVehicleID', '__vInfoVOs', '__vStatsVOs', '__viStatsVOs', '__prbStats', '__playersVIDs', '__weakref__')
def __init__(self):
super(ArenaDataProvider, self).__init__()
self.__playerTeam = avatar_getter.getPlayerTeam()
self.__playerVehicleID = avatar_getter.getPlayerVehicleID()
self.__vInfoVOs = {}
self.__vStatsVOs = VehicleArenaStatsDict()
self.__prbStats = {}
self.__playersVIDs = {}
self.__viStatsVOs = VehicleArenaInteractiveStatsDict()
def __del__(self):
LOG_DEBUG('Deleted:', self)
def clearInfo(self):
self.__vInfoVOs.clear()
self.__prbStats.clear()
self.__playersVIDs.clear()
def clearStats(self):
self.__vStatsVOs.clear()
self.__viStatsVOs.clear()
def clear(self):
self.clearInfo()
self.clearStats()
def defaultInfo(self):
self.clearInfo()
for team in TEAM_RANGE:
self.__prbStats[team] = defaultdict(list)
def buildVehiclesData(self, vehicles, arenaGuiType):
self.defaultInfo()
for vID, vData in vehicles.iteritems():
self.__addVehicleInfoVO(vID, VehicleArenaInfoVO(vID, **vData))
self.__findSquads(arenaGuiType)
def buildStatsData(self, stats):
self.clearStats()
for vID, vStats in stats.iteritems():
self.__vStatsVOs[vID] = VehicleArenaStatsVO(vID, **vStats)
def addVehicleInfo(self, vID, vInfo, arenaGuiType):
vInfoVO = None
if vID not in self.__vInfoVOs:
vInfoVO = VehicleArenaInfoVO(vID, **vInfo)
if self.__addVehicleInfoVO(vID, vInfoVO):
self.__findSquads(arenaGuiType)
else:
LOG_WARNING('Vehicle already exists', self.__vInfoVOs[vID])
return vInfoVO
def updateVehicleInfo(self, vID, vInfo, arenaGuiType):
vInfoVO = self.__vInfoVOs[vID]
flags = vInfoVO.update(**vInfo)
prebattleId = vInfoVO.prebattleID
if flags & INVALIDATE_OP.PREBATTLE_CHANGED:
self.__prbStats[vInfoVO.team][prebattleId].append(vID)
self.__findSquads(arenaGuiType)
dbID = vInfoVO.player.accountDBID
if dbID:
self.__playersVIDs[dbID] = vID
return (flags, vInfoVO)
def updateVehicleStatus(self, vID, vInfo):
vInfoVO = self.__vInfoVOs[vID]
flags = vInfoVO.updateVehicleStatus(**vInfo)
return (flags, vInfoVO)
def updatePlayerStatus(self, vID, vInfo):
vInfoVO = self.__vInfoVOs[vID]
flags = vInfoVO.updatePlayerStatus(**vInfo)
return (flags, vInfoVO)
def updateVehicleStats(self, vID, vStats):
vStatsVO = self.__vStatsVOs[vID]
flags = vStatsVO.update(**vStats)
return (flags, vStatsVO)
def updateVehicleInteractiveStats(self, iStats):
self.__viStatsVOs.clear()
for (vID, _), iStat in iStats.iteritems():
vStatsVO = self.__viStatsVOs[vID]
vStatsVO.update(*iStat)
def isRequiredDataExists(self):
return self.__checkRequiredData()
def getNumberOfTeam(self, enemy = False):
if enemy:
return invertTeam(self.__playerTeam)
return self.__playerTeam
def getPlayerVehicleID(self):
return self.__playerVehicleID
def getVehicleInfo(self, vID = None):
if vID is None:
vID = self.getPlayerVehicleID()
try:
result = self.__vInfoVOs[vID]
except KeyError:
result = VehicleArenaInfoVO(vID)
return result
def getVehicleStats(self, vID = None):
if vID is None:
vID = self.getPlayerVehicleID()
return self.__vStatsVOs[vID]
def getVehIDsByPrebattleID(self, team, prebattleID):
vehIDs = None
if team in self.__prbStats and prebattleID in self.__prbStats[team]:
vehIDs = self.__prbStats[team][prebattleID]
return vehIDs
def getPrbVehCount(self, team, prebattleID):
vehIDs = self.getVehIDsByPrebattleID(team, prebattleID)
if vehIDs:
return len(vehIDs)
return 0
def getVehicleInteractiveStats(self, vID = None):
if vID is None:
vID = self.getPlayerVehicleID()
return self.__viStatsVOs[vID]
def isSquadMan(self, vID, prebattleID = None):
if prebattleID is None:
if not self.__playerVehicleID:
self.__playerVehicleID = avatar_getter.getPlayerVehicleID()
if self.__playerVehicleID in self.__vInfoVOs:
prebattleID = self.__vInfoVOs[self.__playerVehicleID].prebattleID
return self.__getStateFlag(vID, 'isSquadMan', playerTeam=self.__playerTeam, prebattleID=prebattleID)
def isTeamKiller(self, vID):
return self.__getStateFlag(vID, 'isTeamKiller', playerTeam=self.__playerTeam)
def isObserver(self, vID):
return self.__getStateFlag(vID, 'isObserver')
def getVehIDByAccDBID(self, dbID):
try:
vID = self.__playersVIDs[dbID]
except KeyError:
vID = 0
return vID
def getTeamIterator(self, enemy = False):
team = self.getNumberOfTeam(enemy)
if not team:
LOG_NOTE('Team not found')
data = []
else:
data = sorted(filter(lambda vInfoVO: vInfoVO.team == team, self.__vInfoVOs.itervalues()))
for vInfoVO in data:
yield (vInfoVO, self.__vStatsVOs[vInfoVO.vehicleID], self.__viStatsVOs[vInfoVO.vehicleID])
def getVehiclesIDsIterator(self, enemy = False):
team = self.getNumberOfTeam(enemy)
if not team:
LOG_NOTE('Team not found')
data = []
else:
data = sorted(filter(lambda vInfoVO: vInfoVO.team == team, self.__vInfoVOs.itervalues()))
for vInfoVO in data:
yield vInfoVO.vehicleID
def getVehiclesIDs(self, enemy = False):
return list(self.getVehiclesIDsIterator(enemy))
def __findSquads(self, arenaGuiType):
if arenaGuiType not in (ARENA_GUI_TYPE.RANDOM, ARENA_GUI_TYPE.EVENT_BATTLES):
return
for team in self.__prbStats.iterkeys():
squads = filter(lambda item: len(item[1]) in SQUAD_RANGE_TO_SHOW, self.__prbStats[team].iteritems())
if len(squads):
squads = sorted(squads, key=lambda item: item[0])
for index, (prbID, vIDs) in enumerate(squads):
squadsIndex = index + 1
for vID in vIDs:
vInfoVO = self.__vInfoVOs[vID]
vInfoVO.squadIndex = squadsIndex
vInfoVO.updatePlayerStatus(isSquadMan=True)
def __addVehicleInfoVO(self, vID, vInfoVO):
dbID = vInfoVO.player.accountDBID
hasPrbID = False
self.__vInfoVOs[vID] = vInfoVO
if dbID:
self.__playersVIDs[dbID] = vID
prebattleID = vInfoVO.prebattleID
if prebattleID > 0:
self.__prbStats[vInfoVO.team][prebattleID].append(vID)
hasPrbID = True
return hasPrbID
def __getStateFlag(self, vID, flagName, **kwargs):
result = False
if vID in self.__vInfoVOs:
result = getattr(self.__vInfoVOs[vID], flagName)(**kwargs)
return result
def __checkRequiredData(self):
result = self.__playerTeam > 0 and self.__playerVehicleID > 0
if not result:
requestToFind = False
self.__playerTeam = avatar_getter.getPlayerTeam()
if not self.__playerTeam:
requestToFind = True
LOG_NOTE("Player's team not found.")
self.__playerVehicleID = avatar_getter.getPlayerVehicleID()
if not self.__playerVehicleID:
requestToFind = True
LOG_NOTE("Player's vehicle ID not found.")
if not requestToFind:
return
playerName = avatar_getter.getPlayerName()
LOG_NOTE('Uses slow player search by name')
for vo in self.__vInfoVOs.itervalues():
if vo.player.name == playerName:
self.__playerTeam = vo.team
self.__playerVehicleID = vo.vehicleID
result = True
break
return result
class ArenaDataProvider(object):
__slots__ = ('__playerTeam', '__playerVehicleID', '__vInfoVOs',
'__vStatsVOs', '__viStatsVOs', '__prbStats', '__playersVIDs',
'__weakref__', '__teamsOnArena', '__teamsVIStats')
def __init__(self, avatar=None):
super(ArenaDataProvider, self).__init__()
self.__playerTeam = avatar_getter.getPlayerTeam(avatar)
self.__teamsOnArena = range(
1,
avatar_getter.getMaxTeamsOnArena(avatar) + 1)
self.__playerVehicleID = avatar_getter.getPlayerVehicleID(avatar)
self.__vInfoVOs = {}
self.__vStatsVOs = VehicleArenaStatsDict()
self.__prbStats = {}
self.__playersVIDs = {}
self.__viStatsVOs = VehicleArenaInteractiveStatsDict()
self.__teamsVIStats = {}
fo_precache.add(settings.UNKNOWN_CONTOUR_ICON_RES_PATH)
def __del__(self):
LOG_DEBUG('Deleted:', self)
def clearInfo(self):
self.__vInfoVOs.clear()
self.__prbStats.clear()
self.__playersVIDs.clear()
def clearStats(self):
self.__vStatsVOs.clear()
self.__viStatsVOs.clear()
self.__teamsVIStats.clear()
def clear(self):
fo_precache.clear()
self.clearInfo()
self.clearStats()
def defaultInfo(self):
self.clearInfo()
for team in self.getTeamsOnArena():
self.__prbStats[team] = defaultdict(list)
def buildVehiclesData(self, vehicles, arenaGuiType):
self.defaultInfo()
for vID, vData in vehicles.iteritems():
self.__addVehicleInfoVO(vID, VehicleArenaInfoVO(vID, **vData))
self.__findSquads(arenaGuiType)
def buildStatsData(self, stats):
self.clearStats()
for vID, vStats in stats.iteritems():
self.__vStatsVOs[vID] = VehicleArenaStatsVO(vID, **vStats)
def addVehicleInfo(self, vID, vInfo, arenaGuiType):
vInfoVO = None
if vID not in self.__vInfoVOs:
vInfoVO = VehicleArenaInfoVO(vID, **vInfo)
if self.__addVehicleInfoVO(vID, vInfoVO):
self.__findSquads(arenaGuiType)
else:
LOG_WARNING('Vehicle already exists', self.__vInfoVOs[vID])
return vInfoVO
def updateVehicleInfo(self, vID, vInfo, arenaGuiType):
vInfoVO = self.__vInfoVOs[vID]
flags = vInfoVO.update(**vInfo)
prebattleId = vInfoVO.prebattleID
if flags & settings.INVALIDATE_OP.PREBATTLE_CHANGED:
self.__updateStats(vInfoVO.team, prebattleId, vID)
self.__findSquads(arenaGuiType)
dbID = vInfoVO.player.accountDBID
if dbID:
self.__playersVIDs[dbID] = vID
return (flags, vInfoVO)
def updateVehicleStatus(self, vID, vInfo):
vInfoVO = self.__vInfoVOs[vID]
flags = vInfoVO.updateVehicleStatus(**vInfo)
return (flags, vInfoVO)
def updatePlayerStatus(self, vID, vInfo):
vInfoVO = self.__vInfoVOs[vID]
flags = vInfoVO.updatePlayerStatus(**vInfo)
return (flags, vInfoVO)
def updateVehicleStats(self, vID, vStats):
vStatsVO = self.__vStatsVOs[vID]
flags = vStatsVO.update(**vStats)
return (flags, vStatsVO)
def updateVehicleInteractiveStats(self, iStats):
self.__viStatsVOs.clear()
self.__teamsVIStats.clear()
for (vID, _), iStat in iStats.iteritems():
vStatsVO = self.__viStatsVOs[vID]
vStatsVO.update(*iStat)
vInfo = self.__vInfoVOs[vID]
team = vInfo.team
if team in self.__teamsVIStats:
self.__teamsVIStats[team][vID] = vStatsVO
else:
self.__teamsVIStats[team] = {vID: vStatsVO}
def isRequiredDataExists(self):
return self.__checkRequiredData()
def getTeamsOnArena(self):
return self.__teamsOnArena
def getAllyTeams(self):
return (self.__playerTeam, )
def getEnemyTeams(self):
allyTeams = self.getAllyTeams()
return filter(lambda t: t not in allyTeams, self.__teamsOnArena)
def isEnemyTeam(self, team):
return team not in self.getAllyTeams()
def isAllyTeam(self, team):
return team in self.getAllyTeams()
def isMultipleTeams(self):
return len(self.__teamsOnArena) > TEAMS_IN_ARENA.MIN_TEAMS
def getMultiTeamsType(self):
if self.isMultipleTeams():
squadTeamNumber = 0
for team in self.__prbStats.itervalues():
squads = filter(
lambda item: len(item[1]) in settings.SQUAD_RANGE_TO_SHOW,
team.iteritems())
if len(squads):
squadTeamNumber += 1
if squadTeamNumber == 0:
return MULTIPLE_TEAMS_TYPE.FFA
if squadTeamNumber == len(self.__teamsOnArena):
return MULTIPLE_TEAMS_TYPE.TDM
return MULTIPLE_TEAMS_TYPE.MIXED
def getMultiTeamsIndexes(self):
lastTeamIdx = 0
result = {}
for team in self.__teamsOnArena:
teamIdx = 0
vInfoVO, _, _ = next(
self._getVehiclesIterator(lambda v: v.team == team),
(None, None, None))
if vInfoVO is not None and vInfoVO.prebattleID > 0:
lastTeamIdx += 1
teamIdx = lastTeamIdx
result[team] = teamIdx
return result
def getTeamIDsIterator(self):
allyTeams = self.getAllyTeams()
for teamIdx in self.__teamsOnArena:
yield (teamIdx not in allyTeams, teamIdx)
def getNumberOfTeam(self, enemy=False):
if enemy:
return first(self.getEnemyTeams())
else:
return self.__playerTeam
def getPlayerVehicleID(self):
return self.__playerVehicleID
def getVehicleInfo(self, vID=None):
if vID is None:
vID = self.getPlayerVehicleID()
try:
result = self.__vInfoVOs[vID]
except KeyError:
result = VehicleArenaInfoVO(vID)
return result
def getVehicleStats(self, vID=None):
if vID is None:
vID = self.getPlayerVehicleID()
return self.__vStatsVOs[vID]
def getTeamStats(self, team=None):
if team is None:
team = self.__playerTeam
return self.__teamsVIStats.get(team, {})
def getVehIDsByPrebattleID(self, team, prebattleID):
vehIDs = None
if team in self.__prbStats and prebattleID in self.__prbStats[team]:
vehIDs = list(self.__prbStats[team][prebattleID])
return vehIDs
def getPrbVehCount(self, team, prebattleID):
vehIDs = self.getVehIDsByPrebattleID(team, prebattleID)
if vehIDs:
return len(vehIDs)
return 0
def getVehicleInteractiveStats(self, vID=None):
if vID is None:
vID = self.getPlayerVehicleID()
return self.__viStatsVOs[vID]
def getPlayerGuiProps(self, vID, team):
if team in self.getAllyTeams():
if self.isSquadMan(vID=vID):
return PLAYER_GUI_PROPS.squadman
if self.isTeamKiller(vID=vID):
return PLAYER_GUI_PROPS.teamKiller
return PLAYER_GUI_PROPS.ally
return PLAYER_GUI_PROPS.enemy
def isSquadMan(self, vID, prebattleID=None):
if prebattleID is None:
if not self.__playerVehicleID:
self.__playerVehicleID = avatar_getter.getPlayerVehicleID()
if self.__playerVehicleID in self.__vInfoVOs:
prebattleID = self.__vInfoVOs[
self.__playerVehicleID].prebattleID
return self.__getStateFlag(vID,
'isSquadMan',
playerTeam=self.__playerTeam,
prebattleID=prebattleID)
def isTeamKiller(self, vID):
return self.__getStateFlag(vID,
'isTeamKiller',
playerTeam=self.__playerTeam)
def isObserver(self, vID):
return self.__getStateFlag(vID, 'isObserver')
def getVehIDByAccDBID(self, dbID):
try:
vID = self.__playersVIDs[dbID]
except KeyError:
vID = 0
return vID
def getTeamIterator(self, teamIdx):
return self._getVehiclesIterator(lambda v: v.team == teamIdx)
def getVehiclesIterator(self, enemy=False):
if enemy:
teams = self.getEnemyTeams()
else:
teams = self.getAllyTeams()
return self._getVehiclesIterator(lambda v: v.team in teams)
def getAllVehiclesIterator(self):
return self._getVehiclesIterator()
def getAllVehiclesIteratorByTeamScore(self):
def sortByScore(vInfoVOX, vInfoVOY):
teamStatsX = self.getTeamStats(vInfoVOX.team)
teamStatsY = self.getTeamStats(vInfoVOY.team)
teamScoreX = sum(
map(operator.attrgetter('winPoints'), teamStatsX.itervalues()))
teamScoreY = sum(
map(operator.attrgetter('winPoints'), teamStatsY.itervalues()))
res = cmp(teamScoreY, teamScoreX)
if res:
return res
return cmp(vInfoVOX, vInfoVOY)
return self._getVehiclesIterator(sortFunction=sortByScore)
def getAllVehiclesIDsIterator(self):
for vInfoVO, _, _ in self.getAllVehiclesIterator():
yield vInfoVO.vehicleID
def getAllVehiclesIDs(self):
return list(self.getAllVehiclesIDsIterator())
def getVehiclesIDsIterator(self, enemy=False):
for vInfoVO, _, _ in self.getVehiclesIterator(enemy):
yield vInfoVO.vehicleID
def getVehiclesIDs(self, enemy=False):
return list(self.getVehiclesIDsIterator(enemy))
def _getVehiclesIterator(self, filterPredicate=None, sortFunction=None):
if filterPredicate is None:
data = self.__vInfoVOs.itervalues()
else:
data = filter(filterPredicate, self.__vInfoVOs.itervalues())
if sortFunction is None:
sortedData = sorted(data)
else:
sortedData = sorted(data, cmp=sortFunction)
for vInfoVO in sortedData:
yield (vInfoVO, self.__vStatsVOs[vInfoVO.vehicleID],
self.__viStatsVOs[vInfoVO.vehicleID])
def __findSquads(self, arenaGuiType):
if arenaGuiType not in (ARENA_GUI_TYPE.RANDOM,
ARENA_GUI_TYPE.EVENT_BATTLES):
return
for team in self.__prbStats.itervalues():
squads = filter(
lambda item: len(item[1]) in settings.SQUAD_RANGE_TO_SHOW,
team.iteritems())
if len(squads):
squads = sorted(squads, key=lambda item: item[0])
for index, (prbID, vIDs) in enumerate(squads):
squadsIndex = index + 1
for vID in vIDs:
vInfoVO = self.__vInfoVOs[vID]
vInfoVO.squadIndex = squadsIndex
vInfoVO.updatePlayerStatus(isSquadMan=True)
def __addVehicleInfoVO(self, vID, vInfoVO):
dbID = vInfoVO.player.accountDBID
hasPrbID = False
self.__vInfoVOs[vID] = vInfoVO
if dbID:
self.__playersVIDs[dbID] = vID
prebattleID = vInfoVO.prebattleID
if prebattleID > 0:
self.__updateStats(vInfoVO.team, prebattleID, vID)
hasPrbID = True
return hasPrbID
def __getStateFlag(self, vID, flagName, **kwargs):
result = False
if vID in self.__vInfoVOs:
result = operator.methodcaller(flagName,
**kwargs)(self.__vInfoVOs[vID])
return result
def __checkRequiredData(self):
result = self.__playerTeam > 0 and self.__playerVehicleID > 0
if not result:
requestToFind = False
self.__playerTeam = avatar_getter.getPlayerTeam()
if not self.__playerTeam:
requestToFind = True
LOG_NOTE("Player's team not found.")
self.__playerVehicleID = avatar_getter.getPlayerVehicleID()
if not self.__playerVehicleID:
requestToFind = True
LOG_NOTE("Player's vehicle ID not found.")
if not requestToFind:
return
playerName = avatar_getter.getPlayerName()
LOG_NOTE('Uses slow player search by name')
for vo in self.__vInfoVOs.itervalues():
if vo.player.name == playerName:
self.__playerTeam = vo.team
self.__playerVehicleID = vo.vehicleID
result = True
break
return result
def __updateStats(self, team, prebattleID, vID):
if prebattleID not in self.__prbStats[team]:
self.__prbStats[team][prebattleID] = set()
self.__prbStats[team][prebattleID].add(vID)