From 4e712eb891275c5a6039b1e7fdaf663dd4810c85 Mon Sep 17 00:00:00 2001 From: Christian Quest Date: Mon, 30 May 2016 08:03:17 +0200 Subject: [PATCH] add count in results --- backend.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/backend.py b/backend.py index 4473d67..8cbe9f6 100644 --- a/backend.py +++ b/backend.py @@ -70,10 +70,11 @@ class BaseEvent: "properties": properties } - def rows_to_collection(self, rows): + def rows_to_collection(self, rows, count): return { "type": "FeatureCollection", - "features": [self.row_to_feature(r) for r in rows] + "features": [self.row_to_feature(r) for r in rows], + "count": count } @@ -83,7 +84,7 @@ class EventsResource(BaseEvent): db = db_connect() cur = db.cursor(cursor_factory=psycopg2.extras.DictCursor) cur.execute("SELECT events_id, events_tags, createdate, lastupdate, st_asgeojson(geom) as geometry FROM events JOIN geo ON (hash=events_geo)") - resp.body = dumps(self.rows_to_collection(cur.fetchall())) + resp.body = dumps(self.rows_to_collection(cur.fetchall(), cur.rowcount)) resp.status = falcon.HTTP_200 @@ -174,7 +175,7 @@ class EventResource(BaseEvent): event_bbox=event_bbox, event_what=event_what, event_when=event_when, event_type=event_type) cur.execute(sql) - resp.body = dumps(self.rows_to_collection(cur.fetchall())) + resp.body = dumps(self.rows_to_collection(cur.fetchall(), cur.rowcount)) resp.status = falcon.HTTP_200 else: # Get single event geojson Feature by id.