fix improperly handled CancelledError on shutdown

2019-01-09 11:29:28 -05:00 · 2019-01-09 11:29:28 -05:00 · f4bf260cdb
parent 442138ef36
commit f4bf260cdb
1 changed files with 3 additions and 1 deletions
--- a/torba/client/basenetwork.py
+++ b/torba/client/basenetwork.py
@ -88,7 +88,9 @@ class BaseNetwork:
                self._on_connected_controller.add(True)
                delay = 0.0
                await self.client.on_disconnected.first
-            except (Exception, CancelledError):  # pylint: disable=broad-except
+            except CancelledError:
+                self.running = False
+            except Exception:  # pylint: disable=broad-except
                log.exception("Connecting to %s raised an exception:", connection_string)
            if not self.running:
                return