| """distutils.command.config |
| |
| Implements the Distutils 'config' command, a (mostly) empty command class |
| that exists mainly to be sub-classed by specific module distributions and |
| applications. The idea is that while every "config" command is different, |
| at least they're all named the same, and users always see "config" in the |
| list of standard commands. Also, this is a good place to put common |
| configure-like tasks: "try to compile this C code", or "figure out where |
| this header file lives". |
| """ |
| |
| import os, re |
| |
| from distutils.core import Command |
| from distutils.errors import DistutilsExecError |
| from distutils.sysconfig import customize_compiler |
| from distutils import log |
| |
| LANG_EXT = {"c": ".c", "c++": ".cxx"} |
| |
| class config(Command): |
| |
| description = "prepare to build" |
| |
| user_options = [ |
| ('compiler=', None, |
| "specify the compiler type"), |
| ('cc=', None, |
| "specify the compiler executable"), |
| ('include-dirs=', 'I', |
| "list of directories to search for header files"), |
| ('define=', 'D', |
| "C preprocessor macros to define"), |
| ('undef=', 'U', |
| "C preprocessor macros to undefine"), |
| ('libraries=', 'l', |
| "external C libraries to link with"), |
| ('library-dirs=', 'L', |
| "directories to search for external C libraries"), |
| |
| ('noisy', None, |
| "show every action (compile, link, run, ...) taken"), |
| ('dump-source', None, |
| "dump generated source files before attempting to compile them"), |
| ] |
| |
| |
| # The three standard command methods: since the "config" command |
| # does nothing by default, these are empty. |
| |
| def initialize_options(self): |
| self.compiler = None |
| self.cc = None |
| self.include_dirs = None |
| self.libraries = None |
| self.library_dirs = None |
| |
| # maximal output for now |
| self.noisy = 1 |
| self.dump_source = 1 |
| |
| # list of temporary files generated along-the-way that we have |
| # to clean at some point |
| self.temp_files = [] |
| |
| def finalize_options(self): |
| if self.include_dirs is None: |
| self.include_dirs = self.distribution.include_dirs or [] |
| elif isinstance(self.include_dirs, str): |
| self.include_dirs = self.include_dirs.split(os.pathsep) |
| |
| if self.libraries is None: |
| self.libraries = [] |
| elif isinstance(self.libraries, str): |
| self.libraries = [self.libraries] |
| |
| if self.library_dirs is None: |
| self.library_dirs = [] |
| elif isinstance(self.library_dirs, str): |
| self.library_dirs = self.library_dirs.split(os.pathsep) |
| |
| def run(self): |
| pass |
| |
| # Utility methods for actual "config" commands. The interfaces are |
| # loosely based on Autoconf macros of similar names. Sub-classes |
| # may use these freely. |
| |
| def _check_compiler(self): |
| """Check that 'self.compiler' really is a CCompiler object; |
| if not, make it one. |
| """ |
| # We do this late, and only on-demand, because this is an expensive |
| # import. |
| from distutils.ccompiler import CCompiler, new_compiler |
| if not isinstance(self.compiler, CCompiler): |
| self.compiler = new_compiler(compiler=self.compiler, |
| dry_run=self.dry_run, force=1) |
| customize_compiler(self.compiler) |
| if self.include_dirs: |
| self.compiler.set_include_dirs(self.include_dirs) |
| if self.libraries: |
| self.compiler.set_libraries(self.libraries) |
| if self.library_dirs: |
| self.compiler.set_library_dirs(self.library_dirs) |
| |
| def _gen_temp_sourcefile(self, body, headers, lang): |
| filename = "_configtest" + LANG_EXT[lang] |
| file = open(filename, "w") |
| if headers: |
| for header in headers: |
| file.write("#include <%s>\n" % header) |
| file.write("\n") |
| file.write(body) |
| if body[-1] != "\n": |
| file.write("\n") |
| file.close() |
| return filename |
| |
| def _preprocess(self, body, headers, include_dirs, lang): |
| src = self._gen_temp_sourcefile(body, headers, lang) |
| out = "_configtest.i" |
| self.temp_files.extend([src, out]) |
| self.compiler.preprocess(src, out, include_dirs=include_dirs) |
| return (src, out) |
| |
| def _compile(self, body, headers, include_dirs, lang): |
| src = self._gen_temp_sourcefile(body, headers, lang) |
| if self.dump_source: |
| dump_file(src, "compiling '%s':" % src) |
| (obj,) = self.compiler.object_filenames([src]) |
| self.temp_files.extend([src, obj]) |
| self.compiler.compile([src], include_dirs=include_dirs) |
| return (src, obj) |
| |
| def _link(self, body, headers, include_dirs, libraries, library_dirs, |
| lang): |
| (src, obj) = self._compile(body, headers, include_dirs, lang) |
| prog = os.path.splitext(os.path.basename(src))[0] |
| self.compiler.link_executable([obj], prog, |
| libraries=libraries, |
| library_dirs=library_dirs, |
| target_lang=lang) |
| |
| if self.compiler.exe_extension is not None: |
| prog = prog + self.compiler.exe_extension |
| self.temp_files.append(prog) |
| |
| return (src, obj, prog) |
| |
| def _clean(self, *filenames): |
| if not filenames: |
| filenames = self.temp_files |
| self.temp_files = [] |
| log.info("removing: %s", ' '.join(filenames)) |
| for filename in filenames: |
| try: |
| os.remove(filename) |
| except OSError: |
| pass |
| |
| |
| # XXX these ignore the dry-run flag: what to do, what to do? even if |
| # you want a dry-run build, you still need some sort of configuration |
| # info. My inclination is to make it up to the real config command to |
| # consult 'dry_run', and assume a default (minimal) configuration if |
| # true. The problem with trying to do it here is that you'd have to |
| # return either true or false from all the 'try' methods, neither of |
| # which is correct. |
| |
| # XXX need access to the header search path and maybe default macros. |
| |
| def try_cpp(self, body=None, headers=None, include_dirs=None, lang="c"): |
| """Construct a source file from 'body' (a string containing lines |
| of C/C++ code) and 'headers' (a list of header files to include) |
| and run it through the preprocessor. Return true if the |
| preprocessor succeeded, false if there were any errors. |
| ('body' probably isn't of much use, but what the heck.) |
| """ |
| from distutils.ccompiler import CompileError |
| self._check_compiler() |
| ok = True |
| try: |
| self._preprocess(body, headers, include_dirs, lang) |
| except CompileError: |
| ok = False |
| |
| self._clean() |
| return ok |
| |
| def search_cpp(self, pattern, body=None, headers=None, include_dirs=None, |
| lang="c"): |
| """Construct a source file (just like 'try_cpp()'), run it through |
| the preprocessor, and return true if any line of the output matches |
| 'pattern'. 'pattern' should either be a compiled regex object or a |
| string containing a regex. If both 'body' and 'headers' are None, |
| preprocesses an empty file -- which can be useful to determine the |
| symbols the preprocessor and compiler set by default. |
| """ |
| self._check_compiler() |
| src, out = self._preprocess(body, headers, include_dirs, lang) |
| |
| if isinstance(pattern, str): |
| pattern = re.compile(pattern) |
| |
| file = open(out) |
| match = False |
| while True: |
| line = file.readline() |
| if line == '': |
| break |
| if pattern.search(line): |
| match = True |
| break |
| |
| file.close() |
| self._clean() |
| return match |
| |
| def try_compile(self, body, headers=None, include_dirs=None, lang="c"): |
| """Try to compile a source file built from 'body' and 'headers'. |
| Return true on success, false otherwise. |
| """ |
| from distutils.ccompiler import CompileError |
| self._check_compiler() |
| try: |
| self._compile(body, headers, include_dirs, lang) |
| ok = True |
| except CompileError: |
| ok = False |
| |
| log.info(ok and "success!" or "failure.") |
| self._clean() |
| return ok |
| |
| def try_link(self, body, headers=None, include_dirs=None, libraries=None, |
| library_dirs=None, lang="c"): |
| """Try to compile and link a source file, built from 'body' and |
| 'headers', to executable form. Return true on success, false |
| otherwise. |
| """ |
| from distutils.ccompiler import CompileError, LinkError |
| self._check_compiler() |
| try: |
| self._link(body, headers, include_dirs, |
| libraries, library_dirs, lang) |
| ok = True |
| except (CompileError, LinkError): |
| ok = False |
| |
| log.info(ok and "success!" or "failure.") |
| self._clean() |
| return ok |
| |
| def try_run(self, body, headers=None, include_dirs=None, libraries=None, |
| library_dirs=None, lang="c"): |
| """Try to compile, link to an executable, and run a program |
| built from 'body' and 'headers'. Return true on success, false |
| otherwise. |
| """ |
| from distutils.ccompiler import CompileError, LinkError |
| self._check_compiler() |
| try: |
| src, obj, exe = self._link(body, headers, include_dirs, |
| libraries, library_dirs, lang) |
| self.spawn([exe]) |
| ok = True |
| except (CompileError, LinkError, DistutilsExecError): |
| ok = False |
| |
| log.info(ok and "success!" or "failure.") |
| self._clean() |
| return ok |
| |
| |
| # -- High-level methods -------------------------------------------- |
| # (these are the ones that are actually likely to be useful |
| # when implementing a real-world config command!) |
| |
| def check_func(self, func, headers=None, include_dirs=None, |
| libraries=None, library_dirs=None, decl=0, call=0): |
| """Determine if function 'func' is available by constructing a |
| source file that refers to 'func', and compiles and links it. |
| If everything succeeds, returns true; otherwise returns false. |
| |
| The constructed source file starts out by including the header |
| files listed in 'headers'. If 'decl' is true, it then declares |
| 'func' (as "int func()"); you probably shouldn't supply 'headers' |
| and set 'decl' true in the same call, or you might get errors about |
| a conflicting declarations for 'func'. Finally, the constructed |
| 'main()' function either references 'func' or (if 'call' is true) |
| calls it. 'libraries' and 'library_dirs' are used when |
| linking. |
| """ |
| self._check_compiler() |
| body = [] |
| if decl: |
| body.append("int %s ();" % func) |
| body.append("int main () {") |
| if call: |
| body.append(" %s();" % func) |
| else: |
| body.append(" %s;" % func) |
| body.append("}") |
| body = "\n".join(body) + "\n" |
| |
| return self.try_link(body, headers, include_dirs, |
| libraries, library_dirs) |
| |
| def check_lib(self, library, library_dirs=None, headers=None, |
| include_dirs=None, other_libraries=[]): |
| """Determine if 'library' is available to be linked against, |
| without actually checking that any particular symbols are provided |
| by it. 'headers' will be used in constructing the source file to |
| be compiled, but the only effect of this is to check if all the |
| header files listed are available. Any libraries listed in |
| 'other_libraries' will be included in the link, in case 'library' |
| has symbols that depend on other libraries. |
| """ |
| self._check_compiler() |
| return self.try_link("int main (void) { }", headers, include_dirs, |
| [library] + other_libraries, library_dirs) |
| |
| def check_header(self, header, include_dirs=None, library_dirs=None, |
| lang="c"): |
| """Determine if the system header file named by 'header_file' |
| exists and can be found by the preprocessor; return true if so, |
| false otherwise. |
| """ |
| return self.try_cpp(body="/* No body */", headers=[header], |
| include_dirs=include_dirs) |
| |
| |
| def dump_file(filename, head=None): |
| """Dumps a file content into log.info. |
| |
| If head is not None, will be dumped before the file content. |
| """ |
| if head is None: |
| log.info('%s' % filename) |
| else: |
| log.info(head) |
| file = open(filename) |
| try: |
| log.info(file.read()) |
| finally: |
| file.close() |