"""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