import os import logging from datetime import datetime import asyncio from quart import Quart, url_for, jsonify, render_template, session from quart_session_openid import OpenID from quart_session import Session import settings app: Quart = None peewee = None cache = None openid: OpenID = None async def _setup_database(app: Quart): import peewee import yellow.models models = peewee.Model.__subclasses__() for m in models: m.create_table() async def _setup_openid(app: Quart): global openid openid = OpenID(app, **settings.OPENID_CFG) from yellow.auth import handle_user_login async def _setup_cache(app: Quart): global cache app.config['SESSION_TYPE'] = 'redis' app.config['SESSION_URI'] = settings.REDIS_URI Session(app) async def _setup_error_handlers(app: Quart): @app.errorhandler(500) async def page_error(e): return await render_template('error.html', code=500, msg="Error occurred"), 500 @app.errorhandler(403) async def page_forbidden(e): return await render_template('error.html', code=403, msg="Forbidden"), 403 @app.errorhandler(404) async def page_not_found(e): return await render_template('error.html', code=404, msg="Page not found"), 404 def create_app(): global app app = Quart(__name__) app.logger.setLevel(logging.INFO) app.secret_key = settings.APP_SECRET @app.context_processor def template_variables(): global openid from yellow.models import User current_user = session.get('user') if current_user: current_user = User(**current_user) now = datetime.now() return dict(user=current_user, url_login=openid.endpoint_name_login, year=now.year) @app.before_serving async def startup(): await _setup_cache(app) await _setup_openid(app) await _setup_database(app) await _setup_error_handlers(app) from yellow.routes import bp_routes from yellow.api import bp_api app.register_blueprint(bp_routes) app.register_blueprint(bp_api) return app