def _update_new_agency(self, request, foia):
"""Update the new agency"""
form = AgencyForm(request.POST, instance=foia.agency)
if foia.has_perm(request.user, 'change'):
if form.is_valid():
form.save()
messages.success(request, 'Agency info saved. Thanks for your help!')
else:
messages.success(request, 'The data was invalid! Try again.')
else:
messages.error(request, 'You cannot do that, stop it.')
return redirect(foia)
def update_new_agency(request, foia):
"""Update the new agency"""
form = AgencyForm(
request.POST, instance=foia.agency, prefix=request.POST.get("task", "")
)
if foia.has_perm(request.user, "change"):
if form.is_valid():
form.save()
messages.success(request, "Agency info saved. Thanks for your help!")
else:
messages.error(request, "The data was invalid! Try again.")
else:
messages.error(request, "You cannot do that, stop it.")
return _get_redirect(request, foia)
def task_post_helper(self, request, task):
"""Special post handlers exclusive to NewAgencyTasks"""
if request.POST.get('approve'):
new_agency_form = AgencyForm(request.POST, instance=task.agency)
if new_agency_form.is_valid():
new_agency_form.save()
else:
messages.error(request, 'The agency info form is invalid.')
return
task.approve()
task.resolve(request.user)
if request.POST.get('reject'):
replacement_agency_id = request.POST.get('replacement')
replacement_agency = get_object_or_404(Agency,
id=replacement_agency_id)
task.reject(replacement_agency)
task.resolve(request.user)
return super(NewAgencyTaskList, self).task_post_helper(request, task)
def task_post_helper(self, request, task, form_data=None):
"""Special post handlers exclusive to NewAgencyTasks"""
if request.POST.get("approve"):
new_agency_form = AgencyForm(request.POST,
instance=task.agency,
prefix=request.POST.get("task", ""))
if new_agency_form.is_valid():
new_agency_form.save()
else:
raise ValueError("The agency info form is invalid.")
task.approve()
form_data = new_agency_form.cleaned_data
# phone numbers must be strings not phone number objects to serialize
if form_data.get("phone"):
form_data["phone"] = str(form_data["phone"])
if form_data.get("fax"):
form_data["fax"] = str(form_data["fax"])
if form_data.get("jurisdiction"):
form_data["jurisdiction"] = form_data["jurisdiction"].pk
form_data.update({"approve": True})
task.resolve(request.user, form_data)
elif request.POST.get("replace"):
form = ReplaceNewAgencyForm(request.POST,
prefix=request.POST.get("task", ""))
if form.is_valid():
replace_agency = form.cleaned_data["replace_agency"]
task.reject(replace_agency)
form_data = {
"replace": True,
"replace_agency": replace_agency.pk
}
task.resolve(request.user, form_data)
else:
messages.error(request, "Bad form data")
return None
elif request.POST.get("reject"):
task.reject()
form_data = {"reject": True}
task.resolve(request.user, form_data)
elif request.POST.get("spam"):
task.spam(request.user)
form_data = {"spam": True}
task.resolve(request.user, form_data)
return super(NewAgencyTaskList, self).task_post_helper(request, task)
def task_post_helper(self, request, task, form_data=None):
"""Special post handlers exclusive to NewAgencyTasks"""
if request.POST.get('approve'):
new_agency_form = AgencyForm(request.POST, instance=task.agency)
if new_agency_form.is_valid():
new_agency_form.save()
else:
raise ValueError('The agency info form is invalid.')
task.approve()
form_data = new_agency_form.cleaned_data
# phone numbers must be strings not phone number objects to serialize
if form_data.get('phone'):
form_data['phone'] = unicode(form_data['phone'])
if form_data.get('fax'):
form_data['fax'] = unicode(form_data['fax'])
if form_data.get('jurisdiction'):
form_data['jurisdiction'] = form_data['jurisdiction'].pk
form_data.update({'approve': True})
task.resolve(request.user, form_data)
elif request.POST.get('replace'):
form = ReplaceNewAgencyForm(request.POST)
if form.is_valid():
replace_agency = form.cleaned_data['replace_agency']
task.reject(replace_agency)
form_data = {
'replace': True,
'replace_agency': replace_agency.pk
}
task.resolve(request.user, form_data)
else:
messages.error(request, 'Bad form data')
return
elif request.POST.get('reject'):
task.reject()
form_data = {'reject': True}
task.resolve(request.user, form_data)
elif request.POST.get('spam'):
task.spam()
form_data = {'spam': True}
task.resolve(request.user, form_data)
return super(NewAgencyTaskList, self).task_post_helper(request, task)
