diff --git a/drugstone/models.py b/drugstone/models.py index 5decf3d846c119b59c6099f17552bb932118ef60..f7971115d0eba859c2461ddb5d4d397bce0cbfab 100755 --- a/drugstone/models.py +++ b/drugstone/models.py @@ -317,10 +317,14 @@ class ProteinProteinInteraction(models.Model): class ProteinDrugInteraction(models.Model): id = models.BigAutoField(primary_key=True) pdi_dataset = models.ForeignKey( - PDIDataset, null=True, on_delete=models.CASCADE, related_name='pdi_dataset_relation') - protein = models.ForeignKey('Protein', on_delete=models.CASCADE) - drug = models.ForeignKey('Drug', on_delete=models.CASCADE) - actions = models.CharField(max_length=255, default='[]') + PDIDataset, + null=True, + on_delete=models.CASCADE, + related_name="pdi_dataset_relation", + ) + protein = models.ForeignKey("Protein", on_delete=models.CASCADE) + drug = models.ForeignKey("Drug", on_delete=models.CASCADE) + actions = models.CharField(max_length=255, default="[]") class Meta: unique_together = ("pdi_dataset", "protein", "drug") @@ -329,7 +333,12 @@ class ProteinDrugInteraction(models.Model): return f"{self.pdi_dataset}-{self.protein}-{self.drug}" def __eq__(self, other): - return self.pdi_dataset_id == other.pdi_dataset_id and self.protein_id == other.protein_id and self.drug_id == other.drug_id and self.actions == other.actions + return ( + self.pdi_dataset_id == other.pdi_dataset_id + and self.protein_id == other.protein_id + and self.drug_id == other.drug_id + and self.actions == other.actions + ) def __ne__(self, other): return not self.__eq__(other) diff --git a/drugstone/serializers.py b/drugstone/serializers.py index c4b820c3c3b2473f1545d57cae509d1689569491..79080f8d263638bc041d3039106de546bc9c6687 100755 --- a/drugstone/serializers.py +++ b/drugstone/serializers.py @@ -2,20 +2,31 @@ import json from rest_framework import serializers from drugstone import models -from drugstone.models import Protein, Task, Drug, ProteinDrugInteraction, \ - Tissue, ProteinProteinInteraction, Network, ProteinDisorderAssociation, Disorder, DrugDisorderIndication +from drugstone.models import ( + Protein, + Task, + Drug, + ProteinDrugInteraction, + Tissue, + ProteinProteinInteraction, + Network, + ProteinDisorderAssociation, + Disorder, + DrugDisorderIndication, +) class PDIDatasetSerializer(serializers.ModelSerializer): class Meta: model = models.PDIDataset - fields = '__all__' + fields = "__all__" class PPIDatasetSerializer(serializers.ModelSerializer): class Meta: model = models.PPIDataset - fields = '__all__' + fields = "__all__" + class PDisDatasetSerializer(serializers.ModelSerializer): class Meta: @@ -176,7 +187,7 @@ class ProteinDrugInteractionSerializer(serializers.ModelSerializer): class Meta: model = ProteinDrugInteraction - fields = ['dataset', 'protein', 'drug', 'actions'] + fields = ["dataset", "protein", "drug", "actions"] class ProteinDisorderAssociationSerializer(serializers.ModelSerializer):