|
2231
|
1 #!/usr/bin/env python2 |
|
|
2 # -*- coding: utf-8 -*- |
|
|
3 |
|
|
4 # SAT plugin to detect language (experimental) |
|
|
5 # Copyright (C) 2009-2016 Jérôme Poisson (goffi@goffi.org) |
|
|
6 |
|
|
7 # This program is free software: you can redistribute it and/or modify |
|
|
8 # it under the terms of the GNU Affero General Public License as published by |
|
|
9 # the Free Software Foundation, either version 3 of the License, or |
|
|
10 # (at your option) any later version. |
|
|
11 |
|
|
12 # This program is distributed in the hope that it will be useful, |
|
|
13 # but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
|
14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|
|
15 # GNU Affero General Public License for more details. |
|
|
16 |
|
|
17 # You should have received a copy of the GNU Affero General Public License |
|
|
18 # along with this program. If not, see <http://www.gnu.org/licenses/>. |
|
|
19 |
|
|
20 from sat.core.i18n import _ |
|
|
21 from sat.core.constants import Const as C |
|
|
22 from sat.core.log import getLogger |
|
|
23 log = getLogger(__name__) |
|
|
24 from twisted.internet import defer |
|
|
25 from twisted.words.protocols.jabber import jid |
|
|
26 from twisted.words.xish import domish |
|
|
27 from wokkel import pubsub |
|
|
28 |
|
|
29 |
|
|
30 PLUGIN_INFO = { |
|
|
31 C.PI_NAME: "Event plugin", |
|
|
32 C.PI_IMPORT_NAME: "events", |
|
|
33 C.PI_TYPE: "EXP", |
|
|
34 C.PI_PROTOCOLS: [], |
|
|
35 C.PI_DEPENDENCIES: ["XEP-0060"], |
|
|
36 C.PI_MAIN: "Events", |
|
|
37 C.PI_HANDLER: "no", |
|
|
38 C.PI_DESCRIPTION: _("""Experimental implementation of XMPP events management""") |
|
|
39 } |
|
|
40 |
|
|
41 NS_EVENT = 'org.salut-a-toi.event:0' |
|
|
42 |
|
|
43 |
|
|
44 class Events(object): |
|
|
45 """Q&D module to handle event attendance answer, experimentation only""" |
|
|
46 |
|
|
47 def __init__(self, host): |
|
|
48 log.info(_(u"Event plugin initialization")) |
|
|
49 self.host = host |
|
|
50 self._p = self.host.plugins["XEP-0060"] |
|
|
51 host.bridge.addMethod("eventGet", ".plugin", |
|
|
52 in_sign='sss', out_sign='a{ss}', |
|
|
53 method=self._eventGet, |
|
|
54 async=True) |
|
|
55 host.bridge.addMethod("eventSet", ".plugin", |
|
|
56 in_sign='ssa{ss}s', out_sign='', |
|
|
57 method=self._eventSet, |
|
|
58 async=True) |
|
|
59 |
|
|
60 def _eventGet(self, service, node, profile_key): |
|
|
61 service = jid.JID(service) if service else None |
|
|
62 node = node if node else NS_EVENT |
|
|
63 client = self.host.getClient(profile_key) |
|
|
64 return self.eventGet(client, service, node) |
|
|
65 |
|
|
66 @defer.inlineCallbacks |
|
|
67 def eventGet(self, client, service, node): |
|
|
68 """Retrieve attendance from event node |
|
|
69 |
|
|
70 @param service(unicode, None): PubSub service |
|
|
71 @param node(unicode): PubSub node of the event |
|
|
72 @return (dict): a dict with current attendance status, |
|
|
73 an empty dict is returned if nothing has been answered yed |
|
|
74 """ |
|
|
75 items, metadata = yield self._p.getItems(service, node, item_ids=[client.jid.userhost()], profile_key=client.profile) |
|
|
76 try: |
|
|
77 event_elt = next(items[0].elements((NS_EVENT, u'event'))) |
|
|
78 except IndexError: |
|
|
79 # no item found, event data are not set yet |
|
|
80 defer.returnValue({}) |
|
|
81 data = {} |
|
|
82 for key in (u'attend', u'guests'): |
|
|
83 try: |
|
|
84 data[key] = event_elt[key] |
|
|
85 except KeyError: |
|
|
86 continue |
|
|
87 defer.returnValue(data) |
|
|
88 |
|
|
89 def _eventSet(self, service, node, event_data, profile_key): |
|
|
90 service = jid.JID(service) if service else None |
|
|
91 node = node if node else NS_EVENT |
|
|
92 client = self.host.getClient(profile_key) |
|
|
93 return self.eventSet(client, service, node, event_data) |
|
|
94 |
|
|
95 def eventSet(self, client, service, node, data): |
|
|
96 """Set or update attendance data in event node |
|
|
97 |
|
|
98 @param service(unicode, None): PubSub service |
|
|
99 @param node(unicode): PubSub node of the event |
|
|
100 @param data(dict[unicode, unicode]): data to update |
|
|
101 key can be: |
|
|
102 attend: one of "yes", "no", "maybe" |
|
|
103 guests: an int |
|
|
104 """ |
|
|
105 event_elt = domish.Element((NS_EVENT, 'event')) |
|
|
106 for key in (u'attend', u'guests'): |
|
|
107 try: |
|
|
108 event_elt[key] = data.pop(key) |
|
|
109 except KeyError: |
|
|
110 pass |
|
|
111 item_elt = pubsub.Item(id=client.jid.userhost(), payload=event_elt) |
|
|
112 return self._p.publish(service, node, items=[item_elt], profile_key=client.profile) |
