def create(body):
request_body = RequestBody(body)
result = Model.create(
electionId=request_body.get("electionId"),
issuingOfficeId=request_body.get("issuingOfficeId"),
receivingOfficeId=request_body.get("receivingOfficeId"),
issuedTo=request_body.get("issuedTo"),
)
return Schema().dump(result).data
def get_all(electionId=None,
issuingOfficeId=None,
receivingOfficeId=None,
issuedBy=None,
issuedTo=None):
result = Model.get_all(electionId=electionId,
issuingOfficeId=issuingOfficeId,
receivingOfficeId=receivingOfficeId,
issuedBy=issuedBy,
issuedTo=issuedTo)
return Schema(many=True).dump(result).data
def delete(invoiceId, stationaryItemId):
if Invoice.has_confirmed(invoiceId):
raise ForbiddenException("Stationary items cannot be deleted from confirmed invoices (%d)" % invoiceId)
else:
result = Model.query.filter(
Model.invoiceId == invoiceId,
Model.stationaryItemId == stationaryItemId
).delete()
db.session.flush()
return result
def __init__(self, invoiceId, stationaryItemId):
if Invoice.has_confirmed(invoiceId):
raise ForbiddenException("Stationary items cannot be added to confirmed invoices (%d)" % invoiceId)
elif StationaryItem.is_locked(stationaryItemId):
raise ForbiddenException("Stationary item is not available (%d)" % stationaryItemId)
else:
received_proof = Proof.create(
proofType=ProofTypeEnum.InvoiceStationaryItemReceive
)
super(InvoiceStationaryItemModel, self).__init__(
invoiceId=invoiceId,
stationaryItemId=stationaryItemId,
receivedProofId=received_proof.proofId
)
def create(invoiceId, stationaryItemId):
if Invoice.has_confirmed(invoiceId):
raise ForbiddenException("Stationary items cannot be added to confirmed invoices (%d)" % invoiceId)
elif StationaryItem.is_locked(stationaryItemId):
raise ForbiddenException("Stationary item is not available (%d)" % stationaryItemId)
else:
received_proof = Proof.create(
proofType=ProofTypeEnum.InvoiceStationaryItemReceive
)
print("######################### received_proof ###", received_proof)
print("######################### received_proof.proofId ###", received_proof.proofId)
result = Model(
invoiceId=invoiceId,
stationaryItemId=stationaryItemId,
receivedProofId=received_proof.proofId
)
db.session.add(result)
db.session.commit()
return result
def build_presidential_election(root_election: Election,
party_candidate_dataset_file=None,
polling_station_dataset_file=None,
postal_counting_centers_dataset_file=None,
invalid_vote_categories_dataset_file=None):
postal_election = root_election.add_sub_election(
electionName="Postal", voteType=VoteTypeEnum.Postal)
ordinary_election = root_election.add_sub_election(
electionName="Ordinary", voteType=VoteTypeEnum.NonPostal)
if not party_candidate_dataset_file:
party_candidate_dataset_file = root_election.partyCandidateDataset.get_file_path(
)
if not polling_station_dataset_file:
polling_station_dataset_file = root_election.pollingStationsDataset.get_file_path(
)
if not postal_counting_centers_dataset_file:
postal_counting_centers_dataset_file = root_election.postalCountingCentresDataset.get_file_path(
)
if not invalid_vote_categories_dataset_file:
invalid_vote_categories_dataset_file = root_election.invalidVoteCategoriesDataset.get_file_path(
)
data_stores = {}
def get_data_store(data_store_key):
if data_store_key not in data_stores:
data_stores[data_store_key] = {}
return data_stores[data_store_key]
def get_object_from_data_store(data_key, data_store_key):
data_store = get_data_store(data_store_key)
if data_key in data_store:
return data_store[data_key]
else:
return None
def set_object_to_data_store(data_key, data_store_key, obj):
data_store = get_data_store(data_store_key)
data_store[data_key] = obj
def get_object(election, row, row_key, data_key=None):
if data_key is None:
cell = row[row_key].strip()
data_key = cell
else:
cell = row[data_key].strip()
data_key = cell
data_store_key = row_key
if data_store_key == "TallySheet":
data_key = "%s-%s" % (row["TallySheet"], row["Counting Centre"])
elif data_store_key == "Polling District":
data_key = "%s-%s" % (row["Polling Division"],
row["Polling District"])
obj = get_object_from_data_store(data_key, data_store_key)
if obj is None:
if data_store_key == "Ballot":
obj = Ballot.create(ballotId=cell,
electionId=election.electionId)
elif data_store_key == "Tendered Ballot":
obj = Ballot.create(ballotId=cell,
electionId=election.electionId,
ballotType=BallotTypeEnum.Tendered)
elif data_store_key == "Ballot Box":
obj = BallotBox.create(ballotBoxId=cell,
electionId=election.electionId)
elif data_store_key == "Party":
obj = Party.create(partyName=cell,
partySymbol=row["Party Symbol"],
partyAbbreviation=row["Party Abbreviation"])
elif data_store_key == "Candidate":
obj = Candidate.create(candidateName=cell)
elif data_store_key == "Country":
obj = Country.create(cell, electionId=election.electionId)
TallySheet.create(
tallySheetCode=TallySheetCodeEnum.PRE_ALL_ISLAND_RESULTS,
electionId=election.electionId,
areaId=obj.areaId)
TallySheet.create(
tallySheetCode=TallySheetCodeEnum.
PRE_ALL_ISLAND_RESULTS_BY_ELECTORAL_DISTRICTS,
electionId=election.electionId,
areaId=obj.areaId)
elif data_store_key == "Electoral District":
obj = ElectoralDistrict.create(cell,
electionId=election.electionId)
TallySheet.create(tallySheetCode=TallySheetCodeEnum.PRE_30_ED,
electionId=election.electionId,
areaId=obj.areaId)
TallySheet.create(tallySheetCode=TallySheetCodeEnum.PRE_30_PD,
electionId=postal_election.electionId,
areaId=obj.areaId)
elif data_store_key == "Polling Division":
obj = PollingDivision.create(cell,
electionId=election.electionId)
TallySheet.create(tallySheetCode=TallySheetCodeEnum.PRE_30_PD,
electionId=ordinary_election.electionId,
areaId=obj.areaId)
elif data_store_key == "Polling District":
obj = PollingDistrict.create(cell,
electionId=election.electionId)
elif data_store_key == "Election Commission":
obj = ElectionCommission.create(cell,
electionId=election.electionId)
elif data_store_key == "District Centre":
obj = DistrictCentre.create(cell,
electionId=election.electionId)
elif data_store_key == "Counting Centre":
obj = CountingCentre.create(cell,
electionId=election.electionId)
TallySheet.create(tallySheetCode=TallySheetCodeEnum.PRE_41,
electionId=election.electionId,
areaId=obj.areaId)
TallySheet.create(tallySheetCode=TallySheetCodeEnum.PRE_21,
electionId=election.electionId,
areaId=obj.areaId)
if election.voteType is VoteTypeEnum.NonPostal:
TallySheet.create(tallySheetCode=TallySheetCodeEnum.CE_201,
electionId=election.electionId,
areaId=obj.areaId)
elif election.voteType is VoteTypeEnum.Postal:
TallySheet.create(
tallySheetCode=TallySheetCodeEnum.CE_201_PV,
electionId=election.electionId,
areaId=obj.areaId)
elif data_store_key == "Polling Station":
obj = PollingStation.create(
cell,
electionId=election.electionId,
registeredVotersCount=row["Registered Voters"])
else:
print("------------- Not supported yet : *%s*" %
data_store_key)
set_object_to_data_store(data_key, data_store_key, obj)
return obj
def get_rows_from_csv(csv_path):
csv_file_path = csv_path
if os.path.exists(csv_file_path) is True:
with open(csv_file_path, 'r') as f:
reader = csv.DictReader(f, delimiter=',')
rows = list(reader)
else:
rows = []
return rows
for row in get_rows_from_csv(party_candidate_dataset_file):
party = get_object(root_election, row, "Party")
root_election.add_party(partyId=party.partyId)
candidate = get_object(root_election, row, "Candidate")
root_election.add_candidate(candidateId=candidate.candidateId,
partyId=party.partyId)
for row in get_rows_from_csv(invalid_vote_categories_dataset_file):
root_election.add_invalid_vote_category(
row["Invalid Vote Category Description"])
for row in get_rows_from_csv(polling_station_dataset_file):
print("[ROW] ========= ", row)
country = get_object(root_election, {"Country": "Sri Lanka"},
"Country")
electoralDistrict = get_object(root_election, row,
"Electoral District")
pollingDivision = get_object(root_election, row, "Polling Division")
pollingDistrict = get_object(root_election, row, "Polling District")
electionCommission = get_object(
root_election,
{"Election Commission": "Sri Lanka Election Commission"},
"Election Commission")
districtCentre = get_object(root_election, row, "District Centre")
countingCentre = get_object(ordinary_election, row, "Counting Centre")
if row["Registered Voters"] is None or len(
row["Registered Voters"]) is 0:
registered_voters = 0
else:
registered_voters = row["Registered Voters"].replace(",", "")
pollingStation = get_object(
ordinary_election, {
"Polling Station": row["Polling Station (English)"],
"Registered Voters": registered_voters
}, "Polling Station")
country.add_child(electoralDistrict.areaId)
electoralDistrict.add_child(pollingDivision.areaId)
pollingDivision.add_child(pollingDistrict.areaId)
pollingDistrict.add_child(pollingStation.areaId)
electionCommission.add_child(districtCentre.areaId)
districtCentre.add_child(countingCentre.areaId)
countingCentre.add_child(pollingStation.areaId)
stationaryItems = []
for i in range(1, 4):
box_key = "Ballot Box %d" % i
if box_key in row and row[box_key] is not None and len(
row[box_key]) > 0:
if row[box_key] is not None and len(row[box_key]) > 0:
ballotBox = BallotBox.Model(
ballotBoxId=row[box_key],
electionId=root_election.electionId)
db.session.add(ballotBox)
stationaryItems.append(ballotBox)
# ballotBox = get_object({"Ballot Box": row[box_key]}, "Ballot Box")
if len(row["Ballot - start"]) is not 0 and len(
row["Ballot - end"]) is not 0:
for ballotId in range(int(row["Ballot - start"]),
int(row["Ballot - end"]) + 1):
ballot = Ballot.Model(ballotId=ballotId,
electionId=root_election.electionId)
db.session.add(ballot)
stationaryItems.append(ballot)
# ballot = get_object({"Ballot": str(ballotId)}, "Ballot")
# invoice.add_stationary_item(ballot.stationaryItemId)
if len(row["Tendered Ballot - start"]) is not 0 and len(
row["Tendered Ballot - end"]) is not 0:
for ballotId in range(int(row["Tendered Ballot - start"]),
int(row["Tendered Ballot - end"]) + 1):
ballot = Ballot.Model(ballotId=ballotId,
ballotType=BallotTypeEnum.Tendered,
electionId=root_election.electionId)
db.session.add(ballot)
stationaryItems.append(ballot)
# ballot = get_object({"Tendered Ballot": str(ballotId)}, "Tendered Ballot")
# invoice.add_stationary_item(ballot.stationaryItemId)
invoice = Invoice.create(electionId=root_election.electionId,
issuingOfficeId=countingCentre.areaId,
receivingOfficeId=pollingStation.areaId,
issuedTo=1)
invoice.add_stationary_items([
stationaryItem.stationaryItemId
for stationaryItem in stationaryItems
])
invoice.set_confirmed()
print("[ROW END] ========= ")
for row in get_rows_from_csv(postal_counting_centers_dataset_file):
print("[POSTAL ROW] ========= ", row)
country = get_object(root_election, {"Country": "Sri Lanka"},
"Country")
electoralDistrict = get_object(root_election, row,
"Electoral District")
electionCommission = get_object(
root_election,
{"Election Commission": "Sri Lanka Election Commission"},
"Election Commission")
districtCentre = get_object(root_election, row, "District Centre")
countingCentre = get_object(
postal_election,
{"Counting Centre": row["Postal Vote Counting Centre"]},
"Counting Centre")
country.add_child(electoralDistrict.areaId)
electoralDistrict.add_child(countingCentre.areaId)
districtCentre.add_child(countingCentre.areaId)
electionCommission.add_child(districtCentre.areaId)
print("[POSTAL ROW END] ========= ")
db.session.commit()
return root_election
def build_database(dataset):
global csv_dir
basedir = os.path.abspath(os.path.dirname(__file__))
sample_data_dir = os.path.join(basedir, 'sample-data')
csv_dir = "%s/%s" % (sample_data_dir, dataset)
for row in get_rows_from_csv('party-candidate.csv'):
party = get_object(root_election, row, "Party")
root_election.add_party(partyId=party.partyId)
candidate = get_object(root_election, row, "Candidate")
root_election.add_candidate(candidateId=candidate.candidateId,
partyId=party.partyId)
for row in get_rows_from_csv('invalid-vote-categories.csv'):
root_election.add_invalid_vote_category(
row["Invalid Vote Category Description"])
for row in get_rows_from_csv('data.csv'):
print("[ROW] ========= ", row)
country = get_object(root_election, {"Country": "Sri Lanka"},
"Country")
electoralDistrict = get_object(root_election, row,
"Electoral District")
pollingDivision = get_object(root_election, row, "Polling Division")
pollingDistrict = get_object(root_election, row, "Polling District")
electionCommission = get_object(
root_election,
{"Election Commission": "Sri Lanka Election Commission"},
"Election Commission")
districtCentre = get_object(root_election, row, "District Centre")
countingCentre = get_object(ordinary_election, row, "Counting Centre")
if row["Registered Voters"] is None or len(
row["Registered Voters"]) is 0:
registered_voters = 0
else:
registered_voters = row["Registered Voters"].replace(",", "")
pollingStation = get_object(
ordinary_election, {
"Polling Station": row["Polling Station (English)"],
"Registered Voters": registered_voters
}, "Polling Station")
country.add_child(electoralDistrict.areaId)
electoralDistrict.add_child(pollingDivision.areaId)
pollingDivision.add_child(pollingDistrict.areaId)
pollingDistrict.add_child(pollingStation.areaId)
electionCommission.add_child(districtCentre.areaId)
districtCentre.add_child(countingCentre.areaId)
countingCentre.add_child(pollingStation.areaId)
stationaryItems = []
for i in range(1, 4):
box_key = "Ballot Box %d" % i
if box_key in row and row[box_key] is not None and len(
row[box_key]) > 0:
if row[box_key] is not None and len(row[box_key]) > 0:
ballotBox = BallotBox.Model(
ballotBoxId=row[box_key],
electionId=root_election.electionId)
db.session.add(ballotBox)
stationaryItems.append(ballotBox)
# ballotBox = get_object({"Ballot Box": row[box_key]}, "Ballot Box")
if len(row["Ballot - start"]) is not 0 and len(
row["Ballot - end"]) is not 0:
for ballotId in range(int(row["Ballot - start"]),
int(row["Ballot - end"]) + 1):
ballot = Ballot.Model(ballotId=ballotId,
electionId=root_election.electionId)
db.session.add(ballot)
stationaryItems.append(ballot)
# ballot = get_object({"Ballot": str(ballotId)}, "Ballot")
# invoice.add_stationary_item(ballot.stationaryItemId)
if len(row["Tendered Ballot - start"]) is not 0 and len(
row["Tendered Ballot - end"]) is not 0:
for ballotId in range(int(row["Tendered Ballot - start"]),
int(row["Tendered Ballot - end"]) + 1):
ballot = Ballot.Model(ballotId=ballotId,
ballotType=BallotTypeEnum.Tendered,
electionId=root_election.electionId)
db.session.add(ballot)
stationaryItems.append(ballot)
# ballot = get_object({"Tendered Ballot": str(ballotId)}, "Tendered Ballot")
# invoice.add_stationary_item(ballot.stationaryItemId)
invoice = Invoice.create(electionId=root_election.electionId,
issuingOfficeId=countingCentre.areaId,
receivingOfficeId=pollingStation.areaId,
issuedTo=1)
invoice.add_stationary_items([
stationaryItem.stationaryItemId
for stationaryItem in stationaryItems
])
invoice.set_confirmed()
print("[ROW END] ========= ")
for row in get_rows_from_csv('postal-data.csv'):
print("[POSTAL ROW] ========= ", row)
country = get_object(root_election, {"Country": "Sri Lanka"},
"Country")
electoralDistrict = get_object(root_election, row,
"Electoral District")
electionCommission = get_object(
root_election,
{"Election Commission": "Sri Lanka Election Commission"},
"Election Commission")
districtCentre = get_object(root_election, row, "District Centre")
countingCentre = get_object(
postal_election,
{"Counting Centre": row["Postal Vote Counting Centre"]},
"Counting Centre")
country.add_child(electoralDistrict.areaId)
electoralDistrict.add_child(countingCentre.areaId)
districtCentre.add_child(countingCentre.areaId)
electionCommission.add_child(districtCentre.areaId)
print("[POSTAL ROW END] ========= ")
db.session.commit()
# Generate a token with claims for everything.
key = "jwt_secret"
encoded_jwt_token = jwt.encode(jwt_payload, key)
print("""
##########################################################
JWT_PAYLOAD = %s
JWT_TOKEN = %s
##########################################################
""" % (jwt_payload, encoded_jwt_token))
def delete(invoiceId):
result = Model.delete(invoiceId=invoiceId)
return result
def confirm(invoiceId):
result = Model.update(invoiceId=invoiceId, confirmed=True)
return Schema().dump(result).data
def get_by_id(invoiceId):
result = Model.get_by_id(invoiceId=invoiceId)
return Schema().dump(result).data
def delete(invoiceId):
result = Model.delete(invoiceId=invoiceId)
db.session.commit()
return result
def confirm(invoiceId):
result = Model.update(invoiceId=invoiceId, confirmed=True)
db.session.commit()
return Schema().dump(result).data
