mirror of https://github.com/zeromq/pyzmq.git
97 lines
3.0 KiB
Python
97 lines
3.0 KiB
Python
"""Detect zmq version"""
|
|
#-----------------------------------------------------------------------------
|
|
# Copyright (C) 2011 Brian Granger, Min Ragan-Kelley
|
|
#
|
|
# This file is part of pyzmq, copied and adapted from h5py.
|
|
# h5py source used under the New BSD license
|
|
#
|
|
# h5py: <http://code.google.com/p/h5py/>
|
|
#
|
|
# Distributed under the terms of the New BSD License. The full license is in
|
|
# the file COPYING.BSD, distributed as part of this software.
|
|
#-----------------------------------------------------------------------------
|
|
|
|
import shutil
|
|
import sys
|
|
import os
|
|
import logging
|
|
import platform
|
|
from distutils import ccompiler
|
|
from distutils.sysconfig import customize_compiler
|
|
from subprocess import Popen, PIPE
|
|
|
|
pjoin = os.path.join
|
|
|
|
#-----------------------------------------------------------------------------
|
|
# Utility functions (adapted from h5py: http://h5py.googlecode.com)
|
|
#-----------------------------------------------------------------------------
|
|
|
|
def detect_zmq(basedir, **compiler_attrs):
|
|
"""Compile, link & execute a test program, in empty directory `basedir`.
|
|
|
|
The C compiler will be updated with any keywords given via setattr.
|
|
|
|
Parameters
|
|
----------
|
|
|
|
basedir : path
|
|
The location where the test program will be compiled and run
|
|
**compiler_attrs : dict
|
|
Any extra compiler attributes, which will be set via ``setattr(cc)``.
|
|
|
|
Returns
|
|
-------
|
|
|
|
A dict of properties for zmq compilation, with the following two keys:
|
|
|
|
vers : tuple
|
|
The ZMQ version as a tuple of ints, e.g. (2,2,0)
|
|
settings : dict
|
|
The compiler options used to compile the test function, e.g. `include_dirs`,
|
|
`library_dirs`, `libs`, etc.
|
|
"""
|
|
|
|
cc = ccompiler.new_compiler()
|
|
customize_compiler(cc)
|
|
for name, val in compiler_attrs.items():
|
|
setattr(cc, name, val)
|
|
|
|
cfile = pjoin(basedir, 'vers.c')
|
|
efile = pjoin(basedir, 'vers')
|
|
|
|
shutil.copy(pjoin(os.path.dirname(__file__), 'vers.c'), cfile)
|
|
|
|
cpreargs = lpreargs = None
|
|
if sys.platform == 'darwin':
|
|
# use appropriate arch for compiler
|
|
if platform.architecture()[0]=='32bit':
|
|
cpreargs = ['-arch','i386']
|
|
lpreargs = ['-arch', 'i386', '-undefined', 'dynamic_lookup']
|
|
else:
|
|
# allow for missing UB arch, since it will still work:
|
|
lpreargs = ['-undefined', 'dynamic_lookup']
|
|
|
|
objs = cc.compile([cfile],extra_preargs=cpreargs)
|
|
cc.link_executable(objs, efile, extra_preargs=lpreargs)
|
|
|
|
result = Popen(efile, stdout=PIPE, stderr=PIPE)
|
|
so, se = result.communicate()
|
|
# for py3k:
|
|
so = so.decode()
|
|
se = se.decode()
|
|
if result.returncode:
|
|
msg = "Error running version detection script:\n%s\n%s" % (so,se)
|
|
logging.error(msg)
|
|
raise IOError(msg)
|
|
|
|
handlers = {'vers': lambda val: tuple(int(v) for v in val.split('.'))}
|
|
|
|
props = {}
|
|
for line in (x for x in so.split('\n') if x):
|
|
key, val = line.split(':')
|
|
props[key] = handlers[key](val)
|
|
|
|
props['settings'] = compiler_attrs
|
|
return props
|
|
|