mirror of https://github.com/jumpserver/jumpserver
37 lines
1.4 KiB
Python
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',
|
|
]
|