Skip to content
Snippets Groups Projects
Commit 5b84725b authored by Ben Adida's avatar Ben Adida
Browse files

fixed bulk uploading for new voters model

parent 42ba703f
No related branches found
No related tags found
No related merge requests found
...@@ -492,7 +492,7 @@ class VoterFile(models.Model): ...@@ -492,7 +492,7 @@ class VoterFile(models.Model):
last_alias_num = election.last_alias_num last_alias_num = election.last_alias_num
num_voters = 0 num_voters = 0
voter_uuids = [] new_voters = []
for voter in reader: for voter in reader:
# bad line # bad line
if len(voter) < 1: if len(voter) < 1:
...@@ -509,25 +509,24 @@ class VoterFile(models.Model): ...@@ -509,25 +509,24 @@ class VoterFile(models.Model):
if len(voter) > 2: if len(voter) > 2:
name = voter[2] name = voter[2]
# create the user # create the user -- NO MORE!
user = User.update_or_create(user_type='password', user_id=voter_id, info = {'password': heliosutils.random_string(10), 'email': email, 'name': name}) # user = User.update_or_create(user_type='password', user_id=voter_id, info = {'password': heliosutils.random_string(10), 'email': email, 'name': name})
user.save() # user.save()
# does voter for this user already exist # does voter for this user already exist
voter = Voter.get_by_election_and_user(election, user) voter = Voter.get_by_election_and_voter_id(election, voter_id)
# create the voter # create the voter
if not voter: if not voter:
voter_uuid = str(uuid.uuid4()) voter_uuid = str(uuid.uuid4())
voter = Voter(uuid= voter_uuid, voter_type = 'password', voter_id = voter_id, name = name, election = election) voter = Voter(uuid= voter_uuid, user = None, voter_login_id = voter_id, voter_name = name, election = election)
voter_uuids.append(voter_uuid) new_voters.append(voter)
voter.save() voter.save()
if election.use_voter_aliases: if election.use_voter_aliases:
voter_alias_integers = range(last_alias_num+1, last_alias_num+1+num_voters) voter_alias_integers = range(last_alias_num+1, last_alias_num+1+num_voters)
random.shuffle(voter_alias_integers) random.shuffle(voter_alias_integers)
for i, voter_uuid in enumerate(voter_uuids): for i, voter in enumerate(new_voters):
voter = Voter.get_by_election_and_uuid(election, voter_uuid)
voter.alias = 'V%s' % voter_alias_integers[i] voter.alias = 'V%s' % voter_alias_integers[i]
voter.save() voter.save()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment