from . import base import logging from xml.etree import cElementTree as ET class jobs(base.base_plugin): def plugin_init(self): self.xep = 'pubsubjob' self.description = "Job distribution over Pubsub" def post_init(self): pass #TODO add event def createJobNode(self, host, jid, node, config=None): pass def createJob(self, host, node, jobid=None, payload=None): return self.xmpp.plugin['xep_0060'].setItem(host, node, ((jobid, payload),)) def claimJob(self, host, node, jobid, ifrom=None): return self._setState(host, node, jobid, ET.Element('{http://andyet.net/protocol/pubsubjob}claimed')) def unclaimJob(self, jobid): return self._setState(host, node, jobid, ET.Element('{http://andyet.net/protocol/pubsubjob}unclaimed')) def finishJob(self, host, node, jobid, payload=None): finished = ET.Element('{http://andyet.net/protocol/pubsubjob}finished') if payload is not None: finished.append(payload) return self._setState(host, node, jobid, finished) def _setState(self, host, node, jobid, state, ifrom=None): iq = self.xmpp.Iq() iq['to'] = host if ifrom: iq['from'] = ifrom iq['type'] = 'set' iq['psstate']['node'] = node iq['psstate']['item'] = jobid iq['psstate']['payload'] = state result = iq.send() if result is None or result['type'] != 'result': return False return True