def __init__(self, *args, **kwargs): node = kwargs.pop('node') super(CreateMergeForm, self).__init__(*args, **kwargs) self.fields['free_atoms'].queryset = ConceptAtom.get_unmerged_atoms(node) self.fields['merged_atoms'].queryset = ConceptAtom.get_final_atoms(node)
def __init__(self, *args, **kwargs): node = kwargs.pop('node') super(CreateMergeForm, self).__init__(*args, **kwargs) self.fields['free_atoms'].queryset = ConceptAtom.get_unmerged_atoms( node) self.fields['merged_atoms'].queryset = ConceptAtom.get_final_atoms( node)
def finalize_merge(request, node_id): """ Once the admin has okayed the merged state of the concept atoms, this function updates the database accordingly """ node = get_object_or_404(ConceptNode,pk=node_id) for atom in ConceptAtom.get_unmerged_atoms(node): atom.final_choice = True atom.save() #return redirect(reverse('final sub', args=[node_id])) return HttpResponse('fix finalize_merge')
def merge(request, node, context): """ This function gets all the concepts for a specified node and renders a form that allows an admin to merge/unmerge concept atoms """ # Hack to disable the "admin" dropdown, since it makes no sense on the merge page. context['staff'] = False # Only if we're in the merging stage. if node.node_type != 'M': return render(request, 'nodemanager/stage_error.html') #populate the form with all the existing concept atoms for that node create_form = CreateMergeForm(node=node) edit_formset = UpdateMergeFormSet(initial=[{'pk': atom.pk} for atom in ConceptAtom.get_final_atoms(node)]) context['create_form'] = create_form context['edit_formset'] = edit_formset return render(request, 'nodemanager/node_merge.html', context)
def submit_entry(request, node_id): """ This function processes a user-submitted form containing their brainstormed picks. It handles users deleting/modifying pre-entered picks, as well as entering new ones. """ auth = ContributorProfile.auth_status(request.user) if not auth: raise PermissionDenied if request.method == 'POST': node = get_object_or_404(ConceptNode,pk=node_id) AtomFormSet = formset_factory(AtomForm, can_delete=True, max_num=node.max_children, extra=node.max_children, validate_max=True) formset = AtomFormSet(request.POST) if formset.is_valid(): #We iterate through each form in the formset, which #represents a different concept entered. If there is data #in the form, the user has either edited an existing #concept or entered a new one. for form in formset: form_text = form.cleaned_data.get('text') pk = form.cleaned_data.get('pk') if form_text: if not pk: new_atom = ConceptAtom( concept_node=node, user=request.user, text=form_text, final_choice=False ) new_atom.save() continue #if there is an associated pk, get the model instance atom = ConceptAtom.objects.filter(pk=pk).get() if not form_text == atom.text: #update if necessary atom.text = form_text atom.save() if form in formset.deleted_forms: #or delete it atom.delete() return HttpResponse('{"success": true}') else: reason = '' for (i, form) in enumerate(formset.errors): for field, problems in form.iteritems(): reason += 'Entry %d, %s:\\n' for problem in problems: reason += ' %s\\n' % problem return HttpResponse('{"success": false, "reason": "%s"}' % reason) else: return redirect('stage1 dispatch')
def get_merge(request, node_id, merge_type=None): """ This function handles a submitted form containing information about merging concepts. """ auth = ContributorProfile.auth_status(request.user) if auth != 'staff': raise PermissionDenied if request.method == 'POST': node = get_object_or_404(ConceptNode,pk=node_id) #differentiate between different forms if merge_type == 'add merge': form = CreateMergeForm(request.POST, node=node) elif merge_type == 'subtract merge': form = UpdateMergeFormSet(request.POST) else: #should never get here print "ERROR No Merge type was called" # note that "form" could be either a single form or a formset # depending on whether the user did a 'add merge' or an # 'update merge'. For this reason, we have to do an additional # attribute check if form.is_valid(): #could be a form or a formset depending on request # user merged under a new concept atom if hasattr(form,'add_merge_id') and form.add_merge_id in request.POST: new_atom = ConceptAtom(concept_node=node, user=request.user, text=form.cleaned_data.get('new_atom_name'), final_choice=True,) new_atom.save() new_atom.add_merge_atoms(form.cleaned_data.get('free_atoms')) #user merged atoms to an existing concept atom elif hasattr(form,'subtract_merge_id') and form.subtract_merge_id in request.POST: curr_atom = form.cleaned_data.get('merged_atoms') curr_atom.add_merge_atoms(form.cleaned_data.get('free_atoms')) #user has edited an existing concept atom else: formset = form #small name-change to enhance clarity for form in formset: if form.cleaned_data.get('delete'): atom = ConceptAtom.objects.filter(pk=form.cleaned_data.get('pk')).get() atom.delete() else: for atom in form.cleaned_data.get('choices'): atom.merged_atoms = None atom.save() return redirect('merge', node_id=node.id) #the form is invalid, and we redirect the errors to the user else: render_args = {'node': node, 'user': request.user} #need to set arguments differently based on whether it was #an add merge or subtract merge, since this chooses #between a form or formset if merge_type == 'add merge': render_args['create_form'] = form render_args['edit_formset'] = UpdateMergeFormSet(initial=[{'pk': atom.pk} for atom in ConceptAtom.get_final_atoms(node)]) elif merge_type == 'subtract merge': render_args['create_form'] = CreateMergeForm(node) render_args['edit_formset'] = form else: #again, should never be here print "ERROR unknown merge type" return render(request, 'nodemanager/merge.html', context) else: return redirect('stage1 dispatch')