jumpserver/apps/assets/serializers/asset/category.py

from rest_framework import serializers

from assets.models import DeviceInfo, Host, Database, Networking, Cloud
from .common import AssetSerializer

__all__ = [
    'DeviceSerializer', 'HostSerializer', 'DatabaseSerializer',
    'NetworkingSerializer', 'CloudSerializer',
]


class DeviceSerializer(serializers.ModelSerializer):
    class Meta:
        model = DeviceInfo
        fields = [
            'id', 'vendor', 'model', 'sn', 'cpu_model', 'cpu_count',
            'cpu_cores', 'cpu_vcpus', 'memory', 'disk_total', 'disk_info',
            'os', 'os_version', 'os_arch', 'hostname_raw', 'number',
            'cpu_info', 'hardware_info', 'date_updated'
        ]


class HostSerializer(AssetSerializer):
    device_info = DeviceSerializer(read_only=True, allow_null=True)

    class Meta(AssetSerializer.Meta):
        model = Host
        fields = AssetSerializer.Meta.fields + ['device_info']


class DatabaseSerializer(AssetSerializer):
    class Meta(AssetSerializer.Meta):
        model = Database
        fields = AssetSerializer.Meta.fields + ['db_name']


class NetworkingSerializer(AssetSerializer):
    class Meta(AssetSerializer.Meta):
        model = Networking


class CloudSerializer(AssetSerializer):
    class Meta(AssetSerializer.Meta):
        model = Cloud
        fields = AssetSerializer.Meta.fields + ['cluster']