Save the stream ID when the stream starts.

2011-08-06 00:44:32 -07:00
parent 5be5b8c02b
commit e83fae3a6f
2 changed files with 14 additions and 1 deletions
--- a/sleekxmpp/basexmpp.py
+++ b/sleekxmpp/basexmpp.py
@@ -138,6 +138,17 @@ class BaseXMPP(XMLStream):
        register_stanza_plugin(Message, Nick)
        register_stanza_plugin(Message, HTMLIM)

+    def start_stream_handler(self, xml):
+        """
+        Save the stream ID once the streams have been established.
+
+        Overrides XMLStream.start_stream_handler.
+
+        Arguments:
+            xml -- The incoming stream's root element.
+        """
+        self.stream_id = xml.get('id', '')
+
    def process(self, *args, **kwargs):
        """
        Overrides XMLStream.process.