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')
