From 6c26344376f3978301f5aa9dd47a140d9be9332d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Herv=C3=A9=20=20MENAGER?= <herve.menager@pasteur.fr>
Date: Thu, 16 Mar 2017 22:32:48 +0100
Subject: [PATCH] add draft model code to retrieve uniprot info from Protein

---
 ippisite/ippidb/models.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/ippisite/ippidb/models.py b/ippisite/ippidb/models.py
index 541217ce..ae2a7f35 100644
--- a/ippisite/ippidb/models.py
+++ b/ippisite/ippidb/models.py
@@ -3,7 +3,7 @@ from __future__ import unicode_literals
 from django.db import models
 from django.forms import ModelForm
 
-from .ws import get_pubmed_info, get_epo_info
+from .ws import get_pubmed_info, get_epo_info, get_uniprot_info
 
 class Bibliography(models.Model):
     """
@@ -68,6 +68,12 @@ class Protein(models.Model):
     organism = models.ForeignKey('Taxonomy')
     molecular_functions = models.ManyToManyField(MolecularFunction)
 
+    def save(self, *args, **kwargs):
+        info = get_epo_info(self.id_source)
+        self.recommended_name_long = info['recommended_name']
+        self.organism = info['organism']
+        super(Protein, self).save(*args, **kwargs)
+
 class Domain(models.Model):
     pfam_acc = models.CharField('Pfam Accession', max_length=10, unique=True) 
     pfam_id = models.CharField('Pfam Family Identifier', max_length=20) 
-- 
GitLab