mirror of
https://github.com/msgpack/msgpack-python.git
synced 2025-11-04 03:20:56 +00:00
77 lines
1.7 KiB
Python
77 lines
1.7 KiB
Python
#!/usr/bin/env python
|
|
import os
|
|
import sys
|
|
from setuptools import setup, Extension
|
|
from setuptools.command.build_ext import build_ext
|
|
from setuptools.command.sdist import sdist
|
|
|
|
|
|
PYPY = hasattr(sys, "pypy_version_info")
|
|
|
|
|
|
class NoCython(Exception):
|
|
pass
|
|
|
|
|
|
try:
|
|
import Cython.Compiler.Main as cython_compiler
|
|
|
|
have_cython = True
|
|
except ImportError:
|
|
have_cython = False
|
|
|
|
|
|
def cythonize(src):
|
|
if not have_cython:
|
|
raise Exception("Cython is required for building from checkout")
|
|
sys.stderr.write(f"cythonize: {src!r}\n")
|
|
cython_compiler.compile([src])
|
|
|
|
|
|
def ensure_source(src):
|
|
pyx = os.path.splitext(src)[0] + ".pyx"
|
|
|
|
if not os.path.exists(src) or have_cython and os.stat(src).st_mtime < os.stat(pyx).st_mtime:
|
|
cythonize(pyx)
|
|
|
|
|
|
class BuildExt(build_ext):
|
|
def build_extension(self, ext):
|
|
for src in ext.sources:
|
|
ensure_source(src)
|
|
return build_ext.build_extension(self, ext)
|
|
|
|
|
|
# Cython is required for sdist
|
|
class Sdist(sdist):
|
|
def __init__(self, *args, **kwargs):
|
|
cythonize("msgpack/_cmsgpack.pyx")
|
|
sdist.__init__(self, *args, **kwargs)
|
|
|
|
|
|
libraries = []
|
|
macros = []
|
|
ext_modules = []
|
|
|
|
if sys.platform == "win32":
|
|
libraries.append("ws2_32")
|
|
macros = [("__LITTLE_ENDIAN__", "1")]
|
|
|
|
if not PYPY and not os.environ.get("MSGPACK_PUREPYTHON"):
|
|
ext_modules.append(
|
|
Extension(
|
|
"msgpack._cmsgpack",
|
|
sources=["msgpack/_cmsgpack.c"],
|
|
libraries=libraries,
|
|
include_dirs=["."],
|
|
define_macros=macros,
|
|
)
|
|
)
|
|
del libraries, macros
|
|
|
|
|
|
setup(
|
|
cmdclass={"build_ext": BuildExt, "sdist": Sdist},
|
|
ext_modules=ext_modules,
|
|
packages=["msgpack"],
|
|
)
|