github
/
jumpserver
mirror of https://github.com/jumpserver/jumpserver
1
0
Fork 0
Code Issues Releases Wiki Activity

perf: Add check for SECURITY_COMMAND_EXECUTION settings in ops tasks

pull/14429/head
wangruidong 2024-11-07 15:22:18 +08:00 committed by Bryan
parent 11b3bafd5a
commit 5724912480
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
apps/ops/tasks.py
View File

@ -6,6 +6,7 @@ from celery.exceptions import SoftTimeLimitExceeded
from django.utils import timezone from django.utils import timezone
from django.utils.translation import gettext_lazy as _ from django.utils.translation import gettext_lazy as _
from django_celery_beat.models import PeriodicTask from django_celery_beat.models import PeriodicTask
from django.conf import settings
from common.const.crontab import CRONTAB_AT_AM_TWO from common.const.crontab import CRONTAB_AT_AM_TWO
from common.utils import get_logger, get_object_or_none, get_log_keep_day from common.utils import get_logger, get_object_or_none, get_log_keep_day
@ -56,6 +57,8 @@ def _run_ops_job_execution(execution):
) )
) )
def run_ops_job(job_id): def run_ops_job(job_id):
if not settings.SECURITY_COMMAND_EXECUTION:
return
with tmp_to_root_org(): with tmp_to_root_org():
job = get_object_or_none(Job, id=job_id) job = get_object_or_none(Job, id=job_id)
if not job: if not job:
@ -89,6 +92,8 @@ def job_execution_task_activity_callback(self, execution_id, *args, **kwargs):
) )
) )
def run_ops_job_execution(execution_id, **kwargs): def run_ops_job_execution(execution_id, **kwargs):
if not settings.SECURITY_COMMAND_EXECUTION:
return
with tmp_to_root_org(): with tmp_to_root_org():
execution = get_object_or_none(JobExecution, id=execution_id) execution = get_object_or_none(JobExecution, id=execution_id)