function.py

from rest_framework import serializers
from metagenedb.apps.catalog.models import EggNOGv5, Function, KeggOrthology

from .bulk_list import BulkListSerializer


class FunctionListSerializer(BulkListSerializer):

    class Meta:
        model = Function


class FunctionSerializer(serializers.ModelSerializer):

    class Meta:
        model = Function
        list_serializer_class = FunctionListSerializer
        fields = ('function_id', 'source', 'name')


class EggNOGListSerializer(BulkListSerializer):

    class Meta:
        model = EggNOGv5


class EggNOGSerializer(serializers.ModelSerializer):

    class Meta:
        model = EggNOGv5
        list_serializer_class = EggNOGListSerializer
        fields = ('function_id', 'name', 'functional_categories')


class KeggOrthologyListSerializer(BulkListSerializer):

    class Meta:
        model = KeggOrthology


class KeggOrthologySerializer(serializers.ModelSerializer):
    ec_number = serializers.CharField(required=False)

    class Meta:
        model = KeggOrthology
        list_serializer_class = KeggOrthologyListSerializer
        fields = ('function_id', 'name', 'long_name', 'ec_number')