import os

PDB_ENABLED = os.getenv("PDB", "false").lower() == "true"

def pdb_debug(
    main_func,
	enabled: bool = PDB_ENABLED
):  # 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:
            if enabled:
                _, _, tb = sys.exc_info()
                traceback.print_exc()
                pdb.post_mortem(tb)
            else:
                raise e

    return wrapper