summaryrefslogtreecommitdiff
path: root/sleekxmpp/plugins/jobs.py
blob: 30a76989888f154011ee4b8473fbfb528661bcab (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
from . import base
import logging
from xml.etree import cElementTree as ET
import types

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':
            logging.error("Unable to change %s:%s to %s" % (node, jobid, state))
			return False
		return True