Poodletooth-iLand/toontown/coghq/FactoryManagerAI.py

28 lines
1 KiB
Python
Raw Normal View History

2015-11-14 13:28:53 -06:00
from direct.directnotify import DirectNotifyGlobal
import DistributedMegaCorpAI
import DistributedFactoryAI
from toontown.toonbase import ToontownGlobals
from direct.showbase import DirectObject
class FactoryManagerAI(DirectObject.DirectObject):
notify = DirectNotifyGlobal.directNotify.newCategory('FactoryManagerAI')
factoryId = None
def __init__(self, air):
DirectObject.DirectObject.__init__(self)
self.air = air
def getDoId(self):
return 0
def createFactory(self, factoryId, entranceId, players):
factoryZone = self.air.allocateZone()
if FactoryManagerAI.factoryId is not None:
factoryId = FactoryManagerAI.factoryId
if entranceId == 2:
factory = DistributedMegaCorpAI.DistributedMegaCorpAI(self.air, factoryId, factoryZone, entranceId, players)
else:
factory = DistributedFactoryAI.DistributedFactoryAI(self.air, factoryId, factoryZone, entranceId, players)
factory.generateWithRequired(factoryZone)
return factoryZone