mirror of
https://github.com/jumpserver/jumpserver.git
synced 2026-01-21 14:13:53 +00:00
9a62a7aaab
* pref: 修改 activity log * perf: 优化 acitivity * pref: 修改 activity * fix: 修复一些运行问题 * fix: app.py 中添加 tasks import * fix: 添加 activity_callback * fix: 添加 execute_account_backup_plan activity_callback * fix: 添加 activity_callback -> gather_asset_accounts * fix: 对 celery 任务添加 activity_callback 回调 * fix: 修改翻译 --------- Co-authored-by: ibuler <ibuler@qq.com> Co-authored-by: jiangweidong <weidong.jiang@fit2cloud.com> Co-authored-by: Bai <baijiangjie@gmail.com>
37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
from celery import shared_task
|
|
from django.utils.translation import gettext_lazy as _
|
|
|
|
from orgs.models import Organization
|
|
from orgs.utils import tmp_to_org
|
|
from ops.celery.decorator import register_as_period_task
|
|
from assets.utils import check_node_assets_amount
|
|
|
|
from common.utils.lock import AcquireFailed
|
|
from common.utils import get_logger
|
|
from common.const.crontab import CRONTAB_AT_AM_TWO
|
|
|
|
logger = get_logger(__file__)
|
|
|
|
|
|
@shared_task(verbose_name=_('Check the amount of assets under the node'))
|
|
def check_node_assets_amount_task(org_id=None):
|
|
if org_id is None:
|
|
orgs = Organization.objects.all()
|
|
else:
|
|
orgs = [Organization.get_instance(org_id)]
|
|
|
|
for org in orgs:
|
|
try:
|
|
with tmp_to_org(org):
|
|
check_node_assets_amount()
|
|
except AcquireFailed:
|
|
error = _('The task of self-checking is already running '
|
|
'and cannot be started repeatedly')
|
|
logger.error(error)
|
|
|
|
|
|
@shared_task(verbose_name=_('Periodic check the amount of assets under the node'))
|
|
@register_as_period_task(crontab=CRONTAB_AT_AM_TWO)
|
|
def check_node_assets_amount_period_task():
|
|
check_node_assets_amount_task()
|