pyunity/setup.py at develop · pyunity/pyunity · GitHub

History

171 lines (154 loc) · 6.17 KB

Raw

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

150

151

152

153

154

155

156

157

158

159

160

161

162

163

164

165

166

167

168

169

170

171

## This file is licensed under the MIT License.

## See https://docs.pyunity.x10.bz/en/latest/license.html

from setuptools import Extension, find_packages, setup

from setuptools.command.build_ext import build_ext

from setuptools.command.egg_info import egg_info, manifest_maker

from pathlib import Path

import os

import re

import sys

import glob

import json

import subprocess

try:

from wheel.bdist_wheel import bdist_wheel

except ImportError:

raise Exception("Please install `wheel` to use this script.")

if "cython" not in os.environ:

os.environ["cython"] = "1"

class SaveMeta(egg_info):

def writeVersion(self):

orig = os.getcwd()

os.chdir(Path(__file__).resolve().parent)

if not os.path.isdir(".git"):

return

# Get repo version info

p = subprocess.Popen(["git", "rev-parse", "HEAD"],

stdout=subprocess.PIPE,

stderr=subprocess.PIPE)

stdout, _ = p.communicate()

rev = stdout.decode().rstrip()

if p.returncode != 0:

rev = "unknown"

local = True

else:

p = subprocess.Popen(["git", "branch", "-r", "--contains", rev],

stdout=subprocess.PIPE,

stderr=subprocess.PIPE)

stdout, _ = p.communicate()

out = stdout.decode().rstrip()

if p.returncode != 0:

local = True

else:

local = len(out) == 0

os.chdir(orig)

data = {"revision": rev, "local": local}

self.write_or_delete_file(

"version", Path(self.egg_info) / "version.json", json.dumps(data))

def find_sources(self):

"""Generate SOURCES.txt manifest file"""

self.writeVersion()

sources = Path(self.egg_info) / "SOURCES.txt"

mm = ManifestMaker(self.distribution)

mm.manifest = str(sources)

mm.run()

self.filelist = mm.filelist

class ManifestMaker(manifest_maker):

def add_defaults(self):

super(ManifestMaker, self).add_defaults()

self.filelist.append("prepare.py")

class Cythonize(SaveMeta):

cythonized = False

@staticmethod

def cythonize(cmd):

if Cythonize.cythonized:

return

Cythonize.cythonized = True

if os.environ["cython"] == "1":

cmd.announce("Cython enabled", level=2)

if not os.path.isdir("src"):

cmd.announce(

"src/ directory not found", level=2)

args = [sys.executable, "prepare.py", "cythonize"]

cmd.announce(

"Running command: python prepare.py cythonize",

level=2)

subprocess.check_call(args)

else:

cmd.announce(

"src/ directory found", level=2)

else:

cmd.announce(

"Cython disabled", level=2)

def run(self):

Cythonize.cythonize(self)

super(Cythonize, self).run()

class ParallelBuild(build_ext):

def finalize_options(self):

super(ParallelBuild, self).finalize_options()

if self.parallel == 0:

cores = min(32, (os.cpu_count() or 1))

self.parallel = True

print(f"ParallelBuild selecting {cores} cores for compilation")

class Wheel(bdist_wheel):

def run(self):

Cythonize.cythonize(self)

super(Wheel, self).run()

if os.environ["cython"] == "1":

if not os.path.isdir("src"):

if not os.path.isfile("prepare.py"):

raise Exception("\n".join([

"Source directory `src` not found but `prepare.py` does not exist.",

"If this is not a local clone of the repository, please report",

"this as a bug at https://github.com/pyunity/pyunity/issues."

]))

if not os.path.isdir("pyunity"):

raise Exception("\n".join([

"Source directory `src` not found but `pyunity` does not exist.",

"If this is not a local clone of the repository, please report",

"this as a bug at https://github.com/pyunity/pyunity/issues."

]))

if "" not in sys.path:

sys.path.insert(0, "")

import prepare

cFiles, dataFiles = prepare.getFiles()

versionfile = "pyunity/_version.py"

else:

cFiles = glob.glob("src/**/*.c", recursive=True)

dataFiles = [a for a in glob.glob("src/**/*.*", recursive=True)

if not a.endswith(".c") and not a.endswith(".py")]

versionfile = "src/_version.py"

config = {

"command_options": {"build_ext": {

"parallel": ("setup.py", os.cpu_count())

}},

"cmdclass": {"egg_info": Cythonize, "bdist_wheel": Wheel, "build_ext": ParallelBuild},

"package_dir": {"pyunity": "src"},

"packages": ["pyunity"] + ["pyunity." + package for package in find_packages(where="pyunity")],

"ext_package": "pyunity",

"ext_modules": [Extension(file[4:-2].replace(os.path.sep, "."), [file]) for file in cFiles],

"package_data": {"pyunity": [file[4:] for file in dataFiles]},

"zip_safe": False

}

if os.getenv("PYUNITY_COMPILER", "") == "mingw":

config["command_options"]["build_ext"]["compiler"] = ("setup.py", "mingw32")

config["command_options"]["build_ext"]["define"] = ("setup.py", "MS_WIN64")

else:

dataFiles = [a for a in glob.glob("pyunity/**/*.*", recursive=True) if ".py" not in a]

config = {

"cmdclass": {"egg_info": SaveMeta},

"packages": ["pyunity"] + ["pyunity." + package for package in find_packages(where="pyunity")],

"package_data": {"pyunity": [file[8:] for file in dataFiles]},

"zip_safe": True

}

versionfile = "pyunity/_version.py"

verstrline = open(versionfile, "r").read()

versionexp = r"^versionInfo = VersionInfo\((\d+), (\d+), (\d+)"

match = re.search(versionexp, verstrline, re.M)

if match:

version = f"{match.group(1)}.{match.group(2)}.{match.group(3)}"

else:

raise RuntimeError(f"Unable to find version string in {versionfile}")

config["version"] = version

setup(**config)

pyunity/setup.py at develop · pyunity/pyunity · GitHub

Uh oh!

Expand file tree

Breadcrumbs

setup.py

Latest commit

History

Breadcrumbs

setup.py

File metadata and controls