Changeset 63:b8139f79276f for wokkel/compat.py

Timestamp:

Jul 6, 2009, 9:26:47 AM (12 years ago)

Author:

Ralph Meijer <ralphm@…>

Branch:

default

Convert:

svn:b33ecbfc-034c-dc11-8662-000475d9059e/trunk@175

Message:

Make IQ timeouts work for InternalComponents?.

This introduces wokkel.compat.IQ to work with older Twisted versions, too.

Author: ralphm.
Fixes #53.

File:

• wokkel/compat.py (modified) (2 diffs)

wokkel/compat.py

@@ -1 +1 @@
 # -*- test-case-name: wokkel.test.test_compat -*-
 #
-# Copyright (c) 2001-2008 Twisted Matrix Laboratories.
+# Copyright (c) 2001-2009 Twisted Matrix Laboratories.
 # See LICENSE for details.
 
 from twisted.internet import protocol
 from twisted.words.protocols.jabber import xmlstream
-from twisted.words.xish import domish
 
 class BootstrapMixin(object):
@@ -95 +94 @@
         self.installBootstraps(xs)
         return xs
+
+
+
+class IQ(xmlstream.IQ):
+    def __init__(self, *args, **kwargs):
+        # Make sure we have a reactor parameter
+        try:
+            reactor = kwargs['reactor']
+        except KeyError:
+            from twisted.internet import reactor
+        kwargs['reactor'] = reactor
+
+        # Check if IQ's init accepts the reactor parameter
+        try:
+            xmlstream.IQ.__init__(self, *args, **kwargs)
+        except TypeError:
+            # Guess not. Remove the reactor parameter and try again.
+            del kwargs['reactor']
+            xmlstream.IQ.__init__(self, *args, **kwargs)
+
+            # Patch the XmlStream instance so that it has a _callLater
+            self._xmlstream._callLater = reactor.callLater