917faecdcb
Resolved by always normalizing JIDs to bare form, regardless of if they are JID objects or strings. Also simplified related code to prefer use of JID objects instead of strings so they don't need to be parsed multiple times.
148 lines
4.6 KiB
Python
148 lines
4.6 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
sleekxmpp.xmlstream.jid
|
|
~~~~~~~~~~~~~~~~~~~~~~~
|
|
|
|
This module allows for working with Jabber IDs (JIDs) by
|
|
providing accessors for the various components of a JID.
|
|
|
|
Part of SleekXMPP: The Sleek XMPP Library
|
|
|
|
:copyright: (c) 2011 Nathanael C. Fritz
|
|
:license: MIT, see LICENSE for more details
|
|
"""
|
|
|
|
from __future__ import unicode_literals
|
|
|
|
|
|
class JID(object):
|
|
|
|
"""
|
|
A representation of a Jabber ID, or JID.
|
|
|
|
Each JID may have three components: a user, a domain, and an optional
|
|
resource. For example: user@domain/resource
|
|
|
|
When a resource is not used, the JID is called a bare JID.
|
|
The JID is a full JID otherwise.
|
|
|
|
**JID Properties:**
|
|
:jid: Alias for ``full``.
|
|
:full: The value of the full JID.
|
|
:bare: The value of the bare JID.
|
|
:user: The username portion of the JID.
|
|
:domain: The domain name portion of the JID.
|
|
:server: Alias for ``domain``.
|
|
:resource: The resource portion of the JID.
|
|
|
|
:param string jid: A string of the form ``'[user@]domain[/resource]'``.
|
|
"""
|
|
|
|
def __init__(self, jid):
|
|
"""Initialize a new JID"""
|
|
self.reset(jid)
|
|
|
|
def reset(self, jid):
|
|
"""Start fresh from a new JID string.
|
|
|
|
:param string jid: A string of the form ``'[user@]domain[/resource]'``.
|
|
"""
|
|
if isinstance(jid, JID):
|
|
jid = jid.full
|
|
self._full = self._jid = jid
|
|
self._domain = None
|
|
self._resource = None
|
|
self._user = None
|
|
self._bare = None
|
|
|
|
def __getattr__(self, name):
|
|
"""Handle getting the JID values, using cache if available.
|
|
|
|
:param name: One of: user, server, domain, resource,
|
|
full, or bare.
|
|
"""
|
|
if name == 'resource':
|
|
if self._resource is None and '/' in self._jid:
|
|
self._resource = self._jid.split('/', 1)[-1]
|
|
return self._resource or ""
|
|
elif name == 'user':
|
|
if self._user is None:
|
|
if '@' in self._jid:
|
|
self._user = self._jid.split('@', 1)[0]
|
|
else:
|
|
self._user = self._user
|
|
return self._user or ""
|
|
elif name in ('server', 'domain', 'host'):
|
|
if self._domain is None:
|
|
self._domain = self._jid.split('@', 1)[-1].split('/', 1)[0]
|
|
return self._domain or ""
|
|
elif name in ('full', 'jid'):
|
|
return self._jid or ""
|
|
elif name == 'bare':
|
|
if self._bare is None:
|
|
self._bare = self._jid.split('/', 1)[0]
|
|
return self._bare or ""
|
|
|
|
def __setattr__(self, name, value):
|
|
"""Edit a JID by updating it's individual values, resetting the
|
|
generated JID in the end.
|
|
|
|
Arguments:
|
|
name -- The name of the JID part. One of: user, domain,
|
|
server, resource, full, jid, or bare.
|
|
value -- The new value for the JID part.
|
|
"""
|
|
if name in ('resource', 'user', 'domain'):
|
|
object.__setattr__(self, "_%s" % name, value)
|
|
self.regenerate()
|
|
elif name in ('server', 'domain', 'host'):
|
|
self.domain = value
|
|
elif name in ('full', 'jid'):
|
|
self.reset(value)
|
|
self.regenerate()
|
|
elif name == 'bare':
|
|
if '@' in value:
|
|
u, d = value.split('@', 1)
|
|
object.__setattr__(self, "_user", u)
|
|
object.__setattr__(self, "_domain", d)
|
|
else:
|
|
object.__setattr__(self, "_user", '')
|
|
object.__setattr__(self, "_domain", value)
|
|
self.regenerate()
|
|
else:
|
|
object.__setattr__(self, name, value)
|
|
|
|
def regenerate(self):
|
|
"""Generate a new JID based on current values, useful after editing."""
|
|
jid = ""
|
|
if self.user:
|
|
jid = "%s@" % self.user
|
|
jid += self.domain
|
|
if self.resource:
|
|
jid += "/%s" % self.resource
|
|
self.reset(jid)
|
|
|
|
def __str__(self):
|
|
"""Use the full JID as the string value."""
|
|
return self.full
|
|
|
|
def __repr__(self):
|
|
return self.full
|
|
|
|
def __eq__(self, other):
|
|
"""
|
|
Two JIDs are considered equal if they have the same full JID value.
|
|
"""
|
|
other = JID(other)
|
|
return self.full == other.full
|
|
|
|
def __ne__(self, other):
|
|
"""Two JIDs are considered unequal if they are not equal."""
|
|
return not self == other
|
|
|
|
def __hash__(self):
|
|
"""Hash a JID based on the string version of its full JID."""
|
|
return hash(self.full)
|
|
|
|
def __copy__(self):
|
|
return JID(self.jid)
|