mirror of
https://github.com/msgpack/msgpack-python.git
synced 2025-10-20 04:13:16 +00:00
c8d0751fe3
The following steps have been taken: 1. Black was updated to latest version. The code has been formatted with the new version. 2. The pyupgrade utility is installed. This helped to remove all the code that was needed to support Python < 3.7. Fix #541. Co-authored-by: Inada Naoki <songofacandy@gmail.com>
93 lines
2.3 KiB
Python
Executable file
93 lines
2.3 KiB
Python
Executable file
#!/usr/bin/env python
|
|
import io
|
|
import os
|
|
import sys
|
|
from glob import glob
|
|
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):
|
|
sys.stderr.write(f"cythonize: {src!r}\n")
|
|
cython_compiler.compile([src], cplus=True)
|
|
|
|
|
|
def ensure_source(src):
|
|
pyx = os.path.splitext(src)[0] + ".pyx"
|
|
|
|
if not os.path.exists(src):
|
|
if not have_cython:
|
|
raise NoCython
|
|
cythonize(pyx)
|
|
elif os.path.exists(pyx) and os.stat(src).st_mtime < os.stat(pyx).st_mtime and have_cython:
|
|
cythonize(pyx)
|
|
return src
|
|
|
|
|
|
class BuildExt(build_ext):
|
|
def build_extension(self, ext):
|
|
try:
|
|
ext.sources = list(map(ensure_source, ext.sources))
|
|
except NoCython:
|
|
print("WARNING")
|
|
print("Cython is required for building extension from checkout.")
|
|
print("Install Cython >= 0.16 or install msgpack from PyPI.")
|
|
print("Falling back to pure Python implementation.")
|
|
return
|
|
try:
|
|
return build_ext.build_extension(self, ext)
|
|
except Exception as e:
|
|
print("WARNING: Failed to compile extension modules.")
|
|
print("msgpack uses fallback pure python implementation.")
|
|
print(e)
|
|
|
|
|
|
# Cython is required for sdist
|
|
class Sdist(sdist):
|
|
def __init__(self, *args, **kwargs):
|
|
cythonize("msgpack/_cmsgpack.pyx")
|
|
sdist.__init__(self, *args, **kwargs)
|
|
|
|
|
|
libraries = []
|
|
macros = []
|
|
|
|
if sys.platform == "win32":
|
|
libraries.append("ws2_32")
|
|
macros = [("__LITTLE_ENDIAN__", "1")]
|
|
|
|
ext_modules = []
|
|
if not PYPY and not os.environ.get("MSGPACK_PUREPYTHON"):
|
|
ext_modules.append(
|
|
Extension(
|
|
"msgpack._cmsgpack",
|
|
sources=["msgpack/_cmsgpack.cpp"],
|
|
libraries=libraries,
|
|
include_dirs=["."],
|
|
define_macros=macros,
|
|
)
|
|
)
|
|
del libraries, macros
|
|
|
|
|
|
setup(
|
|
cmdclass={"build_ext": BuildExt, "sdist": Sdist},
|
|
ext_modules=ext_modules,
|
|
packages=["msgpack"],
|
|
)
|