jumpserver/apps/ops/serializers/playbook.py

import os

from rest_framework import serializers

from common.drf.fields import ReadableHiddenField
from ops.models import Playbook
from orgs.mixins.serializers import BulkOrgResourceModelSerializer


def parse_playbook_name(path):
    file_name = os.path.split(path)[-1]
    return file_name.split(".")[-2]


class PlaybookSerializer(BulkOrgResourceModelSerializer):
    creator = ReadableHiddenField(default=serializers.CurrentUserDefault())
    path = serializers.FileField(required=False)

    def create(self, validated_data):
        name = validated_data.get('name')
        if not name:
            path = validated_data.get('path').name
            validated_data['name'] = parse_playbook_name(path)
        return super().create(validated_data)

    class Meta:
        model = Playbook
        read_only_fields = ["id", "date_created", "date_updated"]
        fields = read_only_fields + [
            "id", 'path', "name", "comment", "creator",
        ]