Add initial XEP-0084 support.
It does not auto-retrieve and store avatars yet, but everything is there to do so.
This commit is contained in:
parent
378a42889f
commit
0b7f134021
1
setup.py
1
setup.py
@ -72,6 +72,7 @@ packages = [ 'sleekxmpp',
|
|||||||
'sleekxmpp/plugins/xep_0077',
|
'sleekxmpp/plugins/xep_0077',
|
||||||
'sleekxmpp/plugins/xep_0078',
|
'sleekxmpp/plugins/xep_0078',
|
||||||
'sleekxmpp/plugins/xep_0080',
|
'sleekxmpp/plugins/xep_0080',
|
||||||
|
'sleekxmpp/plugins/xep_0084',
|
||||||
'sleekxmpp/plugins/xep_0085',
|
'sleekxmpp/plugins/xep_0085',
|
||||||
'sleekxmpp/plugins/xep_0086',
|
'sleekxmpp/plugins/xep_0086',
|
||||||
'sleekxmpp/plugins/xep_0092',
|
'sleekxmpp/plugins/xep_0092',
|
||||||
|
@ -32,6 +32,7 @@ __all__ = [
|
|||||||
# 'xep_0078', # Non-SASL auth. Don't automatically load
|
# 'xep_0078', # Non-SASL auth. Don't automatically load
|
||||||
'xep_0080', # User Location
|
'xep_0080', # User Location
|
||||||
'xep_0082', # XMPP Date and Time Profiles
|
'xep_0082', # XMPP Date and Time Profiles
|
||||||
|
'xep_0084', # User Avatar
|
||||||
'xep_0085', # Chat State Notifications
|
'xep_0085', # Chat State Notifications
|
||||||
'xep_0086', # Legacy Error Codes
|
'xep_0086', # Legacy Error Codes
|
||||||
'xep_0092', # Software Version
|
'xep_0092', # Software Version
|
||||||
|
18
sleekxmpp/plugins/xep_0084/__init__.py
Normal file
18
sleekxmpp/plugins/xep_0084/__init__.py
Normal file
@ -0,0 +1,18 @@
|
|||||||
|
"""
|
||||||
|
SleekXMPP: The Sleek XMPP Library
|
||||||
|
Copyright (C) 2012 Nathanael C. Fritz, Lance J.T. Stout
|
||||||
|
This file is part of SleekXMPP.
|
||||||
|
|
||||||
|
See the file LICENSE for copying permission.
|
||||||
|
"""
|
||||||
|
|
||||||
|
from sleekxmpp.plugins.base import register_plugin
|
||||||
|
|
||||||
|
from sleekxmpp.plugins.xep_0084 import stanza
|
||||||
|
from sleekxmpp.plugins.xep_0084.stanza import Data, MetaData
|
||||||
|
from sleekxmpp.plugins.xep_0084.avatar import XEP_0084
|
||||||
|
|
||||||
|
|
||||||
|
register_plugin(XEP_0084)
|
||||||
|
|
||||||
|
|
101
sleekxmpp/plugins/xep_0084/avatar.py
Normal file
101
sleekxmpp/plugins/xep_0084/avatar.py
Normal file
@ -0,0 +1,101 @@
|
|||||||
|
"""
|
||||||
|
SleekXMPP: The Sleek XMPP Library
|
||||||
|
Copyright (C) 2012 Nathanael C. Fritz, Lance J.T. Stout
|
||||||
|
This file is part of SleekXMPP.
|
||||||
|
|
||||||
|
See the file LICENSE for copying permission.
|
||||||
|
"""
|
||||||
|
|
||||||
|
import hashlib
|
||||||
|
import logging
|
||||||
|
|
||||||
|
from sleekxmpp import Iq
|
||||||
|
from sleekxmpp.plugins import BasePlugin
|
||||||
|
from sleekxmpp.xmlstream.handler import Callback
|
||||||
|
from sleekxmpp.xmlstream.matcher import StanzaPath
|
||||||
|
from sleekxmpp.xmlstream import register_stanza_plugin, JID
|
||||||
|
from sleekxmpp.plugins.xep_0084 import stanza, Data, MetaData
|
||||||
|
|
||||||
|
|
||||||
|
log = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
|
class XEP_0084(BasePlugin):
|
||||||
|
|
||||||
|
name = 'xep_0084'
|
||||||
|
description = 'XEP-0084: User Avatar'
|
||||||
|
dependencies = set(['xep_0163', 'xep_0060'])
|
||||||
|
stanza = stanza
|
||||||
|
|
||||||
|
def plugin_init(self):
|
||||||
|
self.xmpp['xep_0163'].register_pep('avatar_metadata', MetaData)
|
||||||
|
|
||||||
|
pubsub_stanza = self.xmpp['xep_0060'].stanza
|
||||||
|
register_stanza_plugin(pubsub_stanza.Item, Data)
|
||||||
|
register_stanza_plugin(pubsub_stanza.EventItem, Data)
|
||||||
|
|
||||||
|
self.xmpp['xep_0060'].map_node_event(Data.namespace, 'avatar_data')
|
||||||
|
|
||||||
|
def retrieve_avatar(self, jid, id, url=None, ifrom=None, block=True,
|
||||||
|
callback=None, timeout=None):
|
||||||
|
return self.xmpp['xep_0060'].get_item(jid, Data.namespace, id,
|
||||||
|
ifrom=ifrom,
|
||||||
|
block=block,
|
||||||
|
callback=callback,
|
||||||
|
timeout=timeout)
|
||||||
|
|
||||||
|
def publish_avatar(self, data, ifrom=None, block=True, callback=None,
|
||||||
|
timeout=None):
|
||||||
|
payload = Data()
|
||||||
|
payload['value'] = data
|
||||||
|
return self.xmpp['xep_0163'].publish(payload,
|
||||||
|
node=Data.namespace,
|
||||||
|
id=hashlib.sha1(data).hexdigest(),
|
||||||
|
ifrom=ifrom,
|
||||||
|
block=block,
|
||||||
|
callback=callback,
|
||||||
|
timeout=timeout)
|
||||||
|
|
||||||
|
def publish_avatar_metadata(self, items=None, pointers=None,
|
||||||
|
ifrom=None, block=True,
|
||||||
|
callback=None, timeout=None):
|
||||||
|
metadata = MetaData()
|
||||||
|
if items is None:
|
||||||
|
items = []
|
||||||
|
for info in items:
|
||||||
|
metadata.add_info(info['id'], info['type'], info['bytes'],
|
||||||
|
height=info.get('height', ''),
|
||||||
|
width=info.get('width', ''),
|
||||||
|
url=info.get('url', ''))
|
||||||
|
for pointer in pointers:
|
||||||
|
metadata.add_pointer(pointer)
|
||||||
|
|
||||||
|
return self.xmpp['xep_0163'].publish(payload,
|
||||||
|
node=Data.namespace,
|
||||||
|
id=hashlib.sha1(data).hexdigest(),
|
||||||
|
ifrom=ifrom,
|
||||||
|
block=block,
|
||||||
|
callback=callback,
|
||||||
|
timeout=timeout)
|
||||||
|
|
||||||
|
def stop(self, ifrom=None, block=True, callback=None, timeout=None):
|
||||||
|
"""
|
||||||
|
Clear existing avatar metadata information to stop notifications.
|
||||||
|
|
||||||
|
Arguments:
|
||||||
|
ifrom -- Specify the sender's JID.
|
||||||
|
block -- Specify if the send call will block until a response
|
||||||
|
is received, or a timeout occurs. Defaults to True.
|
||||||
|
timeout -- The length of time (in seconds) to wait for a response
|
||||||
|
before exiting the send call if blocking is used.
|
||||||
|
Defaults to sleekxmpp.xmlstream.RESPONSE_TIMEOUT
|
||||||
|
callback -- Optional reference to a stream handler function. Will
|
||||||
|
be executed when a reply stanza is received.
|
||||||
|
"""
|
||||||
|
metadata = MetaData()
|
||||||
|
return self.xmpp['xep_0163'].publish(metadata,
|
||||||
|
node=MetaData.namespace,
|
||||||
|
ifrom=ifrom,
|
||||||
|
block=block,
|
||||||
|
callback=callback,
|
||||||
|
timeout=timeout)
|
78
sleekxmpp/plugins/xep_0084/stanza.py
Normal file
78
sleekxmpp/plugins/xep_0084/stanza.py
Normal file
@ -0,0 +1,78 @@
|
|||||||
|
"""
|
||||||
|
SleekXMPP: The Sleek XMPP Library
|
||||||
|
Copyright (C) 2012 Nathanael C. Fritz, Lance J.T. Stout
|
||||||
|
This file is part of SleekXMPP.
|
||||||
|
|
||||||
|
See the file LICENSE for copying permission.
|
||||||
|
"""
|
||||||
|
|
||||||
|
from base64 import b64encode, b64decode
|
||||||
|
from sleekxmpp.thirdparty.suelta.util import bytes
|
||||||
|
|
||||||
|
from sleekxmpp.xmlstream import ET, ElementBase, register_stanza_plugin
|
||||||
|
|
||||||
|
|
||||||
|
class Data(ElementBase):
|
||||||
|
name = 'data'
|
||||||
|
namespace = 'urn:xmpp:avatar:data'
|
||||||
|
plugin_attrib = 'avatar_data'
|
||||||
|
interfaces = set(['value'])
|
||||||
|
|
||||||
|
def get_value(self):
|
||||||
|
if self.xml.text:
|
||||||
|
return b64decode(bytes(self.xml.text))
|
||||||
|
return ''
|
||||||
|
|
||||||
|
def set_value(self, value):
|
||||||
|
if value:
|
||||||
|
self.xml.text = b64encode(bytes(value))
|
||||||
|
else:
|
||||||
|
self.xml.text = ''
|
||||||
|
|
||||||
|
def del_value(self):
|
||||||
|
self.xml.text = ''
|
||||||
|
|
||||||
|
|
||||||
|
class MetaData(ElementBase):
|
||||||
|
name = 'metadata'
|
||||||
|
namespace = 'urn:xmpp:avatar:metadata'
|
||||||
|
plugin_attrib = 'avatar_metadata'
|
||||||
|
interfaces = set()
|
||||||
|
|
||||||
|
def add_info(self, id, itype, ibytes, height=None, width=None, url=None):
|
||||||
|
info = Info()
|
||||||
|
info.values = {'id': id,
|
||||||
|
'type': itype,
|
||||||
|
'bytes': ibytes,
|
||||||
|
'height': height,
|
||||||
|
'width': width,
|
||||||
|
'url': url}
|
||||||
|
self.append(info)
|
||||||
|
|
||||||
|
def add_pointer(self, xml):
|
||||||
|
if not isinstance(xml, Pointer):
|
||||||
|
pointer = Pointer()
|
||||||
|
pointer.append(xml)
|
||||||
|
self.append(pointer)
|
||||||
|
else:
|
||||||
|
self.append(xml)
|
||||||
|
|
||||||
|
|
||||||
|
class Info(ElementBase):
|
||||||
|
name = 'info'
|
||||||
|
namespace = 'urn:xmpp:avatar:metadata'
|
||||||
|
plugin_attrib = 'info'
|
||||||
|
plugin_multi_attrib = 'items'
|
||||||
|
interfaces = set(['bytes', 'height', 'id', 'type', 'url', 'width'])
|
||||||
|
|
||||||
|
|
||||||
|
class Pointer(ElementBase):
|
||||||
|
name = 'pointer'
|
||||||
|
namespace = 'urn:xmpp:avatar:metadata'
|
||||||
|
plugin_attrib = 'pointer'
|
||||||
|
multi_plugin_attrib = 'pointers'
|
||||||
|
interfaces = set()
|
||||||
|
|
||||||
|
|
||||||
|
register_stanza_plugin(MetaData, Info, iterable=True)
|
||||||
|
register_stanza_plugin(MetaData, Pointer, iterable=True)
|
@ -109,6 +109,7 @@ class XEP_0163(BasePlugin):
|
|||||||
node = stanza.namespace
|
node = stanza.namespace
|
||||||
|
|
||||||
return self.xmpp['xep_0060'].publish(ifrom, node,
|
return self.xmpp['xep_0060'].publish(ifrom, node,
|
||||||
|
id=id,
|
||||||
payload=stanza.xml,
|
payload=stanza.xml,
|
||||||
options=options,
|
options=options,
|
||||||
ifrom=ifrom,
|
ifrom=ifrom,
|
||||||
|
Loading…
x
Reference in New Issue
Block a user