28 lines
1.2 KiB
Python
28 lines
1.2 KiB
Python
|
from toontown.toonbase import ToontownGlobals
|
||
|
|
import DistributedNPCToonBaseAI, GloveNPCGlobals, ToonDNA
|
||
|
|
|
||
|
|
class DistributedNPCGloveAI(DistributedNPCToonBaseAI.DistributedNPCToonBaseAI):
|
||
|
|
|
||
|
|
def changeGlove(self, color):
|
||
|
|
avId = self.air.getAvatarIdFromSender()
|
||
|
|
av = self.air.doId2do.get(avId)
|
||
|
|
|
||
|
|
if av is None or not hasattr(av, 'dna'):
|
||
|
|
return
|
||
|
|
elif len(ToonDNA.allColorsList) <= color:
|
||
|
|
self.sendUpdate('changeGloveResult', [avId, GloveNPCGlobals.INVALID_COLOR])
|
||
|
|
return
|
||
|
|
elif av.dna.gloveColor == color:
|
||
|
|
self.sendUpdate('changeGloveResult', [avId, GloveNPCGlobals.SAME_COLOR])
|
||
|
|
return
|
||
|
|
elif av.getTotalMoney() < ToontownGlobals.GloveCost:
|
||
|
|
self.sendUpdate('changeGloveResult', [avId, GloveNPCGlobals.NOT_ENOUGH_MONEY])
|
||
|
|
return
|
||
|
|
|
||
|
|
av.takeMoney(ToontownGlobals.GloveCost)
|
||
|
|
newDNA = ToonDNA.ToonDNA()
|
||
|
|
newDNA.makeFromNetString(av.getDNAString())
|
||
|
|
newDNA.gloveColor = ToonDNA.allColorsList[color]
|
||
|
|
taskMgr.doMethodLater(1.0, lambda task: av.b_setDNAString(newDNA.makeNetString()), 'transform-%d' % avId)
|
||
|
|
self.sendUpdate('changeGloveResult', [avId, GloveNPCGlobals.CHANGE_SUCCESSFUL])
|