Rename to slixmpp

3 files changed, 245 insertions, 0 deletions

diff --git a/slixmpp/plugins/xep_0202/__init__.py b/slixmpp/plugins/xep_0202/__init__.py
new file mode 100644
index 00000000..1197eabc
--- /dev/null
+++ b/slixmpp/plugins/xep_0202/__init__.py
@@ -0,0 +1,20 @@
+"""
+    Slixmpp: The Slick XMPP Library
+    Copyright (C) 2011 Nathanael C. Fritz, Lance J.T. Stout
+    This file is part of Slixmpp.
+
+    See the file LICENSE for copying permission.
+"""
+
+from slixmpp.plugins.base import register_plugin
+
+from slixmpp.plugins.xep_0202 import stanza
+from slixmpp.plugins.xep_0202.stanza import EntityTime
+from slixmpp.plugins.xep_0202.time import XEP_0202
+
+
+register_plugin(XEP_0202)
+
+
+# Retain some backwards compatibility
+xep_0202 = XEP_0202
diff --git a/slixmpp/plugins/xep_0202/stanza.py b/slixmpp/plugins/xep_0202/stanza.py
new file mode 100644
index 00000000..c855663b
--- /dev/null
+++ b/slixmpp/plugins/xep_0202/stanza.py
@@ -0,0 +1,127 @@
+"""
+    Slixmpp: The Slick XMPP Library
+    Copyright (C) 2010 Nathanael C. Fritz
+    This file is part of Slixmpp.
+
+    See the file LICENSE for copying permission.
+"""
+
+import logging
+import datetime as dt
+
+from slixmpp.xmlstream import ElementBase
+from slixmpp.plugins import xep_0082
+from slixmpp.thirdparty import tzutc, tzoffset
+
+
+class EntityTime(ElementBase):
+
+    """
+    The <time> element represents the local time for an XMPP agent.
+    The time is expressed in UTC to make synchronization easier
+    between entities, but the offset for the local timezone is also
+    included.
+
+    Example <time> stanzas:
+        <iq type="result">
+          <time xmlns="urn:xmpp:time">
+            <utc>2011-07-03T11:37:12.234569</utc>
+            <tzo>-07:00</tzo>
+          </time>
+        </iq>
+
+    Stanza Interface:
+        time -- The local time for the entity (updates utc and tzo).
+        utc  -- The UTC equivalent to local time.
+        tzo  -- The local timezone offset from UTC.
+
+    Methods:
+        get_time -- Return local time datetime object.
+        set_time -- Set UTC and TZO fields.
+        del_time -- Remove both UTC and TZO fields.
+        get_utc  -- Return datetime object of UTC time.
+        set_utc  -- Set the UTC time.
+        get_tzo  -- Return tzinfo object.
+        set_tzo  -- Set the local timezone offset.
+    """
+
+    name = 'time'
+    namespace = 'urn:xmpp:time'
+    plugin_attrib = 'entity_time'
+    interfaces = set(('tzo', 'utc', 'time'))
+    sub_interfaces = interfaces
+
+    def set_time(self, value):
+        """
+        Set both the UTC and TZO fields given a time object.
+
+        Arguments:
+            value -- A datetime object or properly formatted
+                     string equivalent.
+        """
+        date = value
+        if not isinstance(value, dt.datetime):
+            date = xep_0082.parse(value)
+        self['utc'] = date
+        self['tzo'] = date.tzinfo
+
+    def get_time(self):
+        """
+        Return the entity's local time based on the UTC and TZO data.
+        """
+        date = self['utc']
+        tz = self['tzo']
+        return date.astimezone(tz)
+
+    def del_time(self):
+        """Remove both the UTC and TZO fields."""
+        del self['utc']
+        del self['tzo']
+
+    def get_tzo(self):
+        """
+        Return the timezone offset from UTC as a tzinfo object.
+        """
+        tzo = self._get_sub_text('tzo')
+        if tzo == '':
+            tzo = 'Z'
+        time = xep_0082.parse('00:00:00%s' % tzo)
+        return time.tzinfo
+
+    def set_tzo(self, value):
+        """
+        Set the timezone offset from UTC.
+
+        Arguments:
+            value -- Either a tzinfo object or the number of
+                     seconds (positive or negative) to offset.
+        """
+        time = xep_0082.time(offset=value)
+        if xep_0082.parse(time).tzinfo == tzutc():
+            self._set_sub_text('tzo', 'Z')
+        else:
+            self._set_sub_text('tzo', time[-6:])
+
+    def get_utc(self):
+        """
+        Return the time in UTC as a datetime object.
+        """
+        value = self._get_sub_text('utc')
+        if value == '':
+            return xep_0082.parse(xep_0082.datetime())
+        return xep_0082.parse('%sZ' % value)
+
+    def set_utc(self, value):
+        """
+        Set the time in UTC.
+
+        Arguments:
+            value -- A datetime object or properly formatted
+                     string equivalent.
+        """
+        date = value
+        if not isinstance(value, dt.datetime):
+            date = xep_0082.parse(value)
+        date = date.astimezone(tzutc())
+        value = xep_0082.format_datetime(date)[:-1]
+        self._set_sub_text('utc', value)
diff --git a/slixmpp/plugins/xep_0202/time.py b/slixmpp/plugins/xep_0202/time.py
new file mode 100644
index 00000000..3b335e28
--- /dev/null
+++ b/slixmpp/plugins/xep_0202/time.py
@@ -0,0 +1,98 @@
+"""

+    Slixmpp: The Slick XMPP Library

+    Copyright (C) 2010 Nathanael C. Fritz

+    This file is part of Slixmpp.

+

+    See the file LICENSE for copying permission.

+"""

