Source code for api_vulnerabilities.serializers

from rest_framework import serializers

from .models import CWEModel, TechnicalImpactModel


class ProductSerializer(serializers.Serializer):
    vendor = serializers.CharField()
    name = serializers.CharField()
    version = serializers.CharField()
    system = serializers.CharField()


class CveWrapperSerializer(serializers.Serializer):
    cve = serializers.CharField()
    cvss_vector = serializers.CharField()
    complexity = serializers.CharField()
    authentication = serializers.CharField()
    vector = serializers.CharField()
    cvss = serializers.CharField()
    cwe = serializers.CharField()
    title = serializers.CharField()
    availability = serializers.CharField()
    confidentiality = serializers.CharField()
    products = ProductSerializer(many=True)


class CWEModelSerializer(serializers.HyperlinkedModelSerializer):
    """Serializowanie obiektu z bazy danych"""

    class Meta:
        model = CWEModel
        fields = ["cwe_id", "title", "description", "likehood"]