def __init__(self):
self._myIPaddress = "127.0.0.1"
self._connAddr = None
self._bindAddr = None
self._activeTransfers = 0
self._uploadSpeed = 0.0
self._downloadSpeed = 0.0
self._files = {}
self.listening = Delegate()
self.connected = Delegate()
self.connectionError = Delegate()
self.disconnected = Delegate()
self.stateChanged = Delegate()
self.fileListUpdated = Delegate()
self.fileUpdated = Delegate()
self.transferFinished = Delegate()
self.session = FileSharingSession()
class SparkApplication(object):
""" Hold the state of the whole application. """
def __init__(self):
self._myIPaddress = "127.0.0.1"
self._connAddr = None
self._bindAddr = None
self._activeTransfers = 0
self._uploadSpeed = 0.0
self._downloadSpeed = 0.0
self._files = {}
self.listening = Delegate()
self.connected = Delegate()
self.connectionError = Delegate()
self.disconnected = Delegate()
self.stateChanged = Delegate()
self.fileListUpdated = Delegate()
self.fileUpdated = Delegate()
self.transferFinished = Delegate()
self.session = FileSharingSession()
def __enter__(self):
return self
def __exit__(self, type, val, tb):
try:
self.session.stop()
except Exception:
logging.exception("Error while stoping the session")
def start(self):
self.session.start()
def start_linked(self):
Process.trap_exit()
self.session.start_linked()
def connect(self, address):
Process.try_send(self.session.pid, Command("connect", address))
def bind(self, address):
Process.try_send(self.session.pid, Command("bind", address))
def disconnect(self):
Process.try_send(self.session.pid, Command("disconnect"))
def listFiles(self, excludeRemoved=True, senderPid=None):
""" Return a copy of the current file table, which maps file IDs to files. """
if not senderPid:
senderPid = Process.current()
Process.try_send(self.session.pid, Command("list-files", excludeRemoved, senderPid))
def addFile(self, path, mimeType=None, senderPid=None):
""" Add the local file with the given path to the list. """
if not senderPid:
senderPid = Process.current()
Process.try_send(self.session.pid, Command("add-file", path, mimeType, senderPid))
def removeFile(self, fileID, senderPid=None):
""" Remove the file (local or remote) with the given ID from the list. """
if not senderPid:
senderPid = Process.current()
Process.try_send(self.session.pid, Command("remove-file", fileID, senderPid))
def startTransfer(self, fileID, path, senderPid=None):
""" Start receiving the remote file with the given ID. """
if not senderPid:
senderPid = Process.current()
Process.try_send(self.session.pid, Command("start-transfer", fileID, path, senderPid))
def stopTransfer(self, fileID, senderPid=None):
""" Stop receiving the remote file with the given ID. """
if not senderPid:
senderPid = Process.current()
Process.try_send(self.session.pid, Command("stop-transfer", fileID, senderPid))
def updateTransferInfo(self):
Process.try_send(self.session.pid, Command("update-transfer-info"))
@property
def files(self):
""" Return the current file list. """
return self._files
@property
def myIPaddress(self):
""" Return the public IP address of the user, if known. """
return self._myIPaddress
@property
def isConnected(self):
""" Determine whether the session is active, i.e. we are connected to a remote peer. """
return self._connAddr is not None
@property
def isListening(self):
""" Determine whether the server is listening for incoming connections. """
return self._bindAddr is not None
@property
def connectionAddress(self):
""" Return the remote peer's address, if a session is active. """
return self._connAddr
@property
def bindAddress(self):
""" Return the address the server is bound to. """
return self._bindAddr
@property
def activeTransfers(self):
""" Return the number of active transfers. """
return self._activeTransfers
@property
def uploadSpeed(self):
""" Return the total upload speed, across all active transfers. """
return self._uploadSpeed
@property
def downloadSpeed(self):
""" Return the total download speed, across all active transfers. """
return self._downloadSpeed
def installHandlers(self, matcher, pid=None):
""" Suscribe to Spark's internal processes events so the state is kept up to date. """
if pid == None:
pid = Process.current()
matcher.addHandlers(self,
self.session.listening.suscribe(pid),
self.session.connected.suscribe(pid),
self.session.connectionError.suscribe(pid),
self.session.disconnected.suscribe(pid),
self.session.stateChanged.suscribe(pid),
self.session.fileListUpdated.suscribe(pid),
self.session.fileUpdated.suscribe(pid),
self.session.ended.suscribe(pid),
Event("list-files", None),
Event("exit", int, None))
def onListening(self, m, bindAddr, *args):
self._bindAddr = bindAddr
self.listening()
def onConnected(self, m, connAddr, *args):
self._connAddr = connAddr
self.connected()
def onConnectionError(self, m, error, *args):
self.connectionError(error)
def onDisconnected(self, m, *args):
self._connAddr = None
self.disconnected()
def onSessionStateChanged(self, m, sessionState, *args):
self._activeTransfers = sessionState["activeTransfers"]
self._uploadSpeed = sessionState["uploadSpeed"]
self._downloadSpeed = sessionState["downloadSpeed"]
self.stateChanged()
def onFileListUpdated(self, m, *args):
self.listFiles()
def onListFiles(self, m, files, *args):
self._files = files
self.fileListUpdated()
def onFileUpdated(self, m, file, *args):
self._files[file.ID] = file
self.fileUpdated(file.ID)
transfer = file.transfer
if transfer and transfer.state == "closed":
self.transferFinished(transfer.transferID, transfer.direction, file.ID)
def onEnded(self, m, *args):
self.sessionEnded()
def onExit(self, m, pid, reason, *args):
if pid == self.session.pid:
self.sessionEnded()
def sessionEnded(self):
self._bindAddr = None
if self._connAddr:
self._connAddr = None
self._activeTransfers = 0
self._uploadSpeed = 0.0
self._downloadSpeed = 0.0
self.disconnected()
self.stateChanged()
