def get_results_from_table(table):
query = db_session.query(table.bwk,
*sum_party_results(table)).group_by(table.bwk)
result = {}
for row in query.all():
result[row[0]] = dict(zip(PARTIES, [int(p) for p in row[1:]]))
return result
def get_county_geojson(bwks=None):
# Please notice the missing filter statement!
if bwks:
query = db_session.query(
MergedDistrictDiff.bwk,
functions.ST_AsGeoJSON(
functions.ST_Union(
functions.ST_Transform(MergedDistrictDiff.geom,
4326))).label("geom"),
*sum_party_results(MergedDistrictDiff)).filter(
MergedDistrictDiff.bwk.in_(bwks)).group_by(
MergedDistrictDiff.bwk)
else:
query = db_session.query(
MergedDistrictDiff.bwk,
functions.ST_AsGeoJSON(
functions.ST_Union(
functions.ST_Transform(MergedDistrictDiff.geom,
4326))).label("geom"),
*sum_party_results(MergedDistrictDiff)).group_by(
MergedDistrictDiff.bwk)
geojsons = []
for bwk, geom, cdu, spd, gruene, die_linke, fdp, afd in query.all():
candidates = db_session.query(Candidate).filter(Candidate.bwk == bwk)
geojson = json.loads(geom)
geojson['properties'] = {
'bwk': bwk,
'candidates': {
candidate.party_key(): candidate.get_json()
for candidate in candidates
},
'result': {
'cdu': int(cdu),
'spd': int(spd),
'gruene': int(gruene),
'die_linke': int(die_linke),
'fdp': int(fdp),
'afd': int(afd)
}
}
geojsons.append(geojson)
return geojsons
def get_district_geojson(district):
query = db_session.query(
district,
functions.ST_AsGeoJSON(functions.ST_Transform(district.geom, 4326)))
geojsons = []
for district, geom in query.all():
geojson = json.loads(geom)
geojson["properties"] = district.get_geojson_dict()
geojsons.append(geojson)
return geojsons
def get_geojson_dict(self):
candidates = db_session.query(Candidate).filter(
Candidate.bwk == self.bwk)
return {
'identifier': self.identifier,
'bezname': self.bezname,
'bwk': self.bwk,
'candidates': {
candidate.party: candidate.get_json()
for candidate in candidates
},
'result': self.get_result_dict()
}
def upsert_diff(identifier, bwk):
diff = db_session.query(Diff).filter(Diff.identifier == identifier).first()
diff = diff or Diff(identifier=identifier)
diff.bwk = bwk
db_session.add(diff)
db_session.commit()