def filter_representation(representation, instance, detail_filters):
     qs_type = get_queryset_type(instance)
     if SerializerDetail.ENTRY_OVERVIEW in detail_filters:
         representation["entries"] = ProteinSerializer.to_entries_count_representation(instance)
     if SerializerDetail.ENTRY_MATCH in detail_filters:
         representation["entries"] = ProteinSerializer.to_entries_overview_representation(instance, False)
     if SerializerDetail.ENTRY_DETAIL in detail_filters:
         representation["entries"] = ProteinSerializer.to_entries_overview_representation(instance, True)
     if SerializerDetail.STRUCTURE_HEADERS in detail_filters:
         representation["structures"] = ProteinSerializer.to_structures_count_representation(instance)
     if SerializerDetail.STRUCTURE_OVERVIEW in detail_filters:
         representation["structures"] = ProteinSerializer.to_structures_overview_representation(instance)
     if SerializerDetail.STRUCTURE_DETAIL in detail_filters:
         if qs_type == QuerysetType.PROTEIN:
             representation["structures"] = ProteinSerializer.to_structures_overview_representation(instance, True)
     return representation
Beispiel #2
0
    def filter_representation(representation, instance, detail_filters):
        qs_type = get_queryset_type(instance)
        if SerializerDetail.PROTEIN_OVERVIEW in detail_filters:
            # if qs_type == QuerysetType.STRUCTURE_PROTEIN:
            #     representation["proteins"] = [StructureSerializer.to_chain_representation(instance)]
            # else:
            representation["proteins"] = StructureSerializer.to_proteins_overview_representation(instance)
        if SerializerDetail.PROTEIN_DETAIL in detail_filters:
            # if qs_type == QuerysetType.STRUCTURE:
            representation["proteins"] = StructureSerializer.to_proteins_overview_representation(instance, True)
        # else:
        #     representation["proteins"] = StructureSerializer.to_proteins_detail_representation(instance)
        if SerializerDetail.ENTRY_PROTEIN_HEADERS in detail_filters:
            representation["proteins"] = StructureSerializer.to_proteins_count_representation(instance)
        if SerializerDetail.ENTRY_DETAIL in detail_filters:
            if qs_type == QuerysetType.STRUCTURE:
                representation["entries"] = StructureSerializer.to_entries_overview_representation(instance, True)
        if SerializerDetail.ENTRY_OVERVIEW in detail_filters:
            representation["entries"] = StructureSerializer.to_entries_count_representation(instance)
        if SerializerDetail.ENTRY_MATCH in detail_filters:
            representation["entries"] = StructureSerializer.to_entries_overview_representation(instance)

        return representation