+

+import logging

+

+from slixmpp.stanza.iq import Iq

+from slixmpp.xmlstream import register_stanza_plugin

+from slixmpp.xmlstream.handler import Callback

+from slixmpp.xmlstream.matcher import StanzaPath

+from slixmpp.plugins import BasePlugin

+from slixmpp.plugins import xep_0082

+from slixmpp.plugins.xep_0202 import stanza

+

+

+log = logging.getLogger(__name__)

+

+

+class XEP_0202(BasePlugin):

+

+    """

+    XEP-0202: Entity Time

+    """

+

+    name = 'xep_0202'

+    description = 'XEP-0202: Entity Time'

+    dependencies = set(['xep_0030', 'xep_0082'])

+    stanza = stanza

+    default_config = {

+        #: As a default, respond to time requests with the

+        #: local time returned by XEP-0082. However, a

+        #: custom function can be supplied which accepts

+        #: the JID of the entity to query for the time.

+        'local_time': None,

+        'tz_offset': 0

+    }

+

+    def plugin_init(self):

+        """Start the XEP-0203 plugin."""

+

+        if not self.local_time:

+            def default_local_time(jid):

+                return xep_0082.datetime(offset=self.tz_offset)

+

+            self.local_time = default_local_time

+

+        self.xmpp.register_handler(

+            Callback('Entity Time',

+                 StanzaPath('iq/entity_time'),

+                 self._handle_time_request))

+        register_stanza_plugin(Iq, stanza.EntityTime)

+

+    def plugin_end(self):

+        self.xmpp['xep_0030'].del_feature(feature='urn:xmpp:time')

+        self.xmpp.remove_handler('Entity Time')

+

+    def session_bind(self, jid):

+        self.xmpp['xep_0030'].add_feature('urn:xmpp:time')

+

+    def _handle_time_request(self, iq):

+        """

+        Respond to a request for the local time.

+

+        The time is taken from self.local_time(), which may be replaced

+        during plugin configuration with a function that maps JIDs to

+        times.

+

+        Arguments:

+            iq -- The Iq time request stanza.

+        """

+        iq.reply()

+        iq['entity_time']['time'] = self.local_time(iq['to'])

+        iq.send()

+

+    def get_entity_time(self, to, ifrom=None, **iqargs):

+        """

+        Request the time from another entity.

+

+        Arguments:

+            to       -- JID of the entity to query.

+            ifrom    -- Specifiy the sender's JID.

+            block    -- If true, block and wait for the stanzas' reply.

+            timeout  -- The time in seconds to block while waiting for

+                        a reply. If None, then wait indefinitely.

+            callback -- Optional callback to execute when a reply is

+                        received instead of blocking and waiting for

+                        the reply.

+        """

+        iq = self.xmpp.Iq()

+        iq['type'] = 'get'

+        iq['to'] = to

+        iq['from'] = ifrom

+        iq.enable('entity_time')

+        return iq.send(**iqargs)