"""DistributedLargeBlobSender module: contains the DistributedLargeBlobSender class"""
from direct.distributed import DistributedObject
from direct.directnotify import DirectNotifyGlobal
from direct.showbase.MessengerGlobal import messenger
from . import LargeBlobSenderConsts
[docs]
class DistributedLargeBlobSender(DistributedObject.DistributedObject):
"""DistributedLargeBlobSender: for sending large chunks of data through
the DC system"""
notify = DirectNotifyGlobal.directNotify.newCategory('DistributedLargeBlobSender')
[docs]
def __init__(self, cr):
DistributedObject.DistributedObject.__init__(self, cr)
[docs]
def generate(self):
DistributedLargeBlobSender.notify.debug('generate')
DistributedObject.DistributedObject.generate(self)
self.complete = 0
self.doneEvent = None
[docs]
def setMode(self, mode):
self.mode = mode
self.useDisk = mode & LargeBlobSenderConsts.USE_DISK
[docs]
def setTargetAvId(self, avId):
self.targetAvId = avId
[docs]
def announceGenerate(self):
DistributedLargeBlobSender.notify.debug('announceGenerate')
DistributedObject.DistributedObject.announceGenerate(self)
if self.targetAvId != base.localAvatar.doId:
# it's not for us
return
if not self.useDisk:
self.blob = ''
[docs]
def setChunk(self, chunk):
DistributedLargeBlobSender.notify.debug('setChunk')
assert not self.useDisk
if len(chunk) > 0:
self.blob += chunk
else:
self.privOnBlobComplete()
[docs]
def setFilename(self, filename):
DistributedLargeBlobSender.notify.debug('setFilename: %s' % filename)
assert self.useDisk
import os
origDir = os.getcwd()
bPath = LargeBlobSenderConsts.getLargeBlobPath()
try:
os.chdir(bPath)
except OSError:
DistributedLargeBlobSender.notify.error(
'could not access %s' % bPath)
f = open(filename, 'rb')
self.blob = f.read()
f.close()
os.unlink(filename)
os.chdir(origDir)
self.privOnBlobComplete()
[docs]
def isComplete(self):
""" returns non-zero if we've got the full blob """
return self.complete
[docs]
def setDoneEvent(self, event):
self.doneEvent = event
[docs]
def privOnBlobComplete(self):
assert not self.isComplete()
self.complete = 1
# announce that we've got it?
if self.doneEvent is not None:
messenger.send(self.doneEvent, [self.blob])
[docs]
def getBlob(self):
""" returns the full blob """
assert self.isComplete()
return self.blob
[docs]
def sendAck(self):
self.sendUpdate('setAck', [])
