From b3781892db30365559f2a3689bdb3bbc8ef165ad Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Herv=C3=A9=20=20MENAGER?= <herve.menager@pasteur.fr>
Date: Tue, 14 Mar 2017 21:34:12 +0100
Subject: [PATCH] retrieve patent info in Bibliography.save() when relevant
---
ippisite/ippidb/models.py | 15 +++++++++------
1 file changed, 9 insertions(+), 6 deletions(-)
diff --git a/ippisite/ippidb/models.py b/ippisite/ippidb/models.py
index 8ab7fe5e..2c3ef422 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
+from .ws import get_pubmed_info, get_epo_info
class Bibliography(models.Model):
"""
@@ -31,11 +31,14 @@ class Bibliography(models.Model):
return reverse('bibliography-detail', args=[str(self.id)])
def save(self, *args, **kwargs):
- pubmed_info = get_pubmed_info(self.id_source)
- self.title = pubmed_info['title']
- self.journal_name = pubmed_info['journal_name']
- self.authors_list = pubmed_info['authors_list']
- self.biblio_year = pubmed_info['biblio_year']
+ if self.source == 'PM':
+ info = get_pubmed_info(self.id_source)
+ else:
+ info = get_epo_info(self.id_source)
+ self.title = info['title']
+ self.journal_name = info['journal_name']
+ self.authors_list = info['authors_list']
+ self.biblio_year = info['biblio_year']
super(Bibliography, self).save(*args, **kwargs)
class Meta:
--
GitLab