50 lines
1.5 KiB
Python
50 lines
1.5 KiB
Python
from . import base
|
|
import logging
|
|
from xml.etree import cElementTree as ET
|
|
import types
|
|
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
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, host, node, 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 type(result) == types.BooleanType or result['type'] != 'result':
|
|
log.error("Unable to change %s:%s to %s" % (node, jobid, state))
|
|
return False
|
|
return True
|
|
|