134 lines
5 KiB
Python
134 lines
5 KiB
Python
# backend.py
|
|
# openeventdatabase
|
|
|
|
import os
|
|
import falcon
|
|
import psycopg2
|
|
import uuid
|
|
import json
|
|
import codecs
|
|
|
|
def db_connect():
|
|
db_host = os.getenv("DB_HOST","localhost")
|
|
db_password = os.getenv("POSTGRES_PASSWORD","")
|
|
db = psycopg2.connect(dbname="oedb",host=db_host,password=db_password)
|
|
return db
|
|
|
|
def standard_headers(resp):
|
|
resp.set_header('X-Powered-By', 'OpenEventDatabase')
|
|
resp.set_header('Access-Control-Allow-Origin', '*')
|
|
resp.set_header('Access-Control-Allow-Headers', 'X-Requested-With')
|
|
|
|
class StatsResource(object):
|
|
def on_get(self, req, resp):
|
|
db = db_connect()
|
|
cur = db.cursor()
|
|
cur.execute("SELECT count(*) as events_count, max(createdate) as last_created, max(lastupdate) as last_updated from events;")
|
|
stat = cur.fetchone()
|
|
cur.close()
|
|
db.close()
|
|
|
|
standard_headers(resp)
|
|
resp.body = """{"events_count": %s, "last_created": "%s", "last_updated": "%s"}""" % (stat[0], stat[1],stat[2])
|
|
resp.status = falcon.HTTP_200
|
|
|
|
class EventsResource(object):
|
|
def on_get(self,req,resp):
|
|
db = db_connect()
|
|
cur = db.cursor()
|
|
# get event geojson Feature
|
|
cur.execute("""
|
|
SELECT format('{"type":"Feature", "id": "'|| events_id::text ||'", "properties": '|| events_tags::text ||', "geometry":'|| st_asgeojson(geom)) ||' }'
|
|
FROM events
|
|
JOIN geo ON (hash=events_geo)""");
|
|
standard_headers(resp)
|
|
resp.body = '{"type": "FeatureCollection","features": ['+','.join([x[0] for x in cur.fetchall()])+']}'
|
|
resp.status = falcon.HTTP_200
|
|
|
|
class EventResource(object):
|
|
def on_get(self, req, resp, id = None):
|
|
db = db_connect()
|
|
cur = db.cursor()
|
|
if id is None:
|
|
cur.execute("""
|
|
SELECT format('{"type": "FeatureCollection","features": [%s]}',string_agg(feature,',')) FROM
|
|
(SELECT '{"type":"Feature", "properties": '|| events_tags::text ||', "geometry":'|| st_asgeojson(geom) ||' }' as feature
|
|
FROM events
|
|
JOIN geo ON (hash=events_geo)
|
|
WHERE events_when @> format('[%s,%s]', now(), now())::tstzrange
|
|
ORDER BY createdate DESC
|
|
LIMIT 50) as f;
|
|
""")
|
|
else:
|
|
# get event geojson Feature
|
|
cur.execute("""
|
|
SELECT format('{"type":"Feature", "properties": '|| events_tags::text ||', "geometry":'|| st_asgeojson(geom)) ||' }'
|
|
FROM events
|
|
JOIN geo ON (hash=events_geo)
|
|
WHERE events_id=%s;""", (id,))
|
|
|
|
e = cur.fetchone()
|
|
standard_headers(resp)
|
|
if e is not None:
|
|
resp.body = e[0]
|
|
resp.status = falcon.HTTP_200
|
|
else:
|
|
resp.status = falcon.HTTP_404
|
|
db.close()
|
|
|
|
def on_post(self, req, resp):
|
|
standard_headers(resp)
|
|
|
|
# get request body payload (geojson Feature)
|
|
body = req.stream.read().decode('utf-8')
|
|
j=json.loads(body)
|
|
if "properties" not in j or "geometry" not in j:
|
|
resp.body = "missing 'geometry' or 'properties' elements"
|
|
resp.status = falcon.HTTP_400
|
|
if "start" not in j['properties']:
|
|
event_start = j['properties']['when']
|
|
else:
|
|
event_start = j['properties']['start']
|
|
if "stop" not in j['properties']:
|
|
event_stop = j['properties']['when']
|
|
else:
|
|
event_stop = j['properties']['stop']
|
|
if event_start == event_stop:
|
|
when = "["+event_start+", "+event_stop+"]"
|
|
else:
|
|
when = "["+event_start+", "+event_stop+")"
|
|
# connect to db and insert
|
|
db = db_connect()
|
|
cur = db.cursor()
|
|
# get the geometry part
|
|
geometry=json.dumps(j['geometry'])
|
|
# insert into geo table if not existing
|
|
cur.execute("""INSERT INTO geo (hash, geom) SELECT * FROM (SELECT md5(ewkt) as hash, st_setsrid(st_geomfromewkt(ewkt),4326) as geom FROM (SELECT st_asewkt(st_geomfromgeojson( %s )) as ewkt) as g) as i ON CONFLICT DO NOTHING RETURNING hash;""",(geometry,))
|
|
# get its id (md5 hash)
|
|
h = cur.fetchone()
|
|
if h is None:
|
|
cur.execute("""SELECT md5(st_asewkt(st_geomfromgeojson( %s )));""",(geometry,))
|
|
h = cur.fetchone()
|
|
cur.execute("""INSERT INTO events ( events_type, events_what, events_when, events_tags, events_geo) VALUES (%s, %s, %s, %s, %s) RETURNING events_id;""",(j['properties']['type'],j['properties']['what'],when,json.dumps(j['properties']),h[0]))
|
|
# get newly created event id
|
|
e = cur.fetchone()
|
|
db.commit()
|
|
cur.close()
|
|
db.close()
|
|
# send back to client
|
|
resp.body = """{"id":"%s"}""" % (e[0])
|
|
resp.status = falcon.HTTP_201
|
|
|
|
# falcon.API instances are callable WSGI apps
|
|
app = falcon.API()
|
|
|
|
# Resources are represented by long-lived class instances
|
|
events = EventsResource()
|
|
event = EventResource()
|
|
stats = StatsResource()
|
|
|
|
# things will handle all requests to the matching URL path
|
|
app.add_route('/events', events)
|
|
app.add_route('/event/{id}', event) # handle single event requests
|
|
app.add_route('/event', event) # handle single event requests
|
|
app.add_route('/stats', stats)
|