25 lines
419 B
Python
25 lines
419 B
Python
"""Simple wrapper for getting/setting
|
|
db key in global Flask object called `g`"""
|
|
|
|
from flask import g
|
|
from flaskext.mysql import MySQL # type: ignore
|
|
|
|
|
|
def set_db(db: MySQL) -> None:
|
|
"""Add the database to `g` object
|
|
|
|
Args:
|
|
db (MySQL): MySQL database
|
|
"""
|
|
|
|
g.db = db
|
|
|
|
|
|
def get_db() -> MySQL:
|
|
"""Get the database from `g` object
|
|
|
|
Returns:
|
|
MySQL database
|
|
"""
|
|
|
|
return g.db
|