From 0cc53ef4bc2c7c69c62e71401a16bfefe2f23fce Mon Sep 17 00:00:00 2001
From: Neels Hofmeyr <neels@hofmeyr.de>
Date: Mon, 29 May 2017 01:35:00 +0200
Subject: 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 | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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 @@ class Origin:
 
     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
-- 
cgit v1.2.3