File size: 5,105 Bytes
a89a522 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 | import os
import platform
import re
import shutil # Added for shutil.rmtree
import subprocess
import sys
from distutils.version import LooseVersion
from setuptools import Command, Extension, find_packages, setup # Added Command
from setuptools.command.build_ext import build_ext
class CMakeExtension(Extension):
def __init__(self, name, sourcedir=""):
Extension.__init__(self, name, sources=[])
self.sourcedir = os.path.abspath(sourcedir)
class CMakeBuild(build_ext):
def run(self):
try:
out = subprocess.check_output(["cmake", "--version"])
except OSError:
raise RuntimeError(
"CMake must be installed to build the following extensions: "
+ ", ".join(e.name for e in self.extensions)
)
if platform.system() == "Windows":
cmake_version = LooseVersion(re.search(r"version\s*([\d.]+)", out.decode()).group(1))
if cmake_version < "3.1.0":
raise RuntimeError("CMake >= 3.1.0 is required on Windows")
for ext in self.extensions:
self.build_extension(ext)
def build_extension(self, ext):
extdir = os.path.abspath(os.path.dirname(self.get_ext_fullpath(ext.name)))
# required for auto-detection of auxiliary "native" libs
if not extdir.endswith(os.path.sep):
extdir += os.path.sep
# Get pybind11 include paths
cmake_args = [
"-DCMAKE_LIBRARY_OUTPUT_DIRECTORY=" + extdir,
"-DPYTHON_EXECUTABLE=" + sys.executable,
"-DCMAKE_BUILD_TYPE=Release",
]
cfg = "Debug" if self.debug else "Release"
build_args = ["--config", cfg]
if platform.system() == "Windows":
cmake_args += ["-DCMAKE_LIBRARY_OUTPUT_DIRECTORY_{}={}".format(cfg.upper(), extdir)]
if sys.maxsize > 2**32:
cmake_args += ["-A", "x64"]
build_args += ["--", "/m"]
else:
cmake_args += ["-DCMAKE_BUILD_TYPE=" + cfg]
build_args += ["--", "-j2"] # Adjust core count as needed
env = os.environ.copy()
env["CXXFLAGS"] = '{} -DVERSION_INFO=\\"{}\\"'.format(
env.get("CXXFLAGS", ""), self.distribution.get_version()
)
if not os.path.exists(self.build_temp):
os.makedirs(self.build_temp)
subprocess.check_call(["cmake", ext.sourcedir] + cmake_args, cwd=self.build_temp, env=env)
subprocess.check_call(["cmake", "--build", "."] + build_args, cwd=self.build_temp)
# New Clean Command
class CleanCommand(Command):
"""Custom clean command to tidy up the project root."""
user_options = []
def initialize_options(self):
pass
def finalize_options(self):
pass
def run(self):
# Remove build directory
if os.path.exists("build"):
print("Removing 'build/' directory")
shutil.rmtree("build")
# Remove .egg-info directory
for item in os.listdir("."):
if item.endswith(".egg-info"):
print(f"Removing '{item}' directory")
shutil.rmtree(item)
for item in os.listdir("."):
if item.endswith(".eggs"):
print(f"Removing '{item}' directory")
shutil.rmtree(item)
# Optionally, remove dist directory if you generate distributions
if os.path.exists("dist"):
print("Removing 'dist/' directory")
shutil.rmtree("dist")
# New Uninstall Command
class UninstallCommand(Command):
"""Custom command to uninstall the package."""
user_options = []
def initialize_options(self):
pass
def finalize_options(self):
pass
def run(self):
package_name = self.distribution.get_name()
print(f"Attempting to uninstall {package_name}...")
try:
subprocess.check_call([sys.executable, "-m", "pip", "uninstall", "-y", package_name])
print(f"{package_name} uninstalled successfully.")
except subprocess.CalledProcessError as e:
print(
f"Failed to uninstall {package_name}. It may not be installed or pip uninstall failed."
)
print(f"Error: {e}")
except FileNotFoundError:
print("pip command not found. Please ensure pip is installed and in your PATH.")
setup(
name="xrobotoolkit_sdk",
version="1.0.2",
author="Zhigen Zhao",
author_email="zhigen.zhao@bytedance.com",
description="A Python binding for XRobotoolkit PC Service SDK using pybind11 and CMake",
long_description="", # Optionally, load from a README.md file
ext_modules=[CMakeExtension("xrobotoolkit_sdk")],
cmdclass=dict(
build_ext=CMakeBuild,
clean=CleanCommand, # Add clean command
uninstall=UninstallCommand, # Add uninstall command
),
zip_safe=False,
python_requires=">=3.10", # Specify your Python version requirement
packages=find_packages(), # If you have other Python packages in your project
)
|