blob: b1aa9ac661997cb79acb74e08061ae6a646cb2ae [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')
Joshua Yang8d7c2232018-02-22 13:52:33 +090014
15# Exclude wiringPi.c. This setup is only for ODROID WiringPi.
16sources = list(set(sources) - set(glob('WiringPi/wiringPi/wiringPi.c')))
17# Exclude template file.
18sources = list(set(sources) - set(glob('WiringPi/wiringPi/odroid_template.c')))
19
Mark Liffiton37ca9702017-08-13 12:52:10 -050020# If we have swig, use it. Otherwise, use the pre-generated
21# wrapper from the source distribution.
22if find_executable('swig'):
23 sources += ['wiringpi.i']
24elif os.path.exists('wiringpi_wrap.c'):
25 sources += ['wiringpi_wrap.c']
26else:
27 print("Error: Building this module requires either that swig is installed\n"
28 " (e.g., 'sudo apt install swig') or that wiringpi_wrap.c from the\n"
29 " source distribution (on pypi) is available.")
30 sys.exit(1)
Phil Howard3f99ba12016-02-29 12:12:28 +000031
Mark Liffiton36449072017-08-12 16:47:21 -050032try:
33 sources.remove('WiringPi/devLib/piFaceOld.c')
34except ValueError:
35 # the file is already excluded in the source distribution
36 pass
Phil Howard3f99ba12016-02-29 12:12:28 +000037
Mark Liffitonf43bfce2017-08-12 23:34:01 -050038
39# Fix so that build_ext runs before build_py
40# Without this, wiringpi.py is generated too late and doesn't
41# end up in the distribution when running setup.py bdist or bdist_wheel.
42# Based on:
43# https://stackoverflow.com/a/29551581/7938656
44# and
45# https://blog.niteoweb.com/setuptools-run-custom-code-in-setup-py/
Mark Liffiton37ca9702017-08-13 12:52:10 -050046class build_py_ext_first(build_py):
Mark Liffitonf43bfce2017-08-12 23:34:01 -050047 def run(self):
48 self.run_command("build_ext")
49 return build_py.run(self)
50
51
Mark Liffiton37ca9702017-08-13 12:52:10 -050052# Make sure wiringpi_wrap.c is available for the source dist, also.
53class sdist_ext_first(sdist):
54 def run(self):
55 self.run_command("build_ext")
56 return sdist.run(self)
57
58
Phil Howard77ce6cd2016-03-09 11:49:38 +000059_wiringpi = Extension(
60 '_wiringpi',
Phil Howard22041762014-07-18 11:48:01 +000061 include_dirs=['WiringPi/wiringPi','WiringPi/devLib'],
neuralassembly91b71d72017-03-29 17:25:14 +090062 sources=sources,
63 extra_link_args=['-lcrypt', '-lrt']
Philip Howard567ee002013-03-27 22:22:00 +000064)
65
66setup(
Phil Howard77ce6cd2016-03-09 11:49:38 +000067 name = 'wiringpi',
Mark Liffiton37ca9702017-08-13 12:52:10 -050068 version = '2.44.4',
Phil Howard77ce6cd2016-03-09 11:49:38 +000069 ext_modules = [ _wiringpi ],
70 py_modules = ["wiringpi"],
Philip Howard567ee002013-03-27 22:22:00 +000071 install_requires=[],
Mark Liffiton37ca9702017-08-13 12:52:10 -050072 cmdclass = {'build_py' : build_py_ext_first, 'sdist' : sdist_ext_first},
Philip Howard567ee002013-03-27 22:22:00 +000073)