mirror of
https://github.com/jlengrand/bugsink.git
synced 2026-03-10 08:01:17 +00:00
30 lines
1.2 KiB
Python
30 lines
1.2 KiB
Python
import json
|
|
|
|
from . import registry
|
|
|
|
from .models import Task, wakeup_server
|
|
from .settings import get_settings
|
|
|
|
|
|
def shared_task(function):
|
|
def delayed_function(*args, **kwargs):
|
|
if get_settings().TASK_ALWAYS_EAGER:
|
|
# To ensure that args and kwargs are json-serializable in automatic tests and when using the single-server
|
|
# setup we do these 2 .dumps calls here. The cost should be negligible in general (args/kwargs are small),
|
|
# so even when single-server is used as a "production" server this shouldn't be a problem.
|
|
json.dumps(args), json.dumps(kwargs)
|
|
function(*args, **kwargs)
|
|
# nothing is returned, because when calling x.delay(...) no result can be expected (it isn't available for
|
|
# the non-eager case either).
|
|
return
|
|
|
|
# No need for a transaction: we just write something (not connected to any other object, and we will never touch
|
|
# it again).
|
|
Task.objects.create(task_name=name, args=json.dumps(args), kwargs=json.dumps(kwargs))
|
|
wakeup_server()
|
|
|
|
name = function.__module__ + "." + function.__name__
|
|
function.delay = delayed_function
|
|
registry[name] = function
|
|
return function
|