diff options
Diffstat (limited to 'setup.py')
| -rw-r--r-- | setup.py | 217 |
1 files changed, 207 insertions, 10 deletions
@@ -30,10 +30,205 @@ CLASSIFIERS = [ "Topic :: Text Processing :: Markup", ] -from ez_setup import use_setuptools -use_setuptools(version='0.6c5') -from setuptools import setup, Extension, Feature +LIBYAML_CHECK = """ +#include <yaml.h> + +int main(void) { + yaml_parser_t parser; + yaml_emitter_t emitter; + + yaml_parser_initialize(&parser); + yaml_parser_delete(&parser); + + yaml_emitter_initialize(&emitter); + yaml_emitter_delete(&emitter); + + return 0; +} +""" + + +from distutils import log +from distutils.core import setup, Command +from distutils.core import Distribution as _Distribution +from distutils.core import Extension as _Extension +from distutils.dir_util import mkpath +from distutils.command.build_ext import build_ext as _build_ext +from distutils.errors import CompileError, LinkError, DistutilsPlatformError + +try: + from Pyrex.Distutils import Extension as _Extension + from Pyrex.Distutils import build_ext as _build_ext + with_pyrex = True +except ImportError: + with_pyrex = False + +import sys, os.path + + +class Distribution(_Distribution): + + def __init__(self, attrs=None): + _Distribution.__init__(self, attrs) + if not self.ext_modules: + return + for idx in range(len(self.ext_modules)-1, -1, -1): + ext = self.ext_modules[idx] + if not isinstance(ext, Extension): + continue + setattr(self, ext.attr_name, None) + self.global_options = [ + (ext.option_name, None, + "include %s (default if %s is available)" + % (ext.feature_description, ext.feature_name)), + (ext.neg_option_name, None, + "exclude %s" % ext.feature_description), + ] + self.global_options + self.negative_opt = self.negative_opt.copy() + self.negative_opt[ext.neg_option_name] = ext.option_name + + +class Extension(_Extension): + + def __init__(self, name, sources, feature_name, feature_description, + feature_check, **kwds): + if not with_pyrex: + for filename in sources[:]: + base, ext = os.path.splitext(filename) + if ext == '.pyx': + sources.remove(filename) + sources.append('%s.c' % base) + _Extension.__init__(self, name, sources, **kwds) + self.feature_name = feature_name + self.feature_description = feature_description + self.feature_check = feature_check + self.attr_name = 'with_' + feature_name.replace('-', '_') + self.option_name = 'with-' + feature_name + self.neg_option_name = 'without-' + feature_name + + +class build_ext(_build_ext): + + def run(self): + optional = True + disabled = True + for ext in self.extensions: + if isinstance(ext, Extension): + with_ext = getattr(self.distribution, ext.attr_name) + if with_ext is None: + disabled = False + elif with_ext: + optional = False + disabled = False + else: + optional = False + disabled = False + break + if disabled: + return + try: + _build_ext.run(self) + except DistutilsPlatformError, exc: + if optional: + log.warn(str(exc)) + log.warn("skipping build_ext") + else: + raise + + def get_source_files(self): + self.check_extensions_list(self.extensions) + filenames = [] + for ext in self.extensions: + if with_pyrex: + self.pyrex_sources(ext.sources, ext) + for filename in ext.sources: + filenames.append(filename) + base = os.path.splitext(filename)[0] + for ext in ['c', 'h', 'pyx', 'pxd']: + filename = '%s.%s' % (base, ext) + if filename not in filenames and os.path.isfile(filename): + filenames.append(filename) + return filenames + + def build_extensions(self): + self.check_extensions_list(self.extensions) + for ext in self.extensions: + if isinstance(ext, Extension): + with_ext = getattr(self.distribution, ext.attr_name) + if with_ext is None: + with_ext = self.check_extension_availability(ext) + if not with_ext: + continue + if with_pyrex: + ext.sources = self.pyrex_sources(ext.sources, ext) + self.build_extension(ext) + + def check_extension_availability(self, ext): + cache = os.path.join(self.build_temp, 'check_%s.out' % ext.feature_name) + if not self.force and os.path.isfile(cache): + data = open(cache).read().strip() + if data == '1': + return True + elif data == '0': + return False + mkpath(self.build_temp) + src = os.path.join(self.build_temp, 'check_%s.c' % ext.feature_name) + open(src, 'w').write(ext.feature_check) + log.info("checking if %s compiles" % ext.feature_name) + try: + [obj] = self.compiler.compile([src], + macros=ext.define_macros+[(undef,) for undef in ext.undef_macros], + include_dirs=ext.include_dirs, + extra_postargs=(ext.extra_compile_args or []), + depends=ext.depends) + except CompileError: + log.warn("%s appears not to be installed" % ext.feature_name) + log.warn("(if %s is installed, you may need to specify" + % ext.feature_name) + log.warn(" the option --include-dirs or uncomment and modify") + log.warn(" the parameter include_dirs in setup.cfg)") + open(cache, 'w').write('0\n') + return False + prog = 'check_%s' % ext.feature_name + log.info("checking if %s links" % ext.feature_name) + try: + self.compiler.link_executable([obj], prog, + output_dir=self.build_temp, + libraries=ext.libraries, + library_dirs=ext.library_dirs, + runtime_library_dirs=ext.runtime_library_dirs, + extra_postargs=(ext.extra_link_args or [])) + except LinkError: + log.warn("unable to link against %s" % ext.feature_name) + log.warn("(if %s is installed correctly, you may need to specify" + % ext.feature_name) + log.warn(" the option --library-dirs or uncomment and modify") + log.warn(" the parameter library_dirs in setup.cfg)") + open(cache, 'w').write('0\n') + return False + open(cache, 'w').write('1\n') + return True + + +class test(Command): + + user_options = [] + + def initialize_options(self): + pass + + def finalize_options(self): + pass + + def run(self): + build_cmd = self.get_finalized_command('build') + build_cmd.run() + sys.path.insert(0, build_cmd.build_lib) + sys.path.insert(0, 'tests') + import test_all + test_all.main() + if __name__ == '__main__': @@ -52,14 +247,16 @@ if __name__ == '__main__': package_dir={'': 'lib'}, packages=['yaml'], + ext_modules=[ + Extension('_yaml', ['ext/_yaml.pyx'], + 'libyaml', "LibYAML bindings", LIBYAML_CHECK, + libraries=['yaml']), + ], - features = { - 'libyaml': Feature( - description="LibYAML bindings", - ext_modules=[ - Extension('_yaml', ['ext/_yaml.pyx'], libraries=['yaml']), - ], - ), + distclass=Distribution, + cmdclass={ + 'build_ext': build_ext, +# 'test': test, }, ) |