"""CLI functions."""
import locale
import os
import pathlib
import platform
import struct
import sys
import black
import pkg_resources
import tomli
from packaging.requirements import Requirement
import evalml
from evalml.utils import get_logger
CONDA_TO_PIP_NAME = {
"python-kaleido": "kaleido",
"py-xgboost": "xgboost",
"matplotlib-base": "matplotlib",
"python-graphviz": "graphviz",
"category-encoders": "category_encoders",
}
[docs]def print_info():
"""Prints information about the system, evalml, and dependencies of evalml."""
logger = get_logger(__name__)
logger.info("EvalML version: %s" % evalml.__version__)
logger.info("EvalML installation directory: %s" % get_evalml_root())
print_sys_info()
print_deps()
[docs]def print_sys_info():
"""Prints system information."""
logger = get_logger(__name__)
logger.info("\nSYSTEM INFO")
logger.info("-----------")
sys_info = get_sys_info()
for title, stat in sys_info:
logger.info("{title}: {stat}".format(title=title, stat=stat))
[docs]def print_deps():
"""Prints the version number of each dependency."""
logger = get_logger(__name__)
logger.info("\nINSTALLED VERSIONS")
logger.info("------------------")
installed_packages = get_installed_packages()
for package, version in installed_packages.items():
logger.info("{package}: {version}".format(package=package, version=version))
# Modified from here
# https://github.com/pandas-dev/pandas/blob/d9a037ec4ad0aab0f5bf2ad18a30554c38299e57/pandas/util/_print_versions.py#L11
[docs]def get_sys_info():
"""Returns system information.
Returns:
List of tuples about system stats.
"""
blob = []
try:
(sysname, nodename, release, version, machine, processor) = platform.uname()
blob.extend(
[
("python", ".".join(map(str, sys.version_info))),
("python-bits", struct.calcsize("P") * 8),
("OS", "{sysname}".format(sysname=sysname)),
("OS-release", "{release}".format(release=release)),
("machine", "{machine}".format(machine=machine)),
("processor", "{processor}".format(processor=processor)),
("byteorder", "{byteorder}".format(byteorder=sys.byteorder)),
("LC_ALL", "{lc}".format(lc=os.environ.get("LC_ALL", "None"))),
("LANG", "{lang}".format(lang=os.environ.get("LANG", "None"))),
("LOCALE", ".".join(map(str, locale.getlocale()))),
],
)
except (KeyError, ValueError):
pass
return blob
[docs]def get_installed_packages():
"""Get dictionary mapping installed package names to their versions.
Returns:
Dictionary mapping installed package names to their versions.
"""
installed_packages = {}
for d in pkg_resources.working_set:
installed_packages[d.project_name.lower()] = d.version
return installed_packages
[docs]def get_evalml_root():
"""Gets location where evalml is installed.
Returns:
Location where evalml is installed.
"""
return os.path.dirname(evalml.__file__)
[docs]def get_evalml_pip_requirements(
evalml_path,
ignore_packages=None,
convert_to_conda=True,
):
"""Gets pip requirements for evalml (with pip packages converted to conda names)
Args:
evalml_path: Path to evalml root.
ignore_packages: List of packages to ignore. Defaults to None.
Returns:
List of pip requirements for evalml.
"""
toml_dict = None
project_metadata_filepath = pathlib.Path(evalml_path, "pyproject.toml")
with open(project_metadata_filepath, "rb") as f:
toml_dict = tomli.load(f)
packages = []
for dep in toml_dict["project"]["dependencies"]:
packages.append(Requirement(dep))
standardized_package_specifiers = standardize_format(
packages=packages,
ignore_packages=ignore_packages,
convert_to_conda=convert_to_conda,
)
return standardized_package_specifiers
[docs]def get_evalml_requirements_file(evalml_path, requirements_file_path):
"""Gets pip requirements for evalml as a requirements file
Args:
evalml_path: Path to evalml root.
requirements_file_path: Path to requirements file.
Returns:
Pip requirements for evalml in a singular string.
"""
requirements = "\n".join(
get_evalml_pip_requirements(evalml_path, convert_to_conda=False),
)
with open(requirements_file_path, "w") as text_file:
text_file.write(requirements)
return requirements
[docs]def get_evalml_black_config(
evalml_path,
):
"""Gets configuration for black.
Args:
evalml_path: Path to evalml root.
Returns:
Dictionary of black configuration.
"""
black_config = None
try:
toml_dict = None
evalml_path = pathlib.Path(evalml_path, "pyproject.toml")
with open(evalml_path, "rb") as f:
toml_dict = tomli.load(f)
black_config = toml_dict["tool"]["black"]
black_config["line_length"] = black_config.pop("line-length")
target_versions = set(
[
black.TargetVersion[ver.upper()]
for ver in black_config.pop("target-version")
],
)
black_config["target_versions"] = target_versions
except Exception:
black_config = {
"line_length": 88,
"target_versions": set([black.TargetVersion["PY39"]]),
}
return black_config
