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_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_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)
