48 lines
1.8 KiB
Python
48 lines
1.8 KiB
Python
from direct.directnotify.DirectNotifyGlobal import directNotify
|
|
from otp.distributed.DistributedDistrictAI import DistributedDistrictAI
|
|
import time
|
|
|
|
class ToontownDistrictAI(DistributedDistrictAI):
|
|
notify = directNotify.newCategory('ToontownDistrictAI')
|
|
created = 0
|
|
|
|
def announceGenerate(self):
|
|
DistributedDistrictAI.announceGenerate(self)
|
|
|
|
# Remember the time of which this district was created:
|
|
self.created = int(time.time())
|
|
|
|
# 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('queryShardStatus', self.handleShardStatusQuery)
|
|
|
|
# Send a shard status update with the information we have:
|
|
status = {
|
|
'available': bool(self.available),
|
|
'name': self.name,
|
|
'created': int(time.time())
|
|
}
|
|
self.air.sendNetEvent('shardStatus', [self.air.ourChannel, status])
|
|
|
|
def handleShardStatusQuery(self):
|
|
# Send a shard status update with the information we have:
|
|
status = {
|
|
'available': bool(self.available),
|
|
'name': self.name,
|
|
'created': int(time.time())
|
|
}
|
|
self.air.sendNetEvent('shardStatus', [self.air.ourChannel, status])
|
|
|
|
def setName(self, name):
|
|
DistributedDistrictAI.setName(self, name)
|
|
|
|
# Send a shard status update containing our name:
|
|
status = {'name': name}
|
|
self.air.sendNetEvent('shardStatus', [self.air.ourChannel, status])
|
|
|
|
def setAvailable(self, available):
|
|
DistributedDistrictAI.setAvailable(self, available)
|
|
|
|
# Send a shard status update containing our availability:
|
|
status = {'available': bool(available)}
|
|
self.air.sendNetEvent('shardStatus', [self.air.ourChannel, status])
|