78 lines
2.3 KiB
Python
78 lines
2.3 KiB
Python
import logging
|
|
|
|
from flask import Flask, jsonify, url_for, redirect, abort, request
|
|
from flask_sqlalchemy import SQLAlchemy
|
|
from sqlalchemy import text
|
|
from raven.contrib.flask import Sentry
|
|
import config
|
|
|
|
app = Flask(__name__)
|
|
if not config.DEBUG:
|
|
sentry = Sentry(app=app, dsn=config.sentry_url, logging=True, level=logging.ERROR)
|
|
app.config['SQLALCHEMY_DATABASE_URI'] = "mysql+pymysql://{user}:{password}@{host}/{db}?{settings}".format(**config.db)
|
|
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
|
|
db = SQLAlchemy(app)
|
|
|
|
|
|
@app.before_first_request
|
|
def setup_logging():
|
|
if not app.debug:
|
|
# In production mode, add log handler to sys.stderr.
|
|
app.logger.addHandler(logging.StreamHandler())
|
|
app.logger.setLevel(logging.INFO)
|
|
|
|
|
|
@app.errorhandler(404)
|
|
@app.errorhandler(400)
|
|
def handle_invalid_usage(error):
|
|
response = jsonify({
|
|
"code": error.code,
|
|
"name": error.name,
|
|
"description": error.description
|
|
})
|
|
response.status_code = error.code
|
|
return response
|
|
|
|
|
|
@app.route('/api/venue/')
|
|
def get_venues():
|
|
sql = text("SELECT * FROM venues")
|
|
results = db.engine.execute(sql).fetchall()
|
|
venues = []
|
|
for row in results:
|
|
venue = dict(row)
|
|
venue["meals_url"] = url_for("get_meals", venueid=venue["id"], _external=True)
|
|
venues.append(venue)
|
|
return jsonify(venues)
|
|
|
|
|
|
@app.route('/api/venue/<int:venueid>/')
|
|
def get_meals(venueid):
|
|
if request.args.get('mode', "week") == "week":
|
|
sql_range = " AND YEARWEEK(date, 1) = YEARWEEK(CURDATE(), 1) + :offset"
|
|
elif request.args.get('mode') == "day":
|
|
sql_range = " AND date=DATE_ADD(CURDATE(), INTERVAL :offset DAY) "
|
|
else:
|
|
return abort(400)
|
|
sql = """SELECT venue, date, name, description
|
|
FROM menus
|
|
JOIN meals ON meals.id = menus.meal
|
|
WHERE venue = :venue"""
|
|
results = db.engine.execute(text(sql + sql_range),
|
|
{"venue": venueid, "offset": request.args.get('offset', 0)}
|
|
).fetchall()
|
|
menus = []
|
|
for row in results:
|
|
menu = dict(row)
|
|
menu["date"] = menu["date"].isoformat()
|
|
menus.append(menu)
|
|
return jsonify(menus)
|
|
|
|
|
|
@app.route("/api/")
|
|
def redirect_to_correct_api():
|
|
return redirect(url_for("get_venues"))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
app.run(debug=True)
|