Updated setup.py
Mark Qvist
parent
3543dc9802
commit
b7fe01237d
149
setup.py
149
setup.py
|
|
@ -1,7 +1,11 @@
|
|||
import setuptools
|
||||
import sys
|
||||
import os
|
||||
import shutil
|
||||
import subprocess
|
||||
|
||||
pure_python = False
|
||||
native_build = False
|
||||
pure_notice = "\n\n**Warning!** *This package is the zero-dependency version of Reticulum. You should almost certainly use the [normal package](https://pypi.org/project/rns) instead. Do NOT install this package unless you know exactly why you are doing it!*"
|
||||
|
||||
if '--pure' in sys.argv:
|
||||
|
|
@ -9,6 +13,11 @@ if '--pure' in sys.argv:
|
|||
sys.argv.remove('--pure')
|
||||
print("Building pure-python wheel")
|
||||
|
||||
if '--native' in sys.argv:
|
||||
native_build = True
|
||||
sys.argv.remove('--native')
|
||||
print("Building native (compiled) wheel")
|
||||
|
||||
exec(open("RNS/_version.py", "r").read())
|
||||
with open("README.md", "r") as fh:
|
||||
long_description = fh.read()
|
||||
|
|
@ -26,6 +35,106 @@ else:
|
|||
requirements = ['cryptography>=3.4.7', 'pyserial>=3.5']
|
||||
|
||||
excluded_modules = ["tests.*", "tests"]
|
||||
|
||||
DIRECTIVES = { "annotation_typing": False }
|
||||
NATIVE_EXCLUDES = { "RNS.Interfaces.AX25KISSInterface",
|
||||
"RNS.Interfaces.I2PInterface",
|
||||
"RNS.Interfaces.KISSInterface",
|
||||
"RNS.Interfaces.RNodeInterface",
|
||||
"RNS.Interfaces.RNodeMultiInterface",
|
||||
"RNS.Interfaces.SerialInterface",
|
||||
"RNS.Interfaces.WeaveInterface",
|
||||
"RNS.Utilities.rngit",
|
||||
"RNS.Utilities.rnsd",
|
||||
"RNS.Utilities.rnsh",
|
||||
"RNS.Utilities.rncp",
|
||||
"RNS.Utilities.rnid",
|
||||
"RNS.Utilities.rnir",
|
||||
"RNS.Utilities.rnodeconf",
|
||||
"RNS.Utilities.rnpkg",
|
||||
"RNS.Utilities.rnx" }
|
||||
|
||||
ext_modules = []
|
||||
compiled_modules = set()
|
||||
build_cmdclass = {}
|
||||
|
||||
from setuptools.command.build_ext import build_ext as _build_ext
|
||||
from setuptools.command.build_py import build_py as _build_py
|
||||
if native_build:
|
||||
from setuptools import Extension
|
||||
from Cython.Build import cythonize
|
||||
|
||||
# native_opt = os.environ.get("RNS_NATIVE_OPT", "-O3")
|
||||
native_opt = os.environ.get("RNS_NATIVE_OPT", "-Os")
|
||||
if sys.platform == "win32": native_cargs = []
|
||||
else: native_cargs = [native_opt, "-g0", "-fno-lto"]
|
||||
|
||||
rns_root = "RNS"
|
||||
for root, dirs, files in os.walk(rns_root):
|
||||
dirs[:] = [d for d in dirs if d != "__pycache__"]
|
||||
for fn in files:
|
||||
if not fn.endswith(".py"): continue
|
||||
if fn == "__init__.py": continue
|
||||
path = os.path.join(root, fn).replace(os.sep, "/")
|
||||
if path.startswith("RNS/vendor/i2plib/"): continue
|
||||
if path == "RNS/vendor/configobj.py": continue
|
||||
modname = path[:-3].replace("/", ".")
|
||||
if modname in NATIVE_EXCLUDES: continue
|
||||
compiled_modules.add(modname)
|
||||
ext_modules.append(Extension(modname, [path], extra_compile_args=list(native_cargs)))
|
||||
|
||||
ext_modules = cythonize(ext_modules, language_level=3, compiler_directives=DIRECTIVES)
|
||||
|
||||
if pure_python: print("Warning: --pure is ignored when building --native")
|
||||
|
||||
# Exclude the CRNS development shim from built wheels
|
||||
excluded_modules += ["CRNS", "CRNS.*"]
|
||||
|
||||
# Trim superfluous symbol tables
|
||||
class NativeBuildExt(_build_ext):
|
||||
def run(self):
|
||||
super().run()
|
||||
if sys.platform == "win32": return
|
||||
strip = shutil.which("strip")
|
||||
if strip is None: return
|
||||
for output in self.get_outputs():
|
||||
if output.endswith((".so", ".pyd")):
|
||||
try:
|
||||
if sys.platform == "darwin": subprocess.run([strip, "-x", output], check=True)
|
||||
else: subprocess.run([strip, "--strip-unneeded", output], check=True)
|
||||
except Exception: pass
|
||||
|
||||
# Exclude sources of files that were compiled
|
||||
# to an so, and inject build info.
|
||||
class NativeBuildPy(_build_py):
|
||||
def run(self):
|
||||
super().run()
|
||||
if native_build: self._write_buildinfo()
|
||||
|
||||
def find_package_modules(self, package, package_dir):
|
||||
modules = super().find_package_modules(package, package_dir)
|
||||
if native_build:
|
||||
modules = [ (pkg, mod, fname) for (pkg, mod, fname) in modules
|
||||
if (pkg + "." + mod) not in compiled_modules ]
|
||||
return modules
|
||||
|
||||
def get_outputs(self, include_bytecode=True):
|
||||
outputs = super().get_outputs(include_bytecode)
|
||||
if native_build: outputs.append(self._buildinfo_path())
|
||||
return outputs
|
||||
|
||||
def _buildinfo_path(self):
|
||||
return os.path.join(self.build_lib, "RNS", "_buildinfo.py")
|
||||
|
||||
def _write_buildinfo(self):
|
||||
path = self._buildinfo_path()
|
||||
os.makedirs(os.path.dirname(path), exist_ok=True)
|
||||
with open(path, "w") as fh:
|
||||
fh.write("# Generated build information\n")
|
||||
fh.write("compiled = True\n")
|
||||
|
||||
if native_build: build_cmdclass = { "build_ext": NativeBuildExt, "build_py": NativeBuildPy }
|
||||
|
||||
packages = setuptools.find_packages(exclude=excluded_modules)
|
||||
|
||||
setuptools.setup(
|
||||
|
|
@ -40,29 +149,27 @@ setuptools.setup(
|
|||
packages=packages,
|
||||
license="Reticulum License",
|
||||
license_files = ("LICENSE"),
|
||||
classifiers=[
|
||||
"Programming Language :: Python :: 3",
|
||||
"Operating System :: OS Independent",
|
||||
"Development Status :: 5 - Production/Stable",
|
||||
],
|
||||
classifiers=[ "Programming Language :: Python :: 3",
|
||||
"Operating System :: OS Independent",
|
||||
"Development Status :: 5 - Production/Stable" ],
|
||||
entry_points= {
|
||||
'console_scripts': [
|
||||
'rnsd=RNS.Utilities.rnsd:main',
|
||||
'rnstatus=RNS.Utilities.rnstatus:main',
|
||||
'rnprobe=RNS.Utilities.rnprobe:main',
|
||||
'rnpath=RNS.Utilities.rnpath:main',
|
||||
'rnid=RNS.Utilities.rnid:main',
|
||||
'rncp=RNS.Utilities.rncp:main',
|
||||
'rnx=RNS.Utilities.rnx:main',
|
||||
'rnir=RNS.Utilities.rnir:main',
|
||||
'rnpkg=RNS.Utilities.rnpkg:main',
|
||||
'rnsh=RNS.Utilities.rnsh.rnsh:main',
|
||||
'rngit=RNS.Utilities.rngit.server:main',
|
||||
'rngcs=RNS.Utilities.rngit.commitsigs:main',
|
||||
'git-remote-rns=RNS.Utilities.rngit.client:main',
|
||||
'rnodeconf=RNS.Utilities.rnodeconf:main',
|
||||
]
|
||||
'console_scripts': [ 'rnsd=RNS.Utilities.rnsd:main',
|
||||
'rnstatus=RNS.Utilities.rnstatus:main',
|
||||
'rnprobe=RNS.Utilities.rnprobe:main',
|
||||
'rnpath=RNS.Utilities.rnpath:main',
|
||||
'rnid=RNS.Utilities.rnid:main',
|
||||
'rncp=RNS.Utilities.rncp:main',
|
||||
'rnx=RNS.Utilities.rnx:main',
|
||||
'rnir=RNS.Utilities.rnir:main',
|
||||
'rnpkg=RNS.Utilities.rnpkg:main',
|
||||
'rnsh=RNS.Utilities.rnsh.rnsh:main',
|
||||
'rngit=RNS.Utilities.rngit.server:main',
|
||||
'rngcs=RNS.Utilities.rngit.commitsigs:main',
|
||||
'git-remote-rns=RNS.Utilities.rngit.client:main',
|
||||
'rnodeconf=RNS.Utilities.rnodeconf:main' ]
|
||||
},
|
||||
install_requires=requirements,
|
||||
python_requires='>=3.7',
|
||||
ext_modules=ext_modules,
|
||||
cmdclass=build_cmdclass,
|
||||
)
|
||||
|
|
|
|||
Loading…
Reference in New Issue