From cf9c2ee3ebcf72d6aab8f3e8c3c0e404e1624e88 Mon Sep 17 00:00:00 2001
From: Rachel TORCHET <racheltorchet@users.noreply.github.com>
Date: Mon, 18 Sep 2017 16:25:33 +0200
Subject: [PATCH] save data to pass between ProteinDomainComplex(Type)Form in
wizard
Former-commit-id: f021cb531522fc7aa44570beefaaa81d906271a3
---
ippisite/ippidb/views.py | 7 ++++++-
1 file changed, 6 insertions(+), 1 deletion(-)
diff --git a/ippisite/ippidb/views.py b/ippisite/ippidb/views.py
index ddc73104..32d5c6ce 100644
--- a/ippisite/ippidb/views.py
+++ b/ippisite/ippidb/views.py
@@ -5,7 +5,7 @@ from django.http import HttpResponseRedirect
from formtools.wizard.views import SessionWizardView,NamedUrlSessionWizardView
from .forms import IdForm, BibliographyForm, PDBForm, ProteinForm, ProteinDomainComplexTypeForm, ProteinDomainComplexForm, PpiForm, ProteinFormSet, PDBFormSet
-from .models import Protein, Bibliography
+from .models import Protein, Bibliography, ProteinDomainComplex
from .ws import get_pdb_uniprot_mapping
def index(request):
@@ -68,6 +68,9 @@ class IppiWizard(NamedUrlSessionWizardView):
if self.steps.current=='ProteinForm':
pks = self.storage.get_step_data('PDBForm').get('pks')
return Protein.objects.filter(id__in=pks)
+ if self.steps.current=='ProteinDomainComplexForm':
+ complexType = self.storage.get_step_data('ProteinDomainComplexTypeForm').get('complexType')
+ return ProteinDomainComplex.objects.get(complexType=complexType)
def process_step(self, form):
"""
@@ -97,6 +100,8 @@ class IppiWizard(NamedUrlSessionWizardView):
p.save()
protein_ids.append(p.id)
data['pks'] = protein_ids
+ if self.steps.current=='ProteinDomainComplexTypeForm':
+ data['complex_type']=form['complexType']
return data
def done(self, form_list, **kwargs):
--
GitLab