From 081565358dd5f873ce7358305272eb23f9039bf3 Mon Sep 17 00:00:00 2001 From: David Wilson Date: Wed, 10 Aug 2016 22:13:57 +0100 Subject: [PATCH] utils.run_with_broker. --- econtext/utils.py | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) diff --git a/econtext/utils.py b/econtext/utils.py index 6b371e34..2e6bd331 100644 --- a/econtext/utils.py +++ b/econtext/utils.py @@ -2,13 +2,16 @@ import econtext +def run_with_broker(func, *args, **kwargs): + broker = econtext.Broker() + try: + return func(broker, *args, **kwargs) + finally: + broker.Finalize() + + def with_broker(func): def wrapper(*args, **kwargs): - broker = econtext.Broker() - try: - return func(broker, *args, **kwargs) - finally: - broker.Finalize() - + return run_with_broker(*args, **kwargs) wrapper.func_name = func.func_name return wrapper