def instrument_celery_loaders_base(module): def force_application_activation(*args, **kwargs): application_instance().activate() wrap_pre_function(module, 'BaseLoader.init_worker', force_application_activation)
def instrument_billiard_pool(module): def force_agent_shutdown(*args, **kwargs): shutdown_agent() if hasattr(module, 'Worker'): wrap_pre_function(module, 'Worker._do_exit', force_agent_shutdown)
def instrument_tastypie_resources(module): _wrap_view = module.Resource.wrap_view module.Resource.wrap_view = ObjectWrapper( _wrap_view, None, outer_fn_wrapper) wrap_pre_function(module, 'Resource._handle_500', wrap_handle_exception)
def instrument_tastypie_resources(module): _wrap_view = module.Resource.wrap_view module.Resource.wrap_view = ObjectWrapper(_wrap_view, None, outer_fn_wrapper) wrap_pre_function(module, 'Resource._handle_500', wrap_handle_exception)