auto close socket

m2mclient/client.py

@@ -37,18 +37,19 @@ def client(self):
     def run(self):
         """Main thread loop."""
         try:
-            for event in self.ws:
-                if event.name == 'rejected':
-                    self.error = event.reason
-                elif event.name == 'disconnected':
-                    if not event.graceful:
+            with self.ws:
+                for event in self.ws:
+                    if event.name == 'rejected':
                         self.error = event.reason
-                elif event.name == 'ready':
-                    self.running = True
-                    self.on_startup()
-                    self.ready_event.set()
-                elif event.name == 'binary':
-                    self.on_binary(event.data)
+                    elif event.name == 'disconnected':
+                        if not event.graceful:
+                            self.error = event.reason
+                    elif event.name == 'ready':
+                        self.running = True
+                        self.on_startup()
+                        self.ready_event.set()
+                    elif event.name == 'binary':
+                        self.on_binary(event.data)
         finally:
             self.running = False
             self.ready_event.set()

setup.py

@@ -27,7 +27,7 @@
     exclude_package_data={'': ['_*', 'docs/*']},
     classifiers=classifiers,
     install_requires=[
-        'lomond==0.1.6',
+        'lomond==0.1.7a0',
         'wsaccel==0.6.2'
     ]
 )