blob: e7919dd2aa92e178c8df360934e7d23083e68cc0 [file] [log] [blame]
Philip Howard567ee002013-03-27 22:22:00 +00001#!/usr/bin/env python
2
Mark Liffiton37ca9702017-08-13 12:52:10 -05003import os
4import sys
5
Mark Liffiton36449072017-08-12 16:47:21 -05006from setuptools import setup, Extension
Mark Liffitonf43bfce2017-08-12 23:34:01 -05007from setuptools.command.build_py import build_py
Mark Liffiton37ca9702017-08-13 12:52:10 -05008from setuptools.command.sdist import sdist
9from distutils.spawn import find_executable
Phil Howard60cc6422015-03-11 12:20:54 +000010from glob import glob
Philip Howard567ee002013-03-27 22:22:00 +000011
Phil Howard3f99ba12016-02-29 12:12:28 +000012sources = glob('WiringPi/devLib/*.c')
13sources += glob('WiringPi/wiringPi/*.c')
Mark Liffiton37ca9702017-08-13 12:52:10 -050014# If we have swig, use it. Otherwise, use the pre-generated
15# wrapper from the source distribution.
16if find_executable('swig'):
17 sources += ['wiringpi.i']
18elif os.path.exists('wiringpi_wrap.c'):
19 sources += ['wiringpi_wrap.c']
20else:
21 print("Error: Building this module requires either that swig is installed\n"
22 " (e.g., 'sudo apt install swig') or that wiringpi_wrap.c from the\n"
23 " source distribution (on pypi) is available.")
24 sys.exit(1)
Phil Howard3f99ba12016-02-29 12:12:28 +000025
Mark Liffiton36449072017-08-12 16:47:21 -050026try:
27 sources.remove('WiringPi/devLib/piFaceOld.c')
28except ValueError:
29 # the file is already excluded in the source distribution
30 pass
Phil Howard3f99ba12016-02-29 12:12:28 +000031
Mark Liffitonf43bfce2017-08-12 23:34:01 -050032
33# Fix so that build_ext runs before build_py
34# Without this, wiringpi.py is generated too late and doesn't
35# end up in the distribution when running setup.py bdist or bdist_wheel.
36# Based on:
37# https://stackoverflow.com/a/29551581/7938656
38# and
39# https://blog.niteoweb.com/setuptools-run-custom-code-in-setup-py/
Mark Liffiton37ca9702017-08-13 12:52:10 -050040class build_py_ext_first(build_py):
Mark Liffitonf43bfce2017-08-12 23:34:01 -050041 def run(self):
42 self.run_command("build_ext")
43 return build_py.run(self)
44
45
Mark Liffiton37ca9702017-08-13 12:52:10 -050046# Make sure wiringpi_wrap.c is available for the source dist, also.
47class sdist_ext_first(sdist):
48 def run(self):
49 self.run_command("build_ext")
50 return sdist.run(self)
51
52
Phil Howard77ce6cd2016-03-09 11:49:38 +000053_wiringpi = Extension(
54 '_wiringpi',
Phil Howard22041762014-07-18 11:48:01 +000055 include_dirs=['WiringPi/wiringPi','WiringPi/devLib'],
neuralassembly91b71d72017-03-29 17:25:14 +090056 sources=sources,
57 extra_link_args=['-lcrypt', '-lrt']
Philip Howard567ee002013-03-27 22:22:00 +000058)
59
60setup(
Phil Howard77ce6cd2016-03-09 11:49:38 +000061 name = 'wiringpi',
Mark Liffiton37ca9702017-08-13 12:52:10 -050062 version = '2.44.4',
Phil Howard77ce6cd2016-03-09 11:49:38 +000063 ext_modules = [ _wiringpi ],
64 py_modules = ["wiringpi"],
Philip Howard567ee002013-03-27 22:22:00 +000065 install_requires=[],
Mark Liffiton37ca9702017-08-13 12:52:10 -050066 cmdclass = {'build_py' : build_py_ext_first, 'sdist' : sdist_ext_first},
Philip Howard567ee002013-03-27 22:22:00 +000067)