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
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