contrib/python-zstandard/setup_zstd.py
changeset 50528 9d1c56a79bb8
parent 44147 5e84a96d865b
child 52651 f066fc0bdc7a
--- a/contrib/python-zstandard/setup_zstd.py	Thu May 18 17:07:43 2023 -0700
+++ b/contrib/python-zstandard/setup_zstd.py	Mon Feb 21 14:44:22 2022 +0100
@@ -145,8 +145,16 @@
 
     include_dirs = set([os.path.join(actual_root, d) for d in ext_includes])
     if not system_zstd:
-        include_dirs.update(
-            [os.path.join(actual_root, d) for d in zstd_includes]
+        from distutils import sysconfig
+        from shlex import quote
+
+        includes = []
+        for incdir in [os.path.join(actual_root, d) for d in zstd_includes]:
+            includes.append('-I' + quote(incdir))
+            include_dirs.add(incdir)
+        config_vars = sysconfig.get_config_vars()
+        config_vars['CFLAGS'] = ' '.join(
+            includes + [config_vars.get('CFLAGS', '')]
         )
         if support_legacy:
             include_dirs.update(