distribute cffi correctly

1 files changed, 35 insertions, 1 deletions

diff --git a/setup.py b/setup.py
index b5674d85..0051ea77 100644
--- a/setup.py
+++ b/setup.py
@@ -1,16 +1,39 @@
+from distutils.command.build import build
+from setuptools.command.install import install
 from setuptools import setup, find_packages
 from codecs import open
 import os
+
 from netlib import version
 
 # Based on https://github.com/pypa/sampleproject/blob/master/setup.py
 # and https://python-packaging-user-guide.readthedocs.org/
+# and https://caremad.io/2014/11/distributing-a-cffi-project/
 
 here = os.path.abspath(os.path.dirname(__file__))
 
 with open(os.path.join(here, 'README.mkd'), encoding='utf-8') as f:
     long_description = f.read()
 
+
+def get_ext_modules():
+    from netlib import certffi
+    return [certffi.xffi.verifier.get_extension()]
+
+
+class CFFIBuild(build):
+
+    def finalize_options(self):
+        self.distribution.ext_modules = get_ext_modules()
+        build.finalize_options(self)
+
+
+class CFFIInstall(install):
+
+    def finalize_options(self):
+        self.distribution.ext_modules = get_ext_modules()
+        install.finalize_options(self)
+
 setup(
     name="netlib",
     version=version.VERSION,
@@ -36,12 +59,18 @@ setup(
     ],
     packages=find_packages(),
     include_package_data=True,
+    zip_safe=False,
     install_requires=[
+        "cffi",
         "pyasn1>=0.1.7",
         "pyOpenSSL>=0.15.1",
         "cryptography>=0.9",
         "passlib>=1.6.2",
         "hpack>=1.0.1"],
+    setup_requires=[
+        "cffi",
+        "pyOpenSSL>=0.15.1",
+    ],
     extras_require={
         'dev': [
             "mock>=1.0.1",
@@ -52,4 +81,9 @@ setup(
             "autoflake>=0.6.6",
             "pathod>=%s, <%s" %
             (version.MINORVERSION,
-             version.NEXT_MINORVERSION)]})
+             version.NEXT_MINORVERSION)]},
+    cmdclass={
+        "build": CFFIBuild,
+        "install": CFFIInstall,
+    },
+)