def pdb_debug(
    main_func,
):  # pylint: disable=missing-function-docstring, broad-exception-caught
    """Run a function, optionally drop to pdb on exception"""

    def wrapper():
        import pdb
        import sys
        import traceback

        try:
            main_func()
        except Exception as e:
            # PDB is actually read from .env
            if PDB:
                _, _, tb = sys.exc_info()
                traceback.print_exc()
                pdb.post_mortem(tb)
            else:
                raise e

    return wrapper