mirror of https://github.com/jumpserver/jumpserver
92 lines
3.5 KiB
Python
92 lines
3.5 KiB
Python
import uuid
|
|
|
|
from django.utils.translation import gettext_lazy as _
|
|
from rest_framework import serializers
|
|
|
|
from assets.models import Asset
|
|
from common.serializers.fields import ReadableHiddenField
|
|
from ops.mixin import PeriodTaskSerializerMixin
|
|
from ops.models import Job, JobExecution
|
|
from orgs.mixins.serializers import BulkOrgResourceModelSerializer
|
|
|
|
|
|
class JobSerializer(BulkOrgResourceModelSerializer, PeriodTaskSerializerMixin):
|
|
creator = ReadableHiddenField(default=serializers.CurrentUserDefault())
|
|
run_after_save = serializers.BooleanField(label=_("Execute after saving"), default=False, required=False)
|
|
nodes = serializers.ListField(required=False, child=serializers.CharField())
|
|
date_last_run = serializers.DateTimeField(label=_('Date last run'), read_only=True)
|
|
name = serializers.CharField(label=_('Name'), max_length=128, allow_blank=True, required=False)
|
|
assets = serializers.PrimaryKeyRelatedField(label=_('Assets'), queryset=Asset.objects, many=True, required=False)
|
|
|
|
def to_internal_value(self, data):
|
|
instant = data.get('instant', False)
|
|
job_type = data.get('type', '')
|
|
_uid = str(uuid.uuid4()).split('-')[-1]
|
|
if instant:
|
|
data['name'] = f'job-{_uid}'
|
|
if job_type == 'upload_file':
|
|
data['name'] = f'upload_file-{_uid}'
|
|
return super().to_internal_value(data)
|
|
|
|
def get_request_user(self):
|
|
request = self.context.get('request')
|
|
user = request.user if request else None
|
|
return user
|
|
|
|
class Meta:
|
|
model = Job
|
|
read_only_fields = [
|
|
"id", "date_last_run", "date_created",
|
|
"date_updated", "average_time_cost"
|
|
]
|
|
fields = read_only_fields + [
|
|
"name", "instant", "type", "module",
|
|
"args", "playbook", "assets",
|
|
"runas_policy", "runas", "creator",
|
|
"use_parameter_define", "parameters_define",
|
|
"timeout", "chdir", "comment", "summary",
|
|
"is_periodic", "interval", "crontab", "nodes",
|
|
"run_after_save"
|
|
]
|
|
extra_kwargs = {
|
|
'average_time_cost': {'label': _('Duration')},
|
|
}
|
|
|
|
|
|
class FileSerializer(serializers.Serializer):
|
|
files = serializers.FileField(allow_empty_file=False, max_length=128)
|
|
|
|
class Meta:
|
|
ref_name = "JobFileSerializer"
|
|
|
|
|
|
class JobTaskStopSerializer(serializers.Serializer):
|
|
task_id = serializers.CharField(max_length=128)
|
|
|
|
class Meta:
|
|
ref_name = "JobTaskStopSerializer"
|
|
|
|
|
|
class JobExecutionSerializer(BulkOrgResourceModelSerializer):
|
|
creator = ReadableHiddenField(default=serializers.CurrentUserDefault())
|
|
job_type = serializers.ReadOnlyField(label=_("Job type"))
|
|
material = serializers.ReadOnlyField(label=_("Command"))
|
|
is_success = serializers.ReadOnlyField(label=_("Is success"))
|
|
is_finished = serializers.ReadOnlyField(label=_("Is finished"))
|
|
time_cost = serializers.ReadOnlyField(label=_("Time cost"))
|
|
|
|
class Meta:
|
|
model = JobExecution
|
|
read_only_fields = ["id", "task_id", "timedelta", "time_cost",
|
|
'is_finished', 'date_start', 'date_finished',
|
|
'date_created', 'is_success', 'job_type',
|
|
'summary', 'material']
|
|
fields = read_only_fields + [
|
|
"job", "parameters", "creator"
|
|
]
|
|
|
|
def validate_job(self, job_obj):
|
|
if job_obj.creator != self.context['request'].user:
|
|
raise serializers.ValidationError(_("You do not have permission for the current job."))
|
|
return job_obj
|