Add timeout API to Plugin.

Ben Weaver · Ben Weaver · commit ccae4c89c860 · 2010-02-05T16:41:27.000-05:00
diff --git a/xmpp/core.py b/xmpp/core.py
@@ -74,10 +74,14 @@ def on_stream_bound(self, bindings):
         self.authJID = bindings.jid
         self.resources = bindings.resources
 
-    def timeout(self, delay, callback):
+    def add_timeout(self, delay, callback):
         self.stream.io.add_timeout(time.time() + delay, callback)
         return self
 
+    def remove_timeout(self, callback):
+        self.stream.io.remove_timeout(callback)
+        return self
+
     ### ---------- Incoming Stream ----------
 
     ## These are callbacks for the XMPPStream.
diff --git a/xmpp/plugin.py b/xmpp/plugin.py
@@ -509,6 +509,14 @@ def close(self):
         self.__core.close()
         return None
 
+    def add_timeout(self, *args):
+        self.__core.add_timeout(*args)
+        return self
+
+    def clear_timeout(self, *args):
+        self.__core.remove_timeout(*args)
+        return self
+
     ## ---------- Low-level Stream ----------
 
     def open_stream(self, *args):
