Poodletooth-iLand/toontown/parties/DistributedPartyFireworksActivityAI.py
2015-03-03 17:10:12 -05:00

38 lines
No EOL
1.5 KiB
Python

from direct.directnotify import DirectNotifyGlobal
from toontown.parties.DistributedPartyActivityAI import DistributedPartyActivityAI
from direct.distributed.ClockDelta import globalClockDelta
from direct.fsm.FSM import FSM
from toontown.effects import FireworkShows
import PartyGlobals
import random
class DistributedPartyFireworksActivityAI(DistributedPartyActivityAI, FSM):
notify = DirectNotifyGlobal.directNotify.newCategory("DistributedPartyFireworksActivityAI")
def __init__(self, air, parent, activityTuple):
DistributedPartyActivityAI.__init__(self, air, parent, activityTuple)
FSM.__init__(self, 'DistributedPartyActivityAI')
self.state = 'Idle'
def getEventId(self):
return PartyGlobals.FireworkShows.Summer
def getShowStyle(self):
return random.randint(0, len(FireworkShows.shows[PartyGlobals.FireworkShows.Summer]) - 1)
def getSongId(self):
return random.randint(0, 1)
def toonJoinRequest(self):
avId = self.air.getAvatarIdFromSender()
host = self.air.doId2do[self.parent].hostId
if avId == host and self.state == 'Idle':
self.request('Active')
return
self.sendUpdateToAvatarId(avId, 'joinRequestDenied', [1])
def enterActive(self):
self.sendUpdate('setState', ['Active', globalClockDelta.getRealNetworkTime()])
def enterIdle(self):
self.sendUpdate('setState', ['Idle', globalClockDelta.getRealNetworkTime()])