Merge branch 'master' of https://github.com/StuartAxelOwen/websocket-client into StuartAxelOwen-master

1 files changed, 13 insertions, 0 deletions

diff --git a/websocket/_core.py b/websocket/_core.py
index 7d73d76..4d7ddec 100644
--- a/websocket/_core.py
+++ b/websocket/_core.py
@@ -154,6 +154,19 @@ class WebSocket(object):
         else:
             self.lock = NoLock()
 
+    def __iter__(self):
+        """
+        Allow iteration over websocket, implying sequential `recv` executions.
+        """
+        while True:
+            yield self.recv()
+
+    def __next__(self):
+        return self.recv()
+
+    def next(self):
+        return self.__next__()
+
     def fileno(self):
         return self.sock.fileno()