72 lines
1.6 KiB
Python
72 lines
1.6 KiB
Python
#!/usr/bin/env python
|
|
# coding: utf-8
|
|
|
|
import json
|
|
|
|
import arrow
|
|
import bottle
|
|
import peewee
|
|
|
|
db = peewee.SqliteDatabase('reports.db')
|
|
|
|
|
|
class Report(peewee.Model):
|
|
type = peewee.CharField(max_length=255)
|
|
lat = peewee.DoubleField()
|
|
lng = peewee.DoubleField()
|
|
datetime = peewee.DateTimeField(
|
|
default=arrow.utcnow().replace(microsecond=0)
|
|
)
|
|
|
|
class Meta:
|
|
database = db
|
|
|
|
|
|
@bottle.hook('after_request')
|
|
def enable_cors():
|
|
"""
|
|
Add CORS headers at each request.
|
|
"""
|
|
# The str() call is required as we import unicode_literal and WSGI
|
|
# headers list should have plain str type.
|
|
bottle.response.headers[str('Access-Control-Allow-Origin')] = str('*')
|
|
bottle.response.headers[str('Access-Control-Allow-Methods')] = str(
|
|
'PUT, GET, POST, DELETE, OPTIONS, PATCH'
|
|
)
|
|
bottle.response.headers[str('Access-Control-Allow-Headers')] = str(
|
|
'Origin, Accept, Content-Type, X-Requested-With, X-CSRF-Token'
|
|
)
|
|
|
|
|
|
@bottle.route('/api/v1/reports', ["POST", "OPTIONS"])
|
|
def postReport():
|
|
"""
|
|
Add a new report in database.
|
|
"""
|
|
if bottle.request.method == 'OPTIONS':
|
|
return {}
|
|
|
|
try:
|
|
payload = json.load(bottle.request.body)
|
|
except ValueError:
|
|
bottle.abort(400, "Invalid JSON payload.")
|
|
|
|
try:
|
|
Report.create(
|
|
type=payload['type'],
|
|
lat=payload['lat'],
|
|
lng=payload['lng']
|
|
)
|
|
except KeyError:
|
|
bottle.abort(400, "Invalid JSON payload.")
|
|
|
|
return {
|
|
"status": "ok"
|
|
}
|
|
|
|
|
|
if __name__ == "__main__":
|
|
db.connect()
|
|
db.create_tables([Report])
|
|
bottle.run(host='0.0.0.0', port='8081')
|