def get_initial(self): pdb_info = get_pdb_info(self.request.session['entry_id']) initial = super(StepTwoView, self).get_initial() initial.update(pdb_info) return initial
def clean(self): file_source = int(self.cleaned_data["file_source"]) entry_id = self.cleaned_data["entry_id"] if file_source == self.FILE_REMOTE: if not get_pdb_info(entry_id): raise forms.ValidationError("PDB id does not exist in RCSB.") elif file_source == self.FILE_LOCAL: task = send_task("virus.check_file_count", args=[entry_id]) if task.get() is not 2: raise forms.ValidationError("PDB and/or CIF file not found locally.") elif file_source == self.FILE_UPLOAD: pass return self.cleaned_data