commit 0cc53ef4bc2c7c69c62e71401a16bfefe2f23fce
author Neels Hofmeyr <neels@hofmeyr.de> Mon May 29 01:35:00 2017 +0200
committer Neels Hofmeyr <nhofmeyr@sysmocom.de> Mon May 29 20:29:09 2017 +0000
tree da7ca4eef276dcfb41fb36dc7834180cd53af8e3
parent bef36bc5e3f3ab952820ff8955865d8a581dffd9

log.Origin: return self on __enter__

Allows using 'with some_origin() as foo:' constructs.
Not used actively, but is sometimes useful during debugging sessions.

Change-Id: I7a6463ee39761775305dd2272c24f248552db4ad

src/osmo_gsm_tester/log.py

diff --git a/src/osmo_gsm_tester/log.py b/src/osmo_gsm_tester/log.py
index 3af7bbe..c4a9861 100644
--- a/src/osmo_gsm_tester/log.py
+++ b/src/osmo_gsm_tester/log.py
@@ -332,8 +332,9 @@
 
     def __enter__(self):
         if not self.set_child_of(Origin._global_current_origin):
-            return
+            return self
         Origin._global_current_origin = self
+        return self
 
     def __exit__(self, *exc_info):
         rc = None