from flask import Flask, redirect, jsonify, abort, request, url_for, make_response
from where.model import with_session, Point, Category, Field
from where.model.field_types import FieldType
app = Flask(__name__)
def create_resource(session, model_cls, data, get_function):
model = model_cls(**data)
session.add(model)
session.commit()
response = make_response(jsonify(model.as_json()), 201)
response.headers['Location'] = url_for(get_function, id=model.id)
return response
def search_resource(session, model_cls, data):
query = session.query(model_cls).filter_by(**data)
results = list(map(lambda m: m.as_json(), query.limit(100).all()))
response = make_response(jsonify(results), 200)
return response
@app.route('/')
def index():
return """
W H E R E
This is the WHERE project.
Click here to nuke the database and make it all be test data.
"""
@app.route('/test_data')
@with_session
def test_data(session):
session.query(Point).delete()
session.query(Field).delete()
session.query(Category).delete()
# Water Fountain, the class.
wf = Category()
wf.name = "Water Fountain"
wf.icon = "https://karel.pw/water.png"
session.add(wf)
session.commit()
# Building
bd = Category()
bd.name = "Building"
bd.icon = "https://hips.hearstapps.com/hmg-prod.s3.amazonaws.com/images/basket-building-news-photo-1572015168.jpg?resize=980:*"
session.add(bd)
session.commit()
# Radius (Really the simplest metric we can have for building size)
rd = Field()
rd.name = "Radius"
rd.slug = "radius"
rd.type = FieldType.FLOAT
rd.category_id = bd.id
session.add(rd)
session.commit()
# coldness
cd = Field()
cd.name = "Coldness"
cd.slug = "coldness"
cd.type = FieldType.RATING
cd.category_id = wf.id
# filler
fl = Field()
fl.slug = "bottle_filler"
fl.name = "Has Bottle Filler"
fl.type = FieldType.BOOLEAN
fl.category_id = wf.id
session.add(cd)
session.add(fl)
session.commit()
# The johnson center
jc = Point()
jc.category = bd
jc.name = "Johnson Center"
jc.lat = 38
jc.lon = -77
jc.parent = None
jc.attributes = {
"radius": {
"value": 2.0
}
}
session.add(jc)
# A water fountain inside the JC
fn = Point()
fn.name = None
fn.lat = 38.829791
fn.lon = -77.307043
# fn.category_id = wf.id
fn.category = wf
fn.parent = jc
fn.attributes = {
"coldness": {
"num_reviews": 32,
"average_rating": 0.5
},
"bottle_filler": {
"value": True
}
}
session.add(fn)
session.commit()
return redirect('/')
@app.route('/category/')
@with_session
def get_category(session, id):
result = session.query(Category).filter_by(id=id).first()
if result:
return jsonify(result.as_json())
else:
abort(404)
@app.route('/category//children')
@with_session
def get_category_children(session, id):
data = dict(request.args)
data['parent_id'] = id
return search_resource(session, Point, data)
@app.route('/point', methods=['GET', 'POST'])
@with_session
def handle_point(session):
if request.method == 'GET':
return search_resource(session, Point, dict(request.args))
elif request.method == 'POST':
data = request.get_json()
data['category'] = session.query(Category).get(data['category'])
data['parent_id'] = data.pop('parent', None)
return create_resource(session, Point, data, 'get_point')
@app.route('/point/', methods=['GET', 'PUT', 'DELETE'])
@with_session
def get_point(session, id):
result = session.query(Point).get(id)
if result:
return jsonify(result.as_json())
else:
abort(404)
@app.route('/point//children', methods=['GET'])
@with_session
def get_point_children(session, id):
data = dict(request.args)
data['parent_id'] = id
return search_resource(session, Point, data)
if __name__ == '__main__':
app.run()