def extension()

in src/redash_stmo/query_runner/presto.py [0:0]

• 6 lines of code
• 2 McCabe index (conditional complexity)

def extension(app):
    logger.info("Loading Redash Extension for the custom Presto query runner")
    # Monkeypatch the pyhive.presto.connect function
    presto.connect = stmo_connect
    # and register our own STMOPresto query runner class
    # which automatically overwrites the default presto query runner
    register(STMOPresto)
    logger.info("Loaded Redash Extension for the custom Presto query runner")
    return stmo_connect