23 lines
1 KiB
Python
23 lines
1 KiB
Python
from RegenTreasurePlannerAI import RegenTreasurePlannerAI
|
|
from direct.directnotify import DirectNotifyGlobal
|
|
from toontown.toonbase import ToontownGlobals
|
|
|
|
class SZTreasurePlannerAI(RegenTreasurePlannerAI):
|
|
notify = DirectNotifyGlobal.directNotify.newCategory('SZTreasurePlannerAI')
|
|
|
|
def __init__(self, zoneId, treasureType, healAmount, spawnPoints, spawnRate, maxTreasures):
|
|
self.zoneId = zoneId
|
|
self.spawnPoints = spawnPoints
|
|
self.healAmount = healAmount
|
|
RegenTreasurePlannerAI.__init__(self, zoneId, treasureType, 'SZTreasurePlanner-%d' % zoneId, spawnRate, maxTreasures)
|
|
|
|
def initSpawnPoints(self):
|
|
pass
|
|
|
|
def validAvatar(self, treasure, av):
|
|
# Avatars can only heal if they are missing some health, but aren't sad.
|
|
if av.getHp() < av.getMaxHp() and av.getHp() > 0:
|
|
av.toonUp(self.healAmount * 2 if simbase.air.newsManager.isHolidayRunning(ToontownGlobals.VALENTOONS_DAY) else self.healAmount)
|
|
return True
|
|
else:
|
|
return False
|