ticket #8605

sreenatha123 · Apr 16, 2013 · 0339974 · sreenatha123 · Apr 16, 2013 · 0339974
1 parent 1ad1f71
commit 0339974
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/stem/response/__init__.py b/stem/response/__init__.py
@@ -143,8 +143,17 @@ def from_str(content):
 
     :returns: stem.response.ControlMessage instance
     """
-
-    return stem.socket.recv_message(StringIO.StringIO(content))
+    message = stem.socket.recv_message(StringIO.StringIO(content))
+    message_content = message.content()
+    keyword = message_content[0][2].split()[0]
+    if keyword == 'PROTOCOLINFO':
+      stem.response.convert("PROTOCOLINFO", message)
+    elif keyword == 'AUTHCHALLENGE':
+      stem.response.convert("AUTHCHALLENGE", message)
+    elif message_content[0][0][0]=='6':
+      stem.response.convert("EVENT", message)
+
+    return message
 
   from_str = staticmethod(from_str)