Use DictCursor for stat query

This commit is contained in:
Yohan Boniface 2016-05-13 21:10:58 +02:00
parent 5458bedd09
commit a9f1220bca

View file

@ -4,6 +4,7 @@
import os import os
import falcon import falcon
import psycopg2 import psycopg2
import psycopg2.extras
import json import json
@ -29,19 +30,14 @@ def standard_headers(resp):
class StatsResource(object): class StatsResource(object):
def on_get(self, req, resp): def on_get(self, req, resp):
db = db_connect() db = db_connect()
cur = db.cursor() cur = db.cursor(cursor_factory=psycopg2.extras.DictCursor)
cur.execute("SELECT count(*) as events_count, max(createdate) as last_created, max(lastupdate) as last_updated from events;") cur.execute("SELECT count(*) as events_count, max(createdate) as last_created, max(lastupdate) as last_updated from events;")
stat = cur.fetchone() stat = cur.fetchone()
cur.close() cur.close()
db.close() db.close()
standard_headers(resp) standard_headers(resp)
body = { resp.body = json.dumps(dict(stat))
"events_count": stat[0],
"last_created": stat[1],
"last_updated": stat[2]
}
resp.body = json.dumps(body)
resp.status = falcon.HTTP_200 resp.status = falcon.HTTP_200