def ballot_item_list_edit_process_view(request):
"""
Process the new or edit ballot form
:param request:
:return:
"""
authority_required = {'verified_volunteer'} # admin, verified_volunteer
if not voter_has_authority(request, authority_required):
return redirect_to_sign_in_page(request, authority_required)
ballot_returned_id = convert_to_int(
request.POST.get('ballot_returned_id', 0))
google_civic_election_id = request.POST.get('google_civic_election_id', 0)
polling_location_id = convert_to_int(
request.POST.get('polling_location_id', 0))
polling_location_city = request.POST.get('polling_location_city', '')
polling_location_zip = request.POST.get('polling_location_zip', '')
contest_office1_id = request.POST.get('contest_office1_id', 0)
contest_office1_order = request.POST.get('contest_office1_order', 0)
contest_measure1_id = request.POST.get('contest_measure1_id', 0)
election_local_id = 0
# Find existing ballot_returned
ballot_returned_found = False
ballot_returned = BallotReturned()
if positive_value_exists(ballot_returned_id):
try:
ballot_returned_query = BallotReturned.objects.filter(
id=ballot_returned_id)
if len(ballot_returned_query):
ballot_returned = ballot_returned_query[0]
ballot_returned_found = True
except Exception as e:
pass
election_manager = ElectionManager()
polling_location_manager = PollingLocationManager()
polling_location = PollingLocation()
polling_location_found = False
try:
if ballot_returned_found:
# Update
# Check to see if this is a We Vote-created election
is_we_vote_google_civic_election_id = True \
if convert_to_int(ballot_returned.google_civic_election_id) >= 1000000 \
else False
results = election_manager.retrieve_election(
ballot_returned.google_civic_election_id)
if results['election_found']:
election = results['election']
election_local_id = election.id
# polling_location must be found
# We cannot change a polling location once saved, so we ignore the incoming polling_location_id here
results = polling_location_manager.retrieve_polling_location_by_id(
0, ballot_returned.polling_location_we_vote_id)
if results['polling_location_found']:
polling_location = results['polling_location']
polling_location_found = True
else:
# Create new ballot_returned entry
# election must be found
election_results = election_manager.retrieve_election(
google_civic_election_id)
if election_results['election_found']:
election = election_results['election']
election_local_id = election.id
state_code = election.get_election_state()
else:
messages.add_message(
request, messages.ERROR, 'Could not find election -- '
'required to save ballot_returned.')
return HttpResponseRedirect(
reverse('ballot:ballot_item_list_edit',
args=(ballot_returned_id, )) +
"?google_civic_election_id=" +
str(google_civic_election_id) + "&polling_location_id=" +
str(polling_location_id) + "&polling_location_city=" +
polling_location_city + "&polling_location_zip=" +
str(polling_location_zip))
# polling_location must be found
if positive_value_exists(polling_location_id):
results = polling_location_manager.retrieve_polling_location_by_id(
polling_location_id)
if results['polling_location_found']:
polling_location = results['polling_location']
polling_location_found = True
if not polling_location_found:
messages.add_message(
request, messages.ERROR,
'Could not find polling_location -- '
'required to save ballot_returned.')
return HttpResponseRedirect(
reverse('ballot:ballot_item_list_edit',
args=(ballot_returned_id, )) +
"?google_civic_election_id=" +
str(google_civic_election_id) + "&polling_location_id=" +
str(polling_location_id) + "&polling_location_city=" +
polling_location_city + "&polling_location_zip=" +
str(polling_location_zip))
ballot_returned = BallotReturned(
election_date=election.election_day_text,
election_description_text=election.election_name,
google_civic_election_id=google_civic_election_id,
polling_location_we_vote_id=polling_location.we_vote_id,
normalized_city=polling_location.city,
normalized_line1=polling_location.line1,
normalized_line2=polling_location.line2,
normalized_state=polling_location.state,
normalized_zip=polling_location.get_formatted_zip(),
text_for_map_search=polling_location.get_text_for_map_search(),
)
ballot_returned.save()
ballot_returned_id = ballot_returned.id
ballot_returned_found = True
messages.add_message(request, messages.INFO,
'New ballot_returned saved.')
# #######################################
# Make sure we have saved a latitude and longitude for the ballot_returned entry
if ballot_returned_found and positive_value_exists(
ballot_returned.text_for_map_search):
if not ballot_returned.latitude or not ballot_returned.longitude:
google_client = get_geocoder_for_service('google')()
location = google_client.geocode(
ballot_returned.text_for_map_search)
if location is None:
status = 'Could not find location matching "{}"'.format(
ballot_returned.text_for_map_search)
else:
ballot_returned.latitude = location.latitude
ballot_returned.longitude = location.longitude
ballot_returned.save()
# #######################################
# Now create new ballot_item entries
# Contest Office 1
ballot_item_manager = BallotItemManager()
contest_office_manager = ContestOfficeManager()
results = contest_office_manager.retrieve_contest_office(
contest_office1_id)
if results['contest_office_found']:
contest_office = results['contest_office']
ballot_item_display_name = contest_office.office_name
google_ballot_placement = 0
measure_subtitle = ''
local_ballot_order = contest_office1_order if positive_value_exists(
contest_office1_order) else 0
results = ballot_item_manager.update_or_create_ballot_item_for_polling_location(
polling_location.we_vote_id, google_civic_election_id,
google_ballot_placement, ballot_item_display_name,
measure_subtitle, local_ballot_order, contest_office.id,
contest_office.we_vote_id)
if results['new_ballot_item_created']:
messages.add_message(request, messages.INFO, 'Office 1 added.')
else:
messages.add_message(request, messages.ERROR,
'Office 1 could not be added.')
# Contest Measure 1
ballot_item_manager = BallotItemManager()
contest_measure_manager = ContestMeasureManager()
results = contest_measure_manager.retrieve_contest_measure(
contest_measure1_id)
if results['contest_measure_found']:
contest_measure = results['contest_measure']
google_ballot_placement = 0
ballot_item_display_name = contest_measure.measure_title
contest_office_id = 0
contest_office_we_vote_id = ''
local_ballot_order = 0
ballot_item_manager.update_or_create_ballot_item_for_polling_location(
polling_location.we_vote_id, google_civic_election_id,
google_ballot_placement, ballot_item_display_name,
contest_measure.measure_subtitle, local_ballot_order,
contest_office_id, contest_office_we_vote_id,
contest_measure.id)
except Exception as e:
messages.add_message(request, messages.ERROR,
'Could not save ballot_returned.')
return HttpResponseRedirect(
reverse('ballot:ballot_item_list_edit', args=(ballot_returned_id, )) +
"?google_civic_election_id=" + str(google_civic_election_id) +
"&polling_location_id=" + str(polling_location_id) +
"&polling_location_city=" + polling_location_city +
"&polling_location_zip=" + str(polling_location_zip))
def ballot_item_list_edit_process_view(request):
"""
Process the new or edit ballot form
:param request:
:return:
"""
authority_required = {'verified_volunteer'} # admin, verified_volunteer
if not voter_has_authority(request, authority_required):
return redirect_to_sign_in_page(request, authority_required)
ballot_returned_id = convert_to_int(request.POST.get('ballot_returned_id', 0))
google_civic_election_id = request.POST.get('google_civic_election_id', 0)
polling_location_id = convert_to_int(request.POST.get('polling_location_id', 0))
polling_location_city = request.POST.get('polling_location_city', '')
polling_location_zip = request.POST.get('polling_location_zip', '')
contest_office1_id = request.POST.get('contest_office1_id', 0)
contest_office1_order = request.POST.get('contest_office1_order', 0)
contest_measure1_id = request.POST.get('contest_measure1_id', 0)
election_local_id = 0
# Find existing ballot_returned
ballot_returned_found = False
ballot_returned = BallotReturned()
if positive_value_exists(ballot_returned_id):
try:
ballot_returned_query = BallotReturned.objects.filter(id=ballot_returned_id)
if len(ballot_returned_query):
ballot_returned = ballot_returned_query[0]
ballot_returned_found = True
except Exception as e:
pass
election_manager = ElectionManager()
polling_location_manager = PollingLocationManager()
polling_location = PollingLocation()
polling_location_found = False
try:
if ballot_returned_found:
# Update
# Check to see if this is a We Vote-created election
is_we_vote_google_civic_election_id = True \
if convert_to_int(ballot_returned.google_civic_election_id) >= 1000000 \
else False
results = election_manager.retrieve_election(ballot_returned.google_civic_election_id)
if results['election_found']:
election = results['election']
election_local_id = election.id
# polling_location must be found
# We cannot change a polling location once saved, so we ignore the incoming polling_location_id here
results = polling_location_manager.retrieve_polling_location_by_id(
0, ballot_returned.polling_location_we_vote_id)
if results['polling_location_found']:
polling_location = results['polling_location']
polling_location_found = True
else:
# Create new ballot_returned entry
# election must be found
election_results = election_manager.retrieve_election(google_civic_election_id)
if election_results['election_found']:
election = election_results['election']
election_local_id = election.id
state_code = election.get_election_state()
else:
messages.add_message(request, messages.ERROR, 'Could not find election -- '
'required to save ballot_returned.')
return HttpResponseRedirect(reverse('ballot:ballot_item_list_edit', args=(ballot_returned_id,)) +
"?google_civic_election_id=" + str(google_civic_election_id) +
"&polling_location_id=" + str(polling_location_id) +
"&polling_location_city=" + polling_location_city +
"&polling_location_zip=" + str(polling_location_zip)
)
# polling_location must be found
if positive_value_exists(polling_location_id):
results = polling_location_manager.retrieve_polling_location_by_id(polling_location_id)
if results['polling_location_found']:
polling_location = results['polling_location']
polling_location_found = True
if not polling_location_found:
messages.add_message(request, messages.ERROR, 'Could not find polling_location -- '
'required to save ballot_returned.')
return HttpResponseRedirect(reverse('ballot:ballot_item_list_edit', args=(ballot_returned_id,)) +
"?google_civic_election_id=" + str(google_civic_election_id) +
"&polling_location_id=" + str(polling_location_id) +
"&polling_location_city=" + polling_location_city +
"&polling_location_zip=" + str(polling_location_zip)
)
ballot_returned = BallotReturned(
election_date=election.election_day_text,
election_description_text=election.election_name,
google_civic_election_id=google_civic_election_id,
polling_location_we_vote_id=polling_location.we_vote_id,
normalized_city=polling_location.city,
normalized_line1=polling_location.line1,
normalized_line2=polling_location.line2,
normalized_state=polling_location.state,
normalized_zip=polling_location.get_formatted_zip(),
text_for_map_search=polling_location.get_text_for_map_search(),
)
ballot_returned.save()
ballot_returned_id = ballot_returned.id
ballot_returned_found = True
messages.add_message(request, messages.INFO, 'New ballot_returned saved.')
# #######################################
# Make sure we have saved a latitude and longitude for the ballot_returned entry
if ballot_returned_found and positive_value_exists(ballot_returned.text_for_map_search):
if not ballot_returned.latitude or not ballot_returned.longitude:
google_client = get_geocoder_for_service('google')()
location = google_client.geocode(ballot_returned.text_for_map_search)
if location is None:
status = 'Could not find location matching "{}"'.format(ballot_returned.text_for_map_search)
else:
ballot_returned.latitude = location.latitude
ballot_returned.longitude = location.longitude
ballot_returned.save()
# #######################################
# Now create new ballot_item entries
# Contest Office 1
ballot_item_manager = BallotItemManager()
contest_office_manager = ContestOfficeManager()
results = contest_office_manager.retrieve_contest_office(contest_office1_id)
if results['contest_office_found']:
contest_office = results['contest_office']
ballot_item_display_name = contest_office.office_name
google_ballot_placement = 0
measure_subtitle = ''
local_ballot_order = contest_office1_order if positive_value_exists(contest_office1_order) else 0
results = ballot_item_manager.update_or_create_ballot_item_for_polling_location(
polling_location.we_vote_id, google_civic_election_id, google_ballot_placement,
ballot_item_display_name, measure_subtitle, local_ballot_order,
contest_office.id, contest_office.we_vote_id)
if results['new_ballot_item_created']:
messages.add_message(request, messages.INFO, 'Office 1 added.')
else:
messages.add_message(request, messages.ERROR, 'Office 1 could not be added.')
# Contest Measure 1
ballot_item_manager = BallotItemManager()
contest_measure_manager = ContestMeasureManager()
results = contest_measure_manager.retrieve_contest_measure(contest_measure1_id)
if results['contest_measure_found']:
contest_measure = results['contest_measure']
google_ballot_placement = 0
ballot_item_display_name = contest_measure.measure_title
contest_office_id = 0
contest_office_we_vote_id = ''
local_ballot_order = 0
ballot_item_manager.update_or_create_ballot_item_for_polling_location(
polling_location.we_vote_id, google_civic_election_id, google_ballot_placement,
ballot_item_display_name, contest_measure.measure_subtitle, local_ballot_order,
contest_office_id, contest_office_we_vote_id,
contest_measure.id)
except Exception as e:
messages.add_message(request, messages.ERROR, 'Could not save ballot_returned.')
return HttpResponseRedirect(reverse('ballot:ballot_item_list_edit', args=(ballot_returned_id,)) +
"?google_civic_election_id=" + str(google_civic_election_id) +
"&polling_location_id=" + str(polling_location_id) +
"&polling_location_city=" + polling_location_city +
"&polling_location_zip=" + str(polling_location_zip)
)
