Merge pull request #45 from AlmightyOatmeal/master

Added context managers to SignalFlowClient()
signalfx · Dec 5, 2016 · 426cdb0 · 426cdb0
2 parents 13d1225 + 93bf9cf
commit 426cdb0
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/signalfx/signalflow/__init__.py b/signalfx/signalflow/__init__.py
@@ -19,6 +19,12 @@ def __init__(self, token, endpoint=constants.DEFAULT_STREAM_ENDPOINT,
         self._transport = transport(token, endpoint, timeout)
         self._computations = set([])
 
+    def __exit__(self):
+        return self.close()
+
+    def __del__(self):
+        return self.close()
+
     def _get_params(self, **kwargs):
         return dict((k, v) for k, v in kwargs.items() if v)