44 lines
1.7 KiB
Python
44 lines
1.7 KiB
Python
|
import time
|
||
|
|
|
||
|
|
from direct.directnotify import DirectNotifyGlobal
|
||
|
|
from direct.distributed.DistributedObjectAI import DistributedObjectAI
|
||
|
|
|
||
|
|
from toontown.catalog.CatalogGenerator import CatalogGenerator
|
||
|
|
from toontown.toonbase import ToontownGlobals
|
||
|
|
|
||
|
|
|
||
|
|
class CatalogManagerAI(DistributedObjectAI):
|
||
|
|
notify = DirectNotifyGlobal.directNotify.newCategory('CatalogManagerAI')
|
||
|
|
|
||
|
|
def __init__(self, air):
|
||
|
|
DistributedObjectAI.__init__(self, air)
|
||
|
|
self.catalogGenerator = CatalogGenerator()
|
||
|
|
|
||
|
|
def startCatalog(self):
|
||
|
|
avId = self.air.getAvatarIdFromSender()
|
||
|
|
av = self.air.doId2do.get(avId)
|
||
|
|
if not av:
|
||
|
|
return
|
||
|
|
|
||
|
|
self.deliverCatalogFor(av)
|
||
|
|
|
||
|
|
def deliverCatalogFor(self, av):
|
||
|
|
previousWeek, previousTime = av.getCatalogSchedule()
|
||
|
|
mailboxContents = av.mailboxNotify
|
||
|
|
currentTime = time.time()
|
||
|
|
if previousTime:
|
||
|
|
elapsedWeeks = int((time.time() - previousTime * 60) // 604800)
|
||
|
|
if elapsedWeeks < 1:
|
||
|
|
currentTime = previousTime * 60
|
||
|
|
else:
|
||
|
|
currentTime = previousTime * 60 + elapsedWeeks * 604800
|
||
|
|
|
||
|
|
nextWeek = currentTime + 604800
|
||
|
|
currentWeek = previousWeek + 1
|
||
|
|
monthlyCatalog = self.catalogGenerator.generateMonthlyCatalog(av, currentTime / 60)
|
||
|
|
weeklyCatalog = self.catalogGenerator.generateWeeklyCatalog(av, currentWeek, monthlyCatalog)
|
||
|
|
backlogCatalog = self.catalogGenerator.generateBackCatalog(av, currentWeek, previousWeek, weeklyCatalog)
|
||
|
|
av.b_setCatalogSchedule(currentWeek, nextWeek / 60)
|
||
|
|
av.b_setCatalog(monthlyCatalog, weeklyCatalog, backlogCatalog)
|
||
|
|
av.b_setCatalogNotify(ToontownGlobals.NewItems, mailboxContents)
|