예제 #1
0
def conn_with_scout():
    with app_with_scout() as app:
        app.config["SQLALCHEMY_DATABASE_URI"] = "sqlite:///:memory:"
        db = SQLAlchemy(app)
        # Setup according to http://help.apm.scoutapp.com/#flask-sqlalchemy
        instrument_sqlalchemy(db)
        conn = db.engine.connect()
        try:
            yield conn
        finally:
            conn.close()
def conn_with_scout():
    with app_with_scout() as app:
        app.config["SQLALCHEMY_DATABASE_URI"] = "sqlite:///:memory:"
        app.config["SQLALCHEMY_TRACK_MODIFICATIONS"] = False
        db = SQLAlchemy(app)
        # Setup according to https://docs.scoutapm.com/#flask-sqlalchemy
        instrument_sqlalchemy(db)
        conn = db.engine.connect()
        try:
            yield conn
        finally:
            conn.close()
예제 #3
0
def configure_scout(app):
    try:
        from scout_apm.flask import ScoutApm
        from scout_apm.flask.sqlalchemy import instrument_sqlalchemy
    except ImportError:
        return False

    app.config.setdefault("SCOUT_NAME", "Zeus")
    app.config.setdefault("SCOUT_MONITOR", True)

    ScoutApm(app)
    instrument_sqlalchemy(db)
예제 #4
0
def app_with_scout():
    with flask_app_with_scout() as app:
        app.config["SQLALCHEMY_DATABASE_URI"] = "sqlite:///:memory:"
        app.config["SQLALCHEMY_TRACK_MODIFICATIONS"] = False
        db = SQLAlchemy(app)
        # Setup according to https://docs.scoutapm.com/#flask-sqlalchemy
        instrument_sqlalchemy(db)
        conn = db.engine.connect()

        @app.route("/sqlalchemy/")
        def sqlalchemy():
            result = conn.execute("SELECT 'Hello from the DB!'")
            return list(result)[0][0]

        try:
            yield app
        finally:
            conn.close()
예제 #5
0
파일: utils.py 프로젝트: pushresume/backend
def load_scout_apm(app, db, celery=False):
    logger = _get_logger(app, celery)
    try:
        if celery:
            import scout_apm.celery
            from scout_apm.api import Config
            Config.set(key=app.config['SCOUT_KEY'],
                       monitor=app.config['SCOUT_MONITOR'],
                       name=app.config['SCOUT_NAME'])
            scout_apm.celery.install()
        else:
            from scout_apm.flask import ScoutApm
            from scout_apm.flask.sqlalchemy import instrument_sqlalchemy
            ScoutApm(app)
            instrument_sqlalchemy(db)
    except ImportError:
        logger.warning('Scout APM modules not found')
    else:
        logger.info('Scout APM initialized')
예제 #6
0
db = SQLAlchemy(app)


class Role(Enum):
	"""These are used with the 'login_required_with_role' function."""

	Normal = 'N'  # Normal users, without admin and mod powers, but with a profile.
	Mod = 'M'  # Mod users, with mod powers but without a profile/account page.
	Admin = 'A'  # Admin user(s), with admin powers but without a profile/account page.


if os.environ.get("HEROKU"):
	# We're running on Heroku, so we'll add this for statistics.
	from scout_apm.flask.sqlalchemy import instrument_sqlalchemy

	instrument_sqlalchemy(db)


def login_required_with_role(role):
	"""Used instead of "login_required" when a specific role is needed."""

	role = role.value  # Because it's an enum, we'll get its value here.

	def wrapper(fn):
		@wraps(fn)
		def decorated_view(*args, **kwargs):
			if not current_user.is_authenticated:
				return login_manager.unauthorized()

			if current_user.role != role:
				return login_manager.unauthorized()