app.py

from flask import Flask

from where.model import sa
from where.model.field_types import FieldType

app = Flask(__name__)


@app.route('/')
def index():
    return """
<head>
</head>
<body>
    <h1>W H E R E</h1>
    <p>This is the WHERE project.</p>
    <a href='/test_data'>Click here to nuke the database and make it all be test data.</a>
</body>
    """


@app.route('/test_data')
def test_data():
    with sa.session_context() as session:
        # session = Session()
        session.query(sa.Point).delete()
        session.query(sa.Field).delete()
        session.query(sa.Category).delete()
        # Water Fountain, the class.
        wf = sa.Category()
        wf.name = "Water Fountain"
        wf.icon = "https://karel.pw/water.png"
        session.add(wf)
        session.commit()
        # coldness
        cd = sa.Field()
        cd.name = "Coldness"
        cd.slug = "coldness"
        cd.type = FieldType.RATING
        cd.category_id = wf.id
        # filler
        fl = sa.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()
        # an actual instance!
        fn = sa.Point()
        fn.name = None
        fn.lat = 38.829791
        fn.lon = -77.307043
        # fn.category_id = wf.id
        fn.category = wf
        fn.parent_id = None
        fn.attributes = {
            "coldness": {
                "num_reviews": 32,
                "average_rating": 0.5
            },
            "bottle_filler": {
                "value": True
            }
        }
        session.add(fn)
        session.commit()
        return "Success!"


if __name__ == '__main__':
    app.run()