A Python script to unload all modules within a package

unload_pkg

"""Python utilies."""

import os
import sys


def unload(pkg):
    """Unload all imports from the given package.

    Args:
        pkg (module): Python module to unload.
    """

    pkg_dir = os.path.abspath(os.path.dirname(pkg.__file__))

    def _is_part_of_pkg(module_):
        mod_path = getattr(module_, "__file__", os.sep)
        mod_dir = os.path.abspath(os.path.dirname(mod_path))

        return mod_dir.startswith(pkg_dir)

    to_unload = [name for name, module in sys.modules.items() if _is_part_of_pkg(module)]

    for name in to_unload:
        sys.modules.pop(name)