Poodletooth-iLand/toontown/distributed/ToontownDistrictStats.py

90 lines
3.2 KiB
Python
Raw Normal View History

2015-03-03 22:10:12 +00:00
from direct.showbase import DirectObject
from direct.directnotify import DirectNotifyGlobal
from direct.distributed import DistributedObject
from direct.task import Task
from direct.distributed import DoInterestManager
from otp.distributed.OtpDoGlobals import *
_ToonTownDistrictStatInterest = None
_ToonTownDistrictStatInterestComplete = 0
_trashObject = DirectObject.DirectObject()
def EventName():
return 'ShardPopulationSet'
def isOpen():
global _ToonTownDistrictStatInterest
return _ToonTownDistrictStatInterest is not None
def isComplete():
global _ToonTownDistrictStatInterestComplete
return _ToonTownDistrictStatInterestComplete
def open(event = None):
global _trashObject
global _ToonTownDistrictStatInterest
if not isOpen():
def _CompleteProc(event):
global _ToonTownDistrictStatInterestComplete
_ToonTownDistrictStatInterestComplete = 1
if event is not None:
messenger.send(event)
return
_trashObject.acceptOnce(EventName(), _CompleteProc)
_ToonTownDistrictStatInterest = base.cr.addInterest(OTP_DO_ID_TOONTOWN, OTP_ZONE_ID_DISTRICTS_STATS, EventName(), EventName())
elif isComplete():
messenger.send(EventName())
def refresh(event = None):
global _ToonTownDistrictStatInterest
if isOpen():
if isComplete():
messenger.send(EventName())
if event is not none:
messenger.send(event)
else:
def _CompleteProc(event):
global _ToonTownDistrictStatInterestComplete
_ToonTownDistrictStatInterestComplete = 1
if event is not None:
messenger.send(event)
close()
return
_trashObject.acceptOnce(EventName(), _CompleteProc, [event])
_ToonTownDistrictStatInterest = base.cr.addInterest(OTP_DO_ID_TOONTOWN, OTP_ZONE_ID_DISTRICTS_STATS, EventName(), EventName())
def close():
global _ToonTownDistrictStatInterest
global _ToonTownDistrictStatInterestComplete
if isOpen():
_ToonTownDistrictStatInterestComplete = 0
base.cr.removeInterest(_ToonTownDistrictStatInterest, None)
_ToonTownDistrictStatInterest = None
class ToontownDistrictStats(DistributedObject.DistributedObject):
neverDisable = 1
def __init__(self, cr):
DistributedObject.DistributedObject.__init__(self, cr)
self.toontownDistrictId = 0
def settoontownDistrictId(self, value):
self.toontownDistrictId = value
def setAvatarCount(self, avatarCount):
if self.toontownDistrictId in self.cr.activeDistrictMap:
self.cr.activeDistrictMap[self.toontownDistrictId].avatarCount = avatarCount
def setNewAvatarCount(self, newAvatarCount):
if self.toontownDistrictId in self.cr.activeDistrictMap:
self.cr.activeDistrictMap[self.toontownDistrictId].newAvatarCount = newAvatarCount
def setInvasionStatus(self, invasionStatus):
if self.toontownDistrictId in self.cr.activeDistrictMap:
self.cr.activeDistrictMap[self.toontownDistrictId].invasionStatus = invasionStatus
def setStats(self, avatarCount, newAvatarCount):
self.setAvatarCount(avatarCount)
self.setNewAvatarCount(newAvatarCount)