import sentry_sdk import click from flask import Flask from flask_bcrypt import Bcrypt from flask.cli import AppGroup from flask_sqlalchemy import SQLAlchemy from flask_migrate import Migrate import flask_login from flask_mailman import Mail from sentry_sdk.integrations.flask import FlaskIntegration from os import environ from authlib.integrations.flask_client import OAuth sentry_sdk.init( dsn=environ.get("SENTRY_DSN"), integrations=[FlaskIntegration()], traces_sample_rate=1.0, ) app = Flask(__name__) app.config.from_object("app.config") bcrypt = Bcrypt(app) db = SQLAlchemy(app) migrate = Migrate(app, db) login_manager = flask_login.LoginManager() oauth = OAuth(app) mail = Mail(app) oauth.register( name="google", server_metadata_url="https://accounts.google.com/.well-known/openid-configuration", client_kwargs={"scope": "openid email profile"}, ) login_manager.init_app(app) from app.models import User database_cli = AppGroup("database") @database_cli.command("create") def create_database(): db.create_all() @database_cli.command("delete") def delete_database(): db.drop_all() @database_cli.command("admin-create") def make_superuser_database(): user = User( first_name="Supersu", email=app.config["ADMIN_EMAIL"], password=app.config["ADMIN_PASSWORD"], role="SUPERUSER", ) user.confirmation = True db.session.add(user) db.session.commit() app.cli.add_command(database_cli) from app.views import main @login_manager.user_loader def load_user(email): return User.query.filter(User.email == email).first()