web.py

from app.models.latest_update import LatestUpdate
from app.models.lift import Lift
from app.db.session import Session
from app.shared.firebase import init_firebase_admin

import os

from starlette.applications import Starlette
from starlette.templating import Jinja2Templates
from starlette.responses import JSONResponse, FileResponse
from starlette.routing import Mount
from starlette.staticfiles import StaticFiles
from firebase_admin import messaging

APP_ENV = os.getenv('APP_ENV', 'development')
SEASON = os.getenv('SEASON', 'Winter')

templates = Jinja2Templates(directory='app/templates')
routes = [
    Mount('/static', app=StaticFiles(directory='app/static'), name="static")]
app = Starlette(routes=routes)

init_firebase_admin()


@app.route('/', methods=['GET'])
async def root(request):
    session = Session()

    try:
        season = request.query_params.get('season', SEASON)

        latest_update = session.query(LatestUpdate).filter(
            LatestUpdate.season == season).order_by(LatestUpdate.id.desc()).first()
        latest_update_dict = None
        if latest_update:
            latest_update_dict = latest_update.for_html()

        lifts = session.query(Lift).filter(
            Lift.season == season).order_by(Lift.last_updated.desc()).all()
        lift_dicts = list(l.for_html() for l in lifts)

        template_params = {
            'request': request, 'latest_update': latest_update_dict, 'season': season, 'lifts': lift_dicts}

        return templates.TemplateResponse('lifts/index.html.j2', template_params)
    finally:
        session.close()


@app.route('/register', methods=['POST'])
async def register(request):
    data = await request.json()
    subscriber_id = data['subscriber_id']
    tokens = [subscriber_id]

    messaging.subscribe_to_topic(tokens, 'updates')

    return JSONResponse({'success': True})


@app.route('/firebase-messaging.js', methods=['GET'])
async def messaging_js(request):
    if APP_ENV == 'production':
        return FileResponse('app/static/firebase-messaging.js')
    else:
        return FileResponse('app/static/firebase-messaging-dev.js')


@app.route('/firebase-messaging-sw.js', methods=['GET'])
async def messaging_sw_js(request):
    if APP_ENV == 'production':
        return FileResponse('app/static/firebase-messaging-sw.js')
    else:
        return FileResponse('app/static/firebase-messaging-dev-sw.js')


@app.route('/api/lifts', methods=['GET'])
async def api_lifts(request):
    session = Session()

    try:
        season = request.query_params.get('season', SEASON)
        lifts = session.query(Lift).filter(
            Lift.season == season).order_by(Lift.last_updated.desc()).all()
        lift_dicts = list(l.for_json() for l in lifts)

        return JSONResponse({'lifts': lift_dicts})

    finally:
        session.close()


@app.route('/api/latest_update', methods=['GET'])
async def api_latest_update(request):
    session = Session()

    try:
        latest_update = session.query(LatestUpdate).order_by(
            LatestUpdate.id.desc()).first()

        return JSONResponse(latest_update.for_json())

    finally:
        session.close()