diff options
author | jaseg <git@jaseg.de> | 2021-02-04 23:11:55 +0100 |
---|---|---|
committer | jaseg <git@jaseg.de> | 2021-02-04 23:11:55 +0100 |
commit | 62622813beef4bebbb5dd66976a21fad7b7def30 (patch) | |
tree | a91c24cca581a5c312c1c1b862cd923502bd41dd /setup.py | |
parent | 6aab099baa7bab76d187f21b85e74848e25a8110 (diff) | |
download | gerbolyze-62622813beef4bebbb5dd66976a21fad7b7def30.tar.gz gerbolyze-62622813beef4bebbb5dd66976a21fad7b7def30.tar.bz2 gerbolyze-62622813beef4bebbb5dd66976a21fad7b7def30.zip |
setup.py: Integrate C++/usvg install
Diffstat (limited to 'setup.py')
-rwxr-xr-x | setup.py | 95 |
1 files changed, 95 insertions, 0 deletions
@@ -1,12 +1,106 @@ #!/usr/bin/env python3 +import os +import sys from setuptools import setup +from setuptools.command.install import install +import subprocess +from multiprocessing import cpu_count +from pathlib import Path def readme(): with open('README.rst') as f: return f.read() + +def get_virtualenv_path(): + """Used to work out path to install compiled binaries to.""" + if hasattr(sys, 'real_prefix'): + return sys.prefix + + if hasattr(sys, 'base_prefix') and sys.base_prefix != sys.prefix: + return sys.prefix + + if 'conda' in sys.prefix: + return sys.prefix + + return '/usr/local' + +def compile_and_install_svgflatten(): + src_path = 'svg-flatten' + + try: + subprocess.run(['make', 'check-deps'], cwd=src_path, check=True) + subprocess.run(['make', '-j', str(cpu_count()), 'all'], cwd=src_path, check=True) + subprocess.run(['make', 'install', f'PREFIX={get_virtualenv_path()}'], cwd=src_path, check=True) + except subprocess.CalledProcessError: + print('Error building svg-flatten C++ binary. Please see log above for details.', file=sys.stderr) + sys.exit(1) + +def has_usvg(): + checks = ['usvg', str(Path.home() / '.cargo' / 'bin' / 'usvg')] + if 'USVG' in os.environ: + checks = [os.environ['USVG'], *checks] + + for check in checks: + try: + subprocess.run(['usvg'], capture_output=True) + return True + + except FileNotFoundError: + pass + + else: + return False + +def install_usvg(): + try: + subprocess.run(['cargo'], check=True, capture_output=True) + + except subprocess.CalledProcessError as e: + if b'no default toolchain set' in e.stderr: + print('No rust installation found. Calling rustup.') + + try: + subprocess.run(['rustup', 'install', 'stable'], check=True) + subprocess.run(['rustup', 'default', 'stable'], check=True) + + except subprocess.FileNotFoundError as e: + print('Cannot find rustup executable. svg-flatten needs usvg, which we install via rustup. Please install rustup or install usvg manually.', file=sys.stderr) + sys.exit(1) + + except subprocess.CalledProcessError as e: + print('Error installing usvg:', e.returncode, file=sys.stderr) + sys.exit(1) + + else: + print('Error installing usvg:', e.returncode, file=sys.stderr) + print(e.stdout.decode()) + print(e.stderr.decode()) + sys.exit(1) + + except subprocess.FileNotFoundError as e: + print('Cannot find cargo executable. svg-flatten needs usvg, which we install via cargo. Please install cargo or install usvg manually.', file=sys.stderr) + sys.exit(1) + + try: + subprocess.run(['cargo', 'install', 'usvg'], check=True) + + except subprocess.CalledProcessError as e: + print('Error installing usvg:', e.returncode, file=sys.stderr) + sys.exit(1) + +class CustomInstall(install): + """Custom handler for the 'install' command.""" + def run(self): + compile_and_install_svgflatten() + if not has_usvg(): + print('usvg not found. Installing.') + install_usvg() + super().run() + setup( + cmdclass={'install': CustomInstall}, name = 'gerbolyze', version = '2.0.0', py_modules = ['gerbolyze'], @@ -40,3 +134,4 @@ setup( 'Topic :: Utilities' ] ) + |