Poodletooth-iLand/toontown/distributed/ToontownDistrictStatsAI.py
2015-07-12 12:51:05 -04:00

64 lines
2.2 KiB
Python
Executable file

from direct.directnotify.DirectNotifyGlobal import directNotify
from direct.distributed.DistributedObjectAI import DistributedObjectAI
class ToontownDistrictStatsAI(DistributedObjectAI):
notify = directNotify.newCategory('ToontownDistrictStatsAI')
districtId = 0
avatarCount = 0
invasionStatus = 0
def announceGenerate(self):
DistributedObjectAI.announceGenerate(self)
# We want to handle shard status queries so that a ShardStatusReceiver
# being created after we're generated will know where we're at:
self.air.accept('shardStatus', self.handleShardStatusQuery)
def handleShardStatusQuery(self):
# Send a shard status update containing our population:
status = {'population': self.avatarCount}
self.air.sendNetEvent('shardStatus', [self.air.ourChannel, status])
def setDistrictId(self, districtId):
self.districtId = districtId
def d_setDistrictId(self, districtId):
self.sendUpdate('setDistrictId', [districtId])
def b_setDistrictId(self, districtId):
self.setDistrictId(districtId)
self.d_setDistrictId(districtId)
def getDistrictId(self):
return self.districtId
def setAvatarCount(self, avatarCount):
self.avatarCount = avatarCount
# Send a shard status update containing our population:
status = {'population': self.avatarCount}
self.air.sendNetEvent('shardStatus', [self.air.ourChannel, status])
def d_setAvatarCount(self, avatarCount):
self.sendUpdate('setAvatarCount', [avatarCount])
def b_setAvatarCount(self, avatarCount):
self.d_setAvatarCount(avatarCount)
self.setAvatarCount(avatarCount)
def getAvatarCount(self):
return self.avatarCount
def setInvasionStatus(self, invasionStatus):
self.invasionStatus = invasionStatus
def d_setInvasionStatus(self, invasionStatus):
self.sendUpdate('setInvasionStatus', [invasionStatus])
def b_setInvasionStatus(self, invasionStatus):
self.setInvasionStatus(invasionStatus)
self.d_setInvasionStatus(invasionStatus)
def getInvasionStatus(self):
return self.invasionStatus