[tor-commits] [stem/master] Context methods for the Circuit class

[atagar@xxxxxxxxxxxxxx]

commit cc95059da5a8e34d10c16f5a226df21ffb5b43c9
Author: Damian Johnson <atagar@xxxxxxxxxxxxxx>
Date:   Tue Apr 17 10:09:14 2018 -0700

    Context methods for the Circuit class
    
    Like our Relay class circuits are intended to be used for a time then closed.
    Decorating it with the methods necessary so it can be used through 'with'
    clauses.
---
 stem/client/__init__.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/stem/client/__init__.py b/stem/client/__init__.py
index 2c6b7ea6..be7e16ad 100644
--- a/stem/client/__init__.py
+++ b/stem/client/__init__.py
@@ -256,3 +256,9 @@ class Circuit(object):
     with self.relay._orport_lock:
       self.relay._orport.send(stem.client.cell.DestroyCell(self.id).pack(self.relay.link_protocol))
       del self.relay._circuits[self.id]
+
+  def __enter__(self):
+    return self
+
+  def __exit__(self, exit_type, value, traceback):
+    self.close()



_______________________________________________
tor-commits mailing list
tor-commits@xxxxxxxxxxxxxxxxxxxx
https://lists.torproject.org/cgi-bin/mailman/listinfo/tor-commits