libervia-backend: src/plugins/plugin_exp

comparison src/plugins/plugin_exp_pipe.py @ 538:2c4016921403

core, frontends, bridgen plugins: fixed methods which were unproperly managing multi-profiles - added profile argument to askConfirmation, actionResult, actionResultExt, entityDataUpdated, confirmationAnswer, getProgress - core, frontends: fixed calls/signals according to new bridge API - user of proper profile namespace for progression indicators and dialogs - memory: getParam* now return bool when param type is bool - memory: added getStringParam* to return string instead of typed value - core, memory, storage, quick_frontend: getHistory now manage properly multi-profiles - plugins XEP-0047, XEP-0054, XEP-0065, XEP-0077, XEP-0096; multi-profiles proper handling

author	Goffi <goffi@goffi.org>
date	Sat, 10 Nov 2012 16:38:16 +0100
parents	2a072735e459
children	ca13633d3b6b

comparison

equal deleted inserted replaced

-:28cddc96c4ed
+:2c4016921403
 along with this program.  If not, see <http://www.gnu.org/licenses/>.
 """
 from logging import debug, info, warning, error
 from twisted.words.xish import domish
-from twisted.internet import protocol
+from twisted.words.protocols.jabber import jid
-from twisted.words.protocols.jabber import client, jid
 from twisted.words.protocols.jabber import error as jab_error
 import os, os.path
 from twisted.internet import reactor
-import pdb
+from sat.core.exceptions import ProfileNotInCacheError
-from zope.interface import implements
+from wokkel import data_form
-from wokkel import disco, iwokkel, data_form
 IQ_SET = '/iq[@type="set"]'
 PROFILE_NAME = "pipe-transfer"
 PROFILE = "http://jabber.org/protocol/si/profile/" + PROFILE_NAME
 """This is a modified version of XEP-0096 to work with named pipes instead of files"""
 def __init__(self, host):
 info(_("Plugin Pipe initialization"))
 self.host = host
-self._waiting_for_approval = {} #key = id, value = [transfer data, IdelayedCall Reactor timeout,
-# current stream method, [failed stream methods], profile]
 self.managed_stream_m = [self.host.plugins["XEP-0065"].NAMESPACE,
 self.host.plugins["XEP-0047"].NAMESPACE] #Stream methods managed
 self.host.plugins["XEP-0095"].registerSIProfile(PROFILE_NAME, self.transferRequest)
 host.bridge.addMethod("pipeOut", ".plugin", in_sign='ssa{ss}s', out_sign='s', method=self.pipeOut)
-def _kill_id(self, approval_id):
+def profileConnected(self, profile):
+client = self.host.getClient(profile)
+client._pipe_waiting_for_approval = {} #key = id, value = [transfer data, IdelayedCall Reactor timeout,
+# current stream method, [failed stream methods], profile]
+def _kill_id(self, approval_id, profile):
 """Delete a waiting_for_approval id, called after timeout
-@param approval_id: id of _waiting_for_approval"""
+@param approval_id: id of _pipe_waiting_for_approval"""
 info(_("SI Pipe Transfer: TimeOut reached for id %s") % approval_id);
 try:
-del self._waiting_for_approval[approval_id]
+client = self.host.getClient(profile)
+del client._pipe_waiting_for_approval[approval_id]
 except KeyError:
 warning(_("kill id called on a non existant approval id"))
 def transferRequest(self, iq_id, from_jid, si_id, si_mime_type, si_el, profile):
 """Called when a pipe transfer is requested
 @param si_mime_type: Mime type of the pipe (or default "application/octet-stream" if unknown)
 @param si_el: domish.Element of the request
 @param profile: %(doc_profile)s"""
 info (_("EXP-PIPE file transfer requested"))
 debug(si_el.toXml())
+client = self.host.getClient(profile)
+if not client:
+raise ProfileNotInCacheError
 pipe_elts = filter(lambda elt: elt.name == 'pipe', si_el.elements())
 feature_elts = self.host.plugins["XEP-0020"].getFeatureElt(si_el)
 if not pipe_elts:
 warning(_("No pipe element found"))
 self.host.plugins["XEP-0095"].sendBadRequestError(iq_id, from_jid, profile)
 return
 #if we are here, the transfer can start, we just need user's agreement
 data={ "id": iq_id, "from":from_jid }
-self._waiting_for_approval[si_id] = [data, reactor.callLater(300, self._kill_id, si_id), stream_method, [], profile]
+client._pipe_waiting_for_approval[si_id] = [data, reactor.callLater(300, self._kill_id, si_id), stream_method, [], profile]
-self.host.askConfirmation(si_id, "PIPE_TRANSFER", data, self.confirmationCB)
+self.host.askConfirmation(si_id, "PIPE_TRANSFER", data, self.confirmationCB, profile)
-def confirmationCB(self, sid, accepted, frontend_data):
+def confirmationCB(self, sid, accepted, frontend_data, profile):
 """Called on confirmation answer
 @param sid: file transfer session id
 @param accepted: True if file transfer is accepted
 @param frontend_data: data sent by frontend"""
-data, timeout, stream_method, failed_methods, profile = self._waiting_for_approval[sid]
+client = self.host.getClient(profile)
+if not client:
+raise ProfileNotInCacheError
+data, timeout, stream_method, failed_methods, profile = client._pipe_waiting_for_approval[sid]
 if accepted:
 if timeout.active():
 timeout.cancel()
 try:
 dest_path = frontend_data['dest_path']
 except KeyError:
 error(_('dest path not found in frontend_data'))
