mirror of https://github.com/jumpserver/jumpserver
28 lines
892 B
Python
28 lines
892 B
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
|
|
|
|
logger = get_logger(__file__)
|
|
|
|
|
|
@shared_task(queue='celery_heavy_tasks')
|
|
def check_node_assets_amount_task(org_id=Organization.ROOT_ID):
|
|
try:
|
|
with tmp_to_org(Organization.get_instance(org_id)):
|
|
check_node_assets_amount()
|
|
except AcquireFailed:
|
|
logger.error(_('The task of self-checking is already running and cannot be started repeatedly'))
|
|
|
|
|
|
@register_as_period_task(crontab='0 2 * * *')
|
|
@shared_task(queue='celery_heavy_tasks')
|
|
def check_node_assets_amount_period_task():
|
|
check_node_assets_amount_task()
|