mirror of https://github.com/jumpserver/jumpserver
24 lines
654 B
Python
24 lines
654 B
Python
|
from django.utils import translation
|
||
|
|
from django.core.cache import cache
|
||
|
|
from celery.signals import task_prerun, before_task_publish
|
||
|
|
|
||
|
|
|
||
|
|
TASK_LANG_CACHE_KEY = 'TASK_LANG_{}'
|
||
|
|
TASK_LANG_CACHE_TTL = 1800
|
||
|
|
|
||
|
|
|
||
|
|
@before_task_publish.connect()
|
||
|
|
def before_task_publish(headers=None, **kwargs):
|
||
|
|
task_id = headers.get('id')
|
||
|
|
current_lang = translation.get_language()
|
||
|
|
key = TASK_LANG_CACHE_KEY.format(task_id)
|
||
|
|
cache.set(key, current_lang, 1800)
|
||
|
|
|
||
|
|
|
||
|
|
@task_prerun.connect()
|
||
|
|
def on_celery_task_pre_run(task_id='', **kwargs):
|
||
|
|
key = TASK_LANG_CACHE_KEY.format(task_id)
|
||
|
|
task_lang = cache.get(key)
|
||
|
|
if task_lang:
|
||
|
|
translation.activate(task_lang)
|