71 lines
2.2 KiB
Python
71 lines
2.2 KiB
Python
|
"""
|
||
|
|
SleekXMPP: The Sleek XMPP Library
|
||
|
|
Copyright (C) 2013 Nathanael C. Fritz, Lance J.T. Stout
|
||
|
|
This file is part of SleekXMPP.
|
||
|
|
|
||
|
|
See the file LICENSE for copying permission.
|
||
|
|
"""
|
||
|
|
|
||
|
|
from datetime import datetime, timedelta
|
||
|
|
|
||
|
|
from sleekxmpp.stanza import Presence
|
||
|
|
from sleekxmpp.plugins import BasePlugin
|
||
|
|
from sleekxmpp.xmlstream import register_stanza_plugin
|
||
|
|
from sleekxmpp.xmlstream.handler import Callback
|
||
|
|
from sleekxmpp.xmlstream.matcher import StanzaPath
|
||
|
|
from sleekxmpp.plugins.xep_0319 import stanza
|
||
|
|
|
||
|
|
|
||
|
|
class XEP_0319(BasePlugin):
|
||
|
|
name = 'xep_0319'
|
||
|
|
description = 'XEP-0319: Last User Interaction in Presence'
|
||
|
|
dependencies = set()
|
||
|
|
stanza = stanza
|
||
|
|
|
||
|
|
def plugin_init(self):
|
||
|
|
self._idle_stamps = {}
|
||
|
|
register_stanza_plugin(Presence, stanza.Idle)
|
||
|
|
self.api.register(self._set_idle,
|
||
|
|
'set_idle',
|
||
|
|
default=True)
|
||
|
|
self.api.register(self._get_idle,
|
||
|
|
'get_idle',
|
||
|
|
default=True)
|
||
|
|
self.xmpp.register_handler(
|
||
|
|
Callback('Idle Presence',
|
||
|
|
StanzaPath('presence/idle'),
|
||
|
|
self._idle_presence))
|
||
|
|
self.xmpp.add_filter('out', self._stamp_idle_presence)
|
||
|
|
|
||
|
|
def session_bind(self, jid):
|
||
|
|
self.xmpp['xep_0030'].add_feature('urn:xmpp:idle:0')
|
||
|
|
|
||
|
|
def plugin_end(self):
|
||
|
|
self.xmpp['xep_0030'].del_feature(feature='urn:xmpp:idle:0')
|
||
|
|
self.xmpp.del_filter('out', self._stamp_idle_presence)
|
||
|
|
self.xmpp.remove_handler('Idle Presence')
|
||
|
|
|
||
|
|
def idle(self, jid=None, since=None):
|
||
|
|
if since is None:
|
||
|
|
since = datetime.now()
|
||
|
|
self.api['set_idle'](jid, None, None, since)
|
||
|
|
|
||
|
|
def active(self, jid=None):
|
||
|
|
self.api['set_idle'](jid, None, None, None)
|
||
|
|
|
||
|
|
def _set_idle(self, jid, node, ifrom, data):
|
||
|
|
self._idle_stamps[jid] = data
|
||
|
|
|
||
|
|
def _get_idle(self, jid, node, ifrom, data):
|
||
|
|
return self._idle_stamps.get(jid, None)
|
||
|
|
|
||
|
|
def _idle_presence(self, pres):
|
||
|
|
self.xmpp.event('presence_idle', pres)
|
||
|
|
|
||
|
|
def _stamp_idle_presence(self, stanza):
|
||
|
|
if isinstance(stanza, Presence):
|
||
|
|
since = self.api['get_idle'](stanza['from'] or self.xmpp.boundjid)
|
||
|
|
if since:
|
||
|
|
stanza['idle']['since'] = since
|
||
|
|
return stanza
|