123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142 |
- """
- Compile a Python script into an executable that embeds CPython and run it.
- Requires CPython to be built as a shared library ('libpythonX.Y').
- Basic usage:
- python cythonrun somefile.py [ARGS]
- """
- from __future__ import absolute_import
- DEBUG = True
- import sys
- import os
- from distutils import sysconfig
- def get_config_var(name, default=''):
- return sysconfig.get_config_var(name) or default
- INCDIR = sysconfig.get_python_inc()
- LIBDIR1 = get_config_var('LIBDIR')
- LIBDIR2 = get_config_var('LIBPL')
- PYLIB = get_config_var('LIBRARY')
- PYLIB_DYN = get_config_var('LDLIBRARY')
- if PYLIB_DYN == PYLIB:
- # no shared library
- PYLIB_DYN = ''
- else:
- PYLIB_DYN = os.path.splitext(PYLIB_DYN[3:])[0] # 'lib(XYZ).so' -> XYZ
- CC = get_config_var('CC', os.environ.get('CC', ''))
- CFLAGS = get_config_var('CFLAGS') + ' ' + os.environ.get('CFLAGS', '')
- LINKCC = get_config_var('LINKCC', os.environ.get('LINKCC', CC))
- LINKFORSHARED = get_config_var('LINKFORSHARED')
- LIBS = get_config_var('LIBS')
- SYSLIBS = get_config_var('SYSLIBS')
- EXE_EXT = sysconfig.get_config_var('EXE')
- def _debug(msg, *args):
- if DEBUG:
- if args:
- msg = msg % args
- sys.stderr.write(msg + '\n')
- def dump_config():
- _debug('INCDIR: %s', INCDIR)
- _debug('LIBDIR1: %s', LIBDIR1)
- _debug('LIBDIR2: %s', LIBDIR2)
- _debug('PYLIB: %s', PYLIB)
- _debug('PYLIB_DYN: %s', PYLIB_DYN)
- _debug('CC: %s', CC)
- _debug('CFLAGS: %s', CFLAGS)
- _debug('LINKCC: %s', LINKCC)
- _debug('LINKFORSHARED: %s', LINKFORSHARED)
- _debug('LIBS: %s', LIBS)
- _debug('SYSLIBS: %s', SYSLIBS)
- _debug('EXE_EXT: %s', EXE_EXT)
- def runcmd(cmd, shell=True):
- if shell:
- cmd = ' '.join(cmd)
- _debug(cmd)
- else:
- _debug(' '.join(cmd))
- try:
- import subprocess
- except ImportError: # Python 2.3 ...
- returncode = os.system(cmd)
- else:
- returncode = subprocess.call(cmd, shell=shell)
- if returncode:
- sys.exit(returncode)
- def clink(basename):
- runcmd([LINKCC, '-o', basename + EXE_EXT, basename+'.o', '-L'+LIBDIR1, '-L'+LIBDIR2]
- + [PYLIB_DYN and ('-l'+PYLIB_DYN) or os.path.join(LIBDIR1, PYLIB)]
- + LIBS.split() + SYSLIBS.split() + LINKFORSHARED.split())
- def ccompile(basename):
- runcmd([CC, '-c', '-o', basename+'.o', basename+'.c', '-I' + INCDIR] + CFLAGS.split())
- def cycompile(input_file, options=()):
- from ..Compiler import Version, CmdLine, Main
- options, sources = CmdLine.parse_command_line(list(options or ()) + ['--embed', input_file])
- _debug('Using Cython %s to compile %s', Version.version, input_file)
- result = Main.compile(sources, options)
- if result.num_errors > 0:
- sys.exit(1)
- def exec_file(program_name, args=()):
- runcmd([os.path.abspath(program_name)] + list(args), shell=False)
- def build(input_file, compiler_args=(), force=False):
- """
- Build an executable program from a Cython module.
- Returns the name of the executable file.
- """
- basename = os.path.splitext(input_file)[0]
- exe_file = basename + EXE_EXT
- if not force and os.path.abspath(exe_file) == os.path.abspath(input_file):
- raise ValueError("Input and output file names are the same, refusing to overwrite")
- if (not force and os.path.exists(exe_file) and os.path.exists(input_file)
- and os.path.getmtime(input_file) <= os.path.getmtime(exe_file)):
- _debug("File is up to date, not regenerating %s", exe_file)
- return exe_file
- cycompile(input_file, compiler_args)
- ccompile(basename)
- clink(basename)
- return exe_file
- def build_and_run(args):
- """
- Build an executable program from a Cython module and runs it.
- Arguments after the module name will be passed verbatimely to the
- program.
- """
- cy_args = []
- last_arg = None
- for i, arg in enumerate(args):
- if arg.startswith('-'):
- cy_args.append(arg)
- elif last_arg in ('-X', '--directive'):
- cy_args.append(arg)
- else:
- input_file = arg
- args = args[i+1:]
- break
- last_arg = arg
- else:
- raise ValueError('no input file provided')
- program_name = build(input_file, cy_args)
- exec_file(program_name, args)
- if __name__ == '__main__':
- build_and_run(sys.argv[1:])
|