diff --git a/ippisite/ippidb/templates/PDBForm.html b/ippisite/ippidb/templates/PDBForm.html index fc081b80415278b3bd22023cb32a52326cde0832..d4dd716e562f2074f160f14456bb4601f258b325 100644 --- a/ippisite/ippidb/templates/PDBForm.html +++ b/ippisite/ippidb/templates/PDBForm.html @@ -25,11 +25,13 @@ {% block custom_form %}{% endblock %} {% if wizard.form.forms %} + {{ wizard.form.management_form }} {{ formset.management_form }} {% for form in wizard.form.forms %} {{ form }} {% endfor %} {% else %} + {{ wizard.form.management_form }} {{ formset.management_form }} {% for form in formset %} {% for field in form %} @@ -41,4 +43,4 @@ <input type="submit" value="{% trans "Next step" %}"/> </form> </div> -{% endblock %} \ No newline at end of file +{% endblock %} diff --git a/ippisite/ippidb/views.py b/ippisite/ippidb/views.py index d9d2b39346e33d1b53d4b5e8c3934e16d49dc2e3..e4e17ee120c3cfea9d9ecd97aef33e5521fc8c19 100644 --- a/ippisite/ippidb/views.py +++ b/ippisite/ippidb/views.py @@ -64,14 +64,18 @@ class IppiWizard(NamedUrlSessionWizardView): form.instance.save() data['pk'] = form.instance.id if self.steps.current=='PDBForm': - uniprot_ids = get_pdb_uniprot_mapping(form.cleaned_data['pdb_id']) + pdb_ids = [form['pdb_id'] for form in form.cleaned_data] + uniprot_ids = [] protein_ids = [] + for pdb_id in pdb_ids: + uniprot_ids += get_pdb_uniprot_mapping(pdb_id) for uniprot_id in uniprot_ids: try: p = Protein.objects.get(uniprot_id=uniprot_id) except Protein.DoesNotExist: p = Protein() p.uniprot_id = uniprot_ids[0] + p.autofill() p.save() protein_ids.append(p.id) data['pks'] = protein_ids