Poodletooth-iLand/toontown/estate/PlantTreeGUI.py
2015-03-03 17:10:12 -05:00

43 lines
1.7 KiB
Python

from direct.directnotify import DirectNotifyGlobal
from direct.showbase.ShowBase import *
from toontown.toonbase import TTLocalizer
import string
from direct.fsm import StateData
from toontown.toonbase.ToontownBattleGlobals import gagIsPaidOnly
from toontown.toontowngui.TeaserPanel import TeaserPanel
class PlantTreeGUI(StateData.StateData):
notify = DirectNotifyGlobal.directNotify.newCategory('PlantTreeGUI')
def __init__(self, doneEvent):
self.doneEvent = doneEvent
self.oldActivateMode = base.localAvatar.inventory.activateMode
self._teaserPanel = None
base.localAvatar.inventory.setActivateMode('plantTree')
base.localAvatar.inventory.show()
self.accept('inventory-selection', self.__handleInventory)
self.accept('inventory-pass', self.__handleCancel)
return
def destroy(self):
self.ignore('inventory-selection')
self.ignore('inventory-pass')
base.localAvatar.inventory.setActivateMode(self.oldActivateMode)
base.localAvatar.inventory.hide()
if self._teaserPanel:
self._teaserPanel.destroy()
self._teaserPanel = None
return
def __handleInventory(self, track, level):
if gagIsPaidOnly(track, level) and not base.cr.isPaid():
self._teaserPanel = TeaserPanel('plantGags')
return
if base.localAvatar.inventory.numItem(track, level) > 0:
messenger.send(self.doneEvent, [True, track, level])
else:
self.notify.error("An item we don't have: track %s level %s was selected." % (track, level))
def __handleCancel(self):
messenger.send(self.doneEvent, [False, None, None])
return