20 lines
708 B
Python
20 lines
708 B
Python
from otp.ai.AIBaseGlobal import *
|
|
from direct.directnotify import DirectNotifyGlobal
|
|
import random
|
|
from toontown.suit import SuitDNA
|
|
from . import CogDisguiseGlobals
|
|
|
|
class CogSuitManagerAI:
|
|
notify = DirectNotifyGlobal.directNotify.newCategory('CogSuitManagerAI')
|
|
|
|
def __init__(self, air):
|
|
self.air = air
|
|
|
|
def recoverPart(self, av, factoryType, suitTrack, zoneId, avList):
|
|
partsRecovered = [
|
|
0, 0, 0, 0]
|
|
part = av.giveGenericCogPart(factoryType, suitTrack)
|
|
if part:
|
|
partsRecovered[CogDisguiseGlobals.dept2deptIndex(suitTrack)] = part
|
|
self.air.questManager.toonRecoveredCogSuitPart(av, zoneId, avList)
|
|
return partsRecovered
|