Source code for app.shell

"""
app.shell
=========

Define attributes regarding use of ``flask shell``.
"""

from flask import Flask

from app.models import Message, Notification, Post, Task, User, Usernames, db



[docs]
def register_models(app: Flask) -> None:
    """Make database models accessible to the ``Flask`` shell.

    :param app: Application object.
    """
    app.shell_context_processor(
        lambda: {
            "db": db,
            "User": User,
            "Post": Post,
            "Message": Message,
            "Notification": Notification,
            "Task": Task,
            "Usernames": Usernames,
        }
    )




[docs]
def init_app(app: Flask) -> None:
    """Register shell context objects.

    :param app: Application object.
    """
    register_models(app)