mirror of https://github.com/jumpserver/jumpserver
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
99 lines
3.7 KiB
99 lines
3.7 KiB
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" |
|
] |
|
extra_kwargs = { |
|
"task_id": { |
|
"label": _("Task id"), |
|
}, |
|
"job": { |
|
"label": _("Job"), |
|
} |
|
} |
|
|
|
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
|
|
|