blob: c06b02653c026bbd488befd24576d1770c0dac76 (
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)
|
