GitHub Viewer
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
Setup script for building clr.pyd and dependencies using mono and into
an egg or wheel.
"""
from setuptools import setup, Extension
from distutils.command.build_ext import build_ext
from distutils.command.install_lib import install_lib
from distutils.command.install_data import install_data
from distutils.sysconfig import get_config_var
from distutils.spawn import find_executable
from distutils import log
from platform import architecture
from subprocess import Popen, CalledProcessError, PIPE, check_call
from glob import glob
import fnmatch
import sys
import os
CONFIG = "Release" # Release or Debug
DEVTOOLS = "MsDev" if sys.platform == "win32" else "Mono"
VERBOSITY = "minimal" # quiet, minimal, normal, detailed, diagnostic
PLATFORM = "x64" if architecture()[0] == "64bit" else "x86"
def _find_msbuild_tool(tool="msbuild.exe", use_windows_sdk=False):
"""Return full path to one of the Microsoft build tools"""
path = find_executable(tool)
if path:
return path
try:
import _winreg
except ImportError:
import winreg as _winreg
keys_to_check = []
if use_windows_sdk:
sdks_root = r"SOFTWARE\Microsoft\Microsoft SDKs\Windows"
kits_root = r"SOFTWARE\Microsoft\Windows Kits\Installed Roots"
kits_suffix = os.path.join("bin", PLATFORM)
keys_to_check.extend([
("Windows Kit 10.0", kits_root, "KitsRoot10", kits_suffix),
("Windows Kit 8.1", kits_root, "KitsRoot81", kits_suffix),
("Windows Kit 8.0", kits_root, "KitsRoot", kits_suffix),
("Windows SDK 7.1A", sdks_root + r"\v7.1A\WinSDK-Win32Tools", "InstallationFolder"),
("Windows SDK 7.1", sdks_root + r"\v7.1\WinSDKWin32Tools", "InstallationFolder"),
("Windows SDK 7.0A", sdks_root + r"\v7.0A\WinSDK-Win32Tools", "InstallationFolder"),
("Windows SDK 7.0", sdks_root + r"\v7.0\WinSDKWin32Tools", "InstallationFolder"),
("Windows SDK 6.0A", sdks_root + r"\v6.0A\WinSDKWin32Tools", "InstallationFolder")
])
else:
vs_root = r"SOFTWARE\Microsoft\MSBuild\ToolsVersions"
keys_to_check.extend([
("MSBuild 14", vs_root + r"\14.0", "MSBuildToolsPath"),
("MSBuild 12", vs_root + r"\12.0", "MSBuildToolsPath"),
("MSBuild 4", vs_root + r"\4.0", "MSBuildToolsPath"),
("MSBuild 3.5", vs_root + r"\3.5", "MSBuildToolsPath"),
("MSBuild 2.0", vs_root + r"\2.0", "MSBuildToolsPath")
])
# read the possible tools paths from the various registry locations
paths_to_check = []
hreg = _winreg.ConnectRegistry(None, _winreg.HKEY_LOCAL_MACHINE)
try:
for key_to_check in keys_to_check:
sdk_name, key, value_name = key_to_check[:3]
suffix = key_to_check[3] if len(key_to_check) > 3 else None
hkey = None
try:
hkey = _winreg.OpenKey(hreg, key)
val, type_ = _winreg.QueryValueEx(hkey, value_name)
if type_ != _winreg.REG_SZ:
continue
if suffix:
val = os.path.join(val, suffix)
paths_to_check.append((sdk_name, val))
except WindowsError:
pass
finally:
if hkey:
hkey.Close()
finally:
hreg.Close()
# Add Visual C++ for Python as a fall-back in case one
# of the other Windows SDKs isn't installed
if use_windows_sdk:
localappdata = os.environ["LOCALAPPDATA"]
pywinsdk = localappdata + r"\Programs\Common\Microsoft\Visual C++ for Python\9.0\WinSDK\Bin"
if PLATFORM == "x64":
pywinsdk += r"\x64"
paths_to_check.append(("Visual C++ for Python", pywinsdk))
for sdk_name, path in paths_to_check:
path = os.path.join(path, tool)
if os.path.exists(path):
log.info("Using %s from %s" % (tool, sdk_name))
return path
raise RuntimeError("%s could not be found" % tool)
if DEVTOOLS == "MsDev":
_xbuild = "\"%s\"" % _find_msbuild_tool("msbuild.exe")
_defines_sep = ";"
_config = "%sWin" % CONFIG
elif DEVTOOLS == "Mono":
_xbuild = "xbuild"
_defines_sep = ","
_config = "%sMono" % CONFIG
else:
raise NotImplementedError(
"DevTools %s not supported (use MsDev or Mono)" % DEVTOOLS)
class PythonNET_BuildExt(build_ext):
def build_extension(self, ext):
"""Builds the .pyd file using msbuild or xbuild"""
if ext.name != "clr":
return build_ext.build_extension(self, ext)
# install packages using nuget
self._install_packages()
dest_file = self.get_ext_fullpath(ext.name)
dest_dir = os.path.dirname(dest_file)
if not os.path.exists(dest_dir):
os.makedirs(dest_dir)
# Up to Python 3.2 sys.maxunicode is used to determine the size of
# Py_UNICODE, but from 3.3 onwards Py_UNICODE is a typedef of wchar_t.
if sys.version_info[:2]