debug: fix 'app' when using pdb in stress tool

e3414b6f · Julien Muchembled · c611c48f · e3414b6f
Commit e3414b6f authored May 28, 2020 by Julien Muchembled
Hide whitespace changes
Inline Side-by-side

Showing with 13 additions and 9 deletions

neo/debug.py neo/debug.py +13 -9

No files found.
--- a/neo/debug.py
+++ b/neo/debug.py
@@ -13,16 +13,20 @@ import sys
 def app_set():
    try:
-        return sys.modules['neo.lib.threaded_app'].app_set
+        app_set = sys.modules['neo.lib.threaded_app'].app_set
    except KeyError:
-        f = sys._getframe(4)
+        pass
-        try:
+    else:
-            while f.f_code.co_name != 'run' or \
+        if app_set:
-                  f.f_locals.get('self').__class__.__name__ != 'Application':
+            return app_set
-                f = f.f_back
+    f = sys._getframe(4)
-            return f.f_locals['self'],
+    try:
-        except AttributeError:
+        while f.f_code.co_name != 'run' or \
-            return ()
+              f.f_locals.get('self').__class__.__name__ != 'Application':
+            f = f.f_back
+        return f.f_locals['self'],
+    except AttributeError:
+        return ()
 def defer(task):
    def wrapper():