-del(self._waiting_for_approval[sid])
+del(client._pipe_waiting_for_approval[sid])
 return
 if stream_method == self.host.plugins["XEP-0065"].NAMESPACE:
 file_obj = open(dest_path, 'w+')
-self.host.plugins["XEP-0065"].prepareToReceive(jid.JID(data['from']), sid, file_obj, None, self._transferSucceeded, self._transferFailed)
+self.host.plugins["XEP-0065"].prepareToReceive(jid.JID(data['from']), sid, file_obj, None, self._transferSucceeded, self._transferFailed, profile)
 elif stream_method == self.host.plugins["XEP-0047"].NAMESPACE:
 file_obj = open(dest_path, 'w+')
-self.host.plugins["XEP-0047"].prepareToReceive(jid.JID(data['from']), sid, file_obj, None, self._transferSucceeded, self._transferFailed)
+self.host.plugins["XEP-0047"].prepareToReceive(jid.JID(data['from']), sid, file_obj, None, self._transferSucceeded, self._transferFailed, profile)
 else:
 error(_("Unknown stream method, this should not happen at this stage, cancelling transfer"))
-del(self._waiting_for_approval[sid])
+del(client._pipe_waiting_for_approval[sid])
 return
 #we can send the iq result
 feature_elt = self.host.plugins["XEP-0020"].chooseOption({'stream-method':stream_method})
 misc_elts = []
 misc_elts.append(domish.Element((PROFILE, "file")))
 self.host.plugins["XEP-0095"].acceptStream(data["id"], data['from'], feature_elt, misc_elts, profile)
 else:
 debug (_("Transfer [%s] refused"), sid)
 self.host.plugins["XEP-0095"].sendRejectedError (data["id"], data['from'], profile=profile)
-del(self._waiting_for_approval[sid])
+del(client._pipe_waiting_for_approval[sid])
-def _transferSucceeded(self, sid, file_obj, stream_method):
+def _transferSucceeded(self, sid, file_obj, stream_method, profile):
 """Called by the stream method when transfer successfuly finished
 @param id: stream id"""
+client = self.host.getClient(profile)
+if not client:
+raise ProfileNotInCacheError
 file_obj.close()
 info(_('Transfer %s successfuly finished') % sid)
-del(self._waiting_for_approval[sid])
+del(client._pipe_waiting_for_approval[sid])
-def _transferFailed(self, sid, file_obj, stream_method, reason):
+def _transferFailed(self, sid, file_obj, stream_method, reason, profile):
 """Called when something went wrong with the transfer
 @param id: stream id
 @param reason: can be TIMEOUT, IO_ERROR, PROTOCOL_ERROR"""
-data, timeout, stream_method, failed_methods, profile = self._waiting_for_approval[sid]
+client = self.host.getClient(profile)
+if not client:
+raise ProfileNotInCacheError
+data, timeout, stream_method, failed_methods, profile = client._pipe_waiting_for_approval[sid]
 warning(_('Transfer %(id)s failed with stream method %(s_method)s') % { 'id': sid,
 's_method': stream_method })
 filepath = file_obj.name
 file_obj.close()
 #TODO: session remenber (within a time limit) when a stream method fail, and avoid that stream method with full jid for the rest of the session
 warning(_("All stream methods failed, can't transfer the file"))
-del(self._waiting_for_approval[sid])
+del(client._pipe_waiting_for_approval[sid])
-def pipeCb(self, profile, filepath, sid, IQ):
+def pipeCb(self, filepath, sid, profile, IQ):
 if IQ['type'] == "error":
 stanza_err = jab_error.exceptionFromStanza(IQ)
 if stanza_err.code == '403' and stanza_err.condition == 'forbidden':
 debug(_("Pipe transfer refused by %s") % IQ['from'])
 self.host.bridge.newAlert(_("The contact %s refused your pipe stream") % IQ['from'], _("Pipe stream refused"), "INFO", profile)
 pipe_elt = domish.Element((PROFILE, 'pipe'))
 pipe_transfer_elts.append(pipe_elt)
 sid, offer = self.host.plugins["XEP-0095"].proposeStream(jid.JID(to_jid), PROFILE, feature_elt, pipe_transfer_elts, profile_key = profile)
-offer.addCallback(self.pipeCb, profile, filepath, sid)
+offer.addCallback(self.pipeCb, filepath, sid, profile)
 return sid
-def sendSuccessCb(self, sid, file_obj, stream_method):
+def sendSuccessCb(self, sid, file_obj, stream_method, profile):
 info(_('Transfer %s successfuly finished') % sid)
 file_obj.close()
-def sendFailureCb(self, sid, file_obj, stream_method, reason):
+def sendFailureCb(self, sid, file_obj, stream_method, reason, profile):
 file_obj.close()
-warning(_('Transfer %(id)s failed with stream method %(s_method)s') % { 'id': sid, "s_method": stream_method })
+warning(_('Transfer %(id)s failed with stream method %(s_method)s %(profile)s') % { 'id': sid, "s_method": stream_method, "profile": profile })

Mercurial > libervia-backend

comparison src/plugins/plugin_exp_pipe.py @ 538:2c4016921403