mirror of
https://github.com/correl/SleekXMPP.git
synced 2024-12-24 03:00:17 +00:00
141 lines
4.5 KiB
Python
141 lines
4.5 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
|