__init__.py

# type:ignore
import string
from typing import Any
from sqlalchemy import MetaData
from flask import Flask
from flask_cors import CORS
from flask_sqlalchemy import SQLAlchemy
from flask_caching import Cache
from flask_compress import Compress
from application.config import CMDConfig, config
import os
import random


convention = {
    "ix": "ix_%(column_0_label)s",
    "uq": "uq_%(table_name)s_%(column_0_name)s",
    "ck": "ck_%(table_name)s_%(constraint_name)s",
    "fk": "fk_%(table_name)s_%(column_0_name)s_%(referred_table_name)s",
    "pk": "pk_%(table_name)s",
}
metadata = MetaData(naming_convention=convention)
sqla = SQLAlchemy(metadata=metadata)
compress = Compress()
cache = Cache()


def create_app(mode: str = "production", conf: any = None) -> Any:
    app = Flask(__name__)
    if not conf:
        app.config.from_object(config[mode])
    else:
        app.config.from_object(conf)
    # CLI passes an explicit URI via ``CMDConfig`` (``cre_main.db_connect``). A
    # leftover ``SQLALCHEMY_DATABASE_URI`` in the environment (e.g. sqlite from a
    # prior shell export) must not override it — that produced sqlite errors
    # against a Heroku ``DATABASE_URL`` session.
    if os.environ.get("SQLALCHEMY_DATABASE_URI") and not isinstance(conf, CMDConfig):
        app.config["SQLALCHEMY_DATABASE_URI"] = os.environ.get(
            "SQLALCHEMY_DATABASE_URI"
        )
    GOOGLE_CLIENT_SECRET = os.environ.get("GOOGLE_CLIENT_SECRET")
    app.secret_key = GOOGLE_CLIENT_SECRET
    if os.environ.get("NO_LOGIN"):
        letters = string.ascii_lowercase
        app.secret_key = "".join(random.choice(letters) for i in range(20))

    # config[mode].init_app(app)
    sqla.init_app(app=app)
    from application.web.web_main import app as app_blueprint

    app.register_blueprint(app_blueprint)

    from application.web.openapi_registry import init_openapi

    init_openapi(app)

    CORS(app)
    app.config["CORS_HEADERS"] = "Content-Type"

    compress.init_app(app)
    cache.init_app(app)

    return app