mirror of https://github.com/jumpserver/jumpserver
54 lines
1.5 KiB
Python
54 lines
1.5 KiB
Python
# ~*~ coding: utf-8 ~*~
|
|
from __future__ import unicode_literals
|
|
from rest_framework import serializers
|
|
|
|
from .models import Task, AdHoc, AdHocRunHistory
|
|
|
|
|
|
class TaskSerializer(serializers.ModelSerializer):
|
|
class Meta:
|
|
model = Task
|
|
fields = '__all__'
|
|
|
|
|
|
class AdHocSerializer(serializers.ModelSerializer):
|
|
class Meta:
|
|
model = AdHoc
|
|
exclude = ('_tasks', '_options', '_hosts', '_become')
|
|
|
|
def get_field_names(self, declared_fields, info):
|
|
fields = super().get_field_names(declared_fields, info)
|
|
fields.extend(['tasks', 'options', 'hosts', 'become', 'short_id'])
|
|
return fields
|
|
|
|
|
|
class AdHocRunHistorySerializer(serializers.ModelSerializer):
|
|
task = serializers.SerializerMethodField()
|
|
adhoc_short_id = serializers.SerializerMethodField()
|
|
stat = serializers.SerializerMethodField()
|
|
|
|
class Meta:
|
|
model = AdHocRunHistory
|
|
exclude = ('_result', '_summary')
|
|
|
|
@staticmethod
|
|
def get_adhoc_short_id(obj):
|
|
return obj.adhoc.short_id
|
|
|
|
@staticmethod
|
|
def get_task(obj):
|
|
return obj.adhoc.task.id
|
|
|
|
@staticmethod
|
|
def get_stat(obj):
|
|
return {
|
|
"total": len(obj.adhoc.hosts),
|
|
"success": len(obj.summary.get("contacted", [])),
|
|
"failed": len(obj.summary.get("dark", [])),
|
|
}
|
|
|
|
def get_field_names(self, declared_fields, info):
|
|
fields = super().get_field_names(declared_fields, info)
|
|
fields.extend(['summary', 'short_id'])
|
|
return fields
|