26 lines
419 B
Python
26 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
|