You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
mitogen/econtext/utils.py

31 lines
668 B
Python

import logging
import econtext
import econtext.core
import econtext.master
def log_to_file(path, level=logging.DEBUG):
log = logging.getLogger('')
fp = open(path, 'w', 1)
econtext.core.set_cloexec(fp.fileno())
log.setLevel(level)
log.handlers.insert(0, logging.StreamHandler(fp))
def run_with_broker(func, *args, **kwargs):
broker = econtext.master.Broker()
try:
return func(broker, *args, **kwargs)
finally:
broker.shutdown()
broker.wait()
def with_broker(func):
def wrapper(*args, **kwargs):
return run_with_broker(*args, **kwargs)
wrapper.func_name = func.func_name
return wrapper