mirror of
https://github.com/Sneed-Group/Poodletooth-iLand
synced 2024-12-24 12:12:36 -06:00
84 lines
3 KiB
Python
Executable file
84 lines
3 KiB
Python
Executable file
from direct.directnotify import DirectNotifyGlobal
|
|
from toontown.estate.DistributedLawnDecorAI import DistributedLawnDecorAI
|
|
import GardenGlobals
|
|
import datetime
|
|
|
|
class DistributedPlantBaseAI(DistributedLawnDecorAI):
|
|
notify = DirectNotifyGlobal.directNotify.newCategory("DistributedPlantBaseAI")
|
|
|
|
def __init__(self, air):
|
|
DistributedLawnDecorAI.__init__(self, air)
|
|
self.air = air
|
|
self.growthLevel = -1
|
|
|
|
def announceGenerate(self):
|
|
DistributedLawnDecorAI.announceGenerate(self)
|
|
|
|
def delete(self):
|
|
DistributedLawnDecorAI.delete(self)
|
|
|
|
def disable(self):
|
|
DistributedLawnDecorAI.disable(self)
|
|
|
|
def setOwnerPlot(self, owner):
|
|
self.ownerPlot = owner
|
|
|
|
def getOwnerPlot(self):
|
|
return self.ownerPlot
|
|
|
|
def setTypeIndex(self, typeIndex):
|
|
self.typeIndex = typeIndex
|
|
self.attributes = GardenGlobals.PlantAttributes[typeIndex]
|
|
self.name = self.attributes['name']
|
|
self.plantType = self.attributes['plantType']
|
|
self.growthThresholds = self.attributes['growthThresholds']
|
|
self.maxWaterLevel = self.attributes['maxWaterLevel']
|
|
self.minWaterLevel = self.attributes['minWaterLevel']
|
|
|
|
def getTypeIndex(self):
|
|
return self.typeIndex
|
|
|
|
def setWaterLevel(self, water):
|
|
self.waterLevel = water
|
|
|
|
def getWaterLevel(self):
|
|
return self.waterLevel
|
|
|
|
def setGrowthLevel(self, growth):
|
|
self.growthLevel = growth
|
|
|
|
def getGrowthLevel(self):
|
|
return self.growthLevel
|
|
|
|
def waterPlant(self, avId):
|
|
# TODO: Proper water threshold for watering can type.
|
|
# Not supposed to be 16 for all watering cans.
|
|
self.lastWateredBy = avId
|
|
newLevel = self.waterLevel + 1
|
|
if newLevel > self.maxWaterLevel:
|
|
self.setWaterLevel(self.maxWaterLevel)
|
|
else:
|
|
self.setWaterLevel(newLevel)
|
|
self.sendUpdate('setMovie', [GardenGlobals.MOVIE_WATER, avId])
|
|
self.sendUpdate('setWaterLevel', [self.getWaterLevel()])
|
|
|
|
def waterPlantDone(self):
|
|
if hasattr(self, 'lastWateredBy'):
|
|
av = simbase.air.doId2do.get(self.lastWateredBy)
|
|
skill = av.getWateringCanSkill()
|
|
skill += GardenGlobals.WateringCanAttributes[av.wateringCan]['skillPts'] / 100
|
|
av.b_setWateringCanSkill(skill)
|
|
del self.lastWateredBy
|
|
estate = simbase.air.doId2do.get(self.getEstate())
|
|
dataIndex = -1
|
|
# TODO: Possibly store this in mongodb/cPickle instead.
|
|
for n, item in enumerate(estate.items[self.getOwnerIndex()]):
|
|
if item[0] == self.getPlot():
|
|
dataIndex = n
|
|
if dataIndex >= 0:
|
|
dtime = int(datetime.datetime.now().strftime('%Y%m%d%H%M'))
|
|
data = list(estate.items[self.getOwnerIndex()][dataIndex])
|
|
data[4] = self.getWaterLevel()
|
|
data[8] = dtime
|
|
estate.items[self.getOwnerIndex()][dataIndex] = tuple(data)
|
|
estate.updateItems()
|