aboutsummaryrefslogtreecommitdiff
path: root/gn3/auth/db_utils.py
blob: c06b02653c026bbd488befd24576d1770c0dac76 (about) (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
"""Some common auth db utilities"""
from typing import Any, Callable
from flask import current_app

from . import db

def with_db_connection(func: Callable[[db.DbConnection], Any]) -> Any:
    """
    Takes a function of one argument `func`, whose one argument is a database
    connection.
    """
    db_uri = current_app.config["AUTH_DB"]
    with db.connection(db_uri) as conn:
        return func(conn)