"""This module defines custom implementation of ``egg_info`` setuptools
command."""
import os
import os.path
from setuptools.command.egg_info import egg_info as _egg_info
from . import set_build_base_mixin
from ..constants import CMAKE_INSTALL_DIR
from ..utils import new_style, to_unix_path
[docs]class egg_info(set_build_base_mixin, new_style(_egg_info)):
"""Custom implementation of ``egg_info`` setuptools command."""
[docs] def finalize_options(self, *args, **kwargs):
# pylint:disable=access-member-before-definition
if self.egg_base is None:
if self.distribution.package_dir is not None and len(self.distribution.package_dir) == 1:
# Recover directory specified in setup() function
# using `package_dir={'':<egg_base>}`
# This is required to successfully update the python path when
# running the test command.
package_name = list(self.distribution.package_dir.keys())[0]
egg_base = to_unix_path(list(self.distribution.package_dir.values())[0])
cmake_install_dir = to_unix_path(CMAKE_INSTALL_DIR())
if egg_base.startswith(cmake_install_dir):
egg_base = egg_base[len(cmake_install_dir) + 1:]
if package_name and egg_base.endswith(package_name):
egg_base = egg_base[:-len(package_name) - 1]
if egg_base == "":
egg_base = "."
# pylint:disable=attribute-defined-outside-init
self.egg_base = egg_base
else:
script_path = os.path.abspath(self.distribution.script_name)
script_dir = os.path.dirname(script_path)
# pylint:disable=attribute-defined-outside-init
self.egg_base = os.path.join(script_dir, self.egg_base)
super(egg_info, self).finalize_options(*args, **kwargs)