jumpserver/apps/ops/serializers/playbook.py
fit2bot c96ae1022b
feat: Supports running adhoc,playbook with variable (#14417)
* perf:Create a job that supports adding node parameters

* feat: add variable model

* feat: Modify Variable and AdHoc models,

* feat: Parameters can be set when running job

* feat: Supports setting  variable type

* feat: Supports running adhoc with parameters

* feat: Supports running playbook with parameters

* fix: Translate

* feat: Support setting variables for scheduled tasks

* perf: Translate

---------

Co-authored-by: wangruidong <940853815@qq.com>
2024-11-07 10:38:34 +08:00

37 lines
1.4 KiB
Python

import os
from django.utils.translation import gettext_lazy as _
from rest_framework import serializers
from common.serializers import WritableNestedModelSerializer
from common.serializers.fields import ReadableHiddenField
from common.serializers.mixin import CommonBulkModelSerializer
from ops.models import Playbook
from .mixin import ScopeSerializerMixin
from ops.serializers.variable import PlaybookVariableSerializer
def parse_playbook_name(path):
file_name = os.path.split(path)[-1]
return file_name.split(".")[-2]
class PlaybookSerializer(ScopeSerializerMixin, CommonBulkModelSerializer, WritableNestedModelSerializer):
creator = ReadableHiddenField(default=serializers.CurrentUserDefault())
path = serializers.FileField(required=False)
variable = PlaybookVariableSerializer(many=True, required=False, allow_null=True, label=_('Variable'))
def to_internal_value(self, data):
name = data.get('name', False)
if not name and data.get('path'):
data['name'] = parse_playbook_name(data['path'].name)
return super().to_internal_value(data)
class Meta:
model = Playbook
read_only_fields = ["id", "date_created", "date_updated", "created_by"]
fields_m2m = ['variable']
fields = read_only_fields + fields_m2m + [
"id", 'path', 'scope', "name", "comment", "creator",
'create_method', 'vcs_url',
]