"""
Utility methods to print system info for debugging

adapted from :func:`pandas.show_versions`
"""
# License: BSD 3 clause

import platform
import sys
from ..utils.fixes import threadpool_info
from .. import __version__


from ._openmp_helpers import _openmp_parallelism_enabled


def _get_sys_info():
    """System information

    Returns
    -------
    sys_info : dict
        system and Python version information

    """
    python = sys.version.replace("\n", " ")

    blob = [
        ("python", python),
        ("executable", sys.executable),
        ("machine", platform.platform()),
    ]

    return dict(blob)


def _get_deps_info():
    """Overview of the installed version of main dependencies

    This function does not import the modules to collect the version numbers
    but instead relies on standard Python package metadata.

    Returns
    -------
    deps_info: dict
        version information on relevant Python libraries

    """
    deps = [
        "pip",
        "setuptools",
        "numpy",
        "scipy",
        "Cython",
        "pandas",
        "matplotlib",
        "joblib",
        "threadpoolctl",
    ]

    deps_info = {
        "sklearn": __version__,
    }

    if sys.version_info < (3, 8):
        # Backwards compatibility with Python < 3.8, primarily for PyPy.
        # TODO: remove once PyPy 3.8 is available on conda-forge and
        # therefore on our CI.
        # https://github.com/conda-forge/conda-forge-pinning-feedstock/issues/2089
        try:
            from pkg_resources import get_distribution, DistributionNotFound

            for modname in deps:
                try:
                    deps_info[modname] = get_distribution(modname).version
                except DistributionNotFound:
                    deps_info[modname] = None

        except ImportError:
            # Setuptools not installed
            for modname in deps:
                deps_info[modname] = None

    else:
        from importlib.metadata import version, PackageNotFoundError

        for modname in deps:
            try:
                deps_info[modname] = version(modname)
            except PackageNotFoundError:
                deps_info[modname] = None

    return deps_info


def show_versions():
    """Print useful debugging information"

    .. versionadded:: 0.20
    """

    sys_info = _get_sys_info()
    deps_info = _get_deps_info()

    print("\nSystem:")
    for k, stat in sys_info.items():
        print("{k:>10}: {stat}".format(k=k, stat=stat))

    print("\nPython dependencies:")
    for k, stat in deps_info.items():
        print("{k:>13}: {stat}".format(k=k, stat=stat))

    print(
        "\n{k}: {stat}".format(
            k="Built with OpenMP", stat=_openmp_parallelism_enabled()
        )
    )

    # show threadpoolctl results
    threadpool_results = threadpool_info()
    if threadpool_results:
        print()
        print("threadpoolctl info:")

        for i, result in enumerate(threadpool_results):
            for key, val in result.items():
                print(f"{key:>15}: {val}")
            if i != len(threadpool_results) - 1:
                print()