setup.py

from distutils.core import setup, Extension, Command 
from distutils.command.build_ext import build_ext
import os
import subprocess
import sys
import shutil

#os.environ['CXX'] = 'g++'
#os.environ['ARCHFLAGS'] ="-arch x86_64"

# add this to fix build on mac osx mavericks
#os.environ['ARCHFLAGS'] ="-Wno-error=unused-command-line-argument-hard-error-in-future"

copy_args = sys.argv[1:]
AAF_ROOT = None
if '--aaf-root' in copy_args:
    index = copy_args.index('--aaf-root')
    AAF_ROOT = copy_args[index+1]
    del copy_args[index]
    del copy_args[index]
else:
    AAF_ROOT = os.environ.get("AAF_ROOT")

space = '   '
if AAF_ROOT is None:

    print space, "Unable to locate AAF Development libraries."
    print space, "Please specify with --aaf-root or AAF_ROOT env variable"
    print space, "AAF SDK can be found from http://aaf.sourceforge.net"
    print space, "Pre-built devel libraries can be found here"
    print space, "http://sourceforge.net/projects/aaf/files/AAF-devel-libs/1.1.6"
    sys.exit(-1)
    
if not os.path.exists(AAF_ROOT):
    print space, "AAF_ROOT direcotry does not exist: %s" % AAF_ROOT
    sys.exit(-1)
    
AAF_INCLUDE = os.path.join(AAF_ROOT,'include')

AAF_LIB = os.path.join(AAF_ROOT,'lib', 'debug')

AAF_COM = os.path.join(AAF_ROOT,'bin', 'debug')

ext_extra = {
    'include_dirs': ['headers',AAF_INCLUDE],
    'library_dirs': [AAF_LIB, AAF_COM],
    'libraries': ['aaflib','aafiid', 'com-api'],
}

if sys.platform.startswith('linux'):
    ext_extra['extra_link_args'] = ['-Wl,-R$ORIGIN']

WIN_ARCH = 'Win32'

if sys.platform.startswith('win'):
    import platform
    if platform.architecture()[0] == '64bit':
        WIN_ARCH = 'x64'
    if '--debug' in copy_args:
        ext_extra['library_dirs'] = [os.path.join(AAF_ROOT,WIN_ARCH ,'Debug','Refimpl')]
        ext_extra['libraries'] = ['AAFD', 'AAFIIDD']
    else:
        ext_extra['library_dirs'] = [os.path.join(AAF_ROOT,WIN_ARCH ,'Release','Refimpl')]
        ext_extra['libraries'] = ['AAF', 'AAFIID']

    ext_extra['library_dirs'].extend([os.path.join(AAF_ROOT, 'lib'),
                                        os.path.join(AAF_ROOT, 'bin')])
    
print "AAF_ROOT =",AAF_ROOT

# Construct the modules that we find in the "build/cython" directory.
ext_modules = []
build_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), 'build', 'cython'))

for dirname, dirnames, filenames in os.walk(build_dir):
    for filename in filenames:
        if filename.startswith('.') or os.path.splitext(filename)[1] != '.cpp':
            continue

        path = os.path.join(dirname, filename)
        name = os.path.splitext(os.path.relpath(path, build_dir))[0].replace(os.sep, '.')
        ext_modules.append(Extension(
            name,
            sources=[path],
            language="c++",
            **ext_extra
        ))
        
class cythonize_command(Command):
    description = "Cythonize .pyx files into c++ .cpp files"
    user_options = []

    def initialize_options(self):
        from Cython.Compiler.Main import main
        self.main = main
        
    def cythonize(self, src):

        cmd = ['--cplus']

        for item in ext_extra['include_dirs']:
            cmd.append('-I%s'% item)

        name, ext = os.path.splitext(src)
        dst = os.path.join(build_dir, name + '.cpp')

        cmd.extend(['-o', dst, src])
        cmd.insert(0,'cython')
        print subprocess.list2cmdline(cmd)

        dirname = os.path.dirname(dst)
        if not os.path.exists(dirname):
            os.makedirs(dirname)
        sys.argv = cmd
        self.main(command_line = 1)
    
    def finalize_options(self):
        pass
    def run(self):
        dest_dir = os.path.join(build_dir, 'aaf')
        for dirname, dirnames, filenames in os.walk('aaf'):
            for filename in filenames:
                if filename.startswith('.') or os.path.splitext(filename)[1] != '.pyx':
                    continue
                self.cythonize(os.path.join(dirname, filename))
        
def get_com_api(debug=True):
    if sys.platform.startswith("win"):
        dir = os.path.join(AAF_ROOT,'%s' % str(WIN_ARCH))
        if debug:
            dir = os.path.join(dir, "Debug")
        else:
            dir = os.path.join(dir, "Release")

        for dirname in ext_extra['library_dirs']:
            com_api = os.path.join(dirname, 'AAFCOAPI.dll')
            libaafintp =  os.path.join(dirname, 'aafext', 'AAFINTP.dll')
            libaafpgapi =  os.path.join(dirname, 'aafext', 'AAFPGAPI.dll')
            print com_api
            if all([os.path.exists(item) for item in (com_api,libaafintp,libaafpgapi)]):
                return com_api, libaafintp, libaafpgapi
        raise Exception("Unable to find AAFCOAPI.dll, AAFINTP.dll, AAFPGAPI.dll")
    ext = '.so'
    if sys.platform == 'darwin':
        ext = '.dylib'
    dir = os.path.join(AAF_ROOT, 'bin')
    if debug:
        dir = os.path.join(dir, 'debug')
    
    com_api = os.path.join(dir, 'libcom-api' + ext)
    libaafintp = os.path.join(dir, 'aafext', 'libaafintp' + ext)
    libaafpgapi = os.path.join(dir, 'aafext', 'libaafpgapi' + ext)
    
    return com_api, libaafintp, libaafpgapi

def copy_com_api(debug=True):
    com_api, libaafintp, libaafpgapi = get_com_api(debug)
    print  com_api, libaafintp, libaafpgapi
    
    dir = os.path.dirname(__file__)
    
    # copy libcom-api
    basename = os.path.basename(com_api)
    dest = os.path.join(dir, 'aaf', basename)
    print com_api, '->', dest
    shutil.copy(com_api, dest)
    
    # create ext dir
    aafext_dir = os.path.join(dir, 'aaf', 'aafext')
    if not os.path.exists(aafext_dir):
        print 'creating', aafext_dir
        os.makedirs(aafext_dir)
        
    # copy libaafintp
    basename = os.path.basename(libaafintp)
    intp_dest = os.path.join(aafext_dir,basename)
    print libaafintp, '->', intp_dest
    shutil.copy(libaafintp, intp_dest)
    # copy libaafpgapi
    basename = os.path.basename(libaafpgapi)
    pgapi_dest = os.path.join(aafext_dir,basename)
    print libaafpgapi, '->', pgapi_dest
    shutil.copy(libaafpgapi, pgapi_dest)
    
    return dest,intp_dest, pgapi_dest

def name_tool_fix_com_api(path):
    
    cmd = ['install_name_tool', '-id', 'libcom-api.dylib', path]
    print subprocess.list2cmdline(cmd)
    subprocess.check_call(cmd)
    
    #'install_name_tool -id libcom-api.dylib aaf/libcom-api.dylib'

def install_name_tool(path):
    cmd = ['sh','fixup_bundle.sh', path]
    subprocess.check_call(cmd)
        
class build_pyaaf_ext(build_ext):

    def build_extensions(self):
        com_api, libaafintp, libaafpgapi = copy_com_api(debug=self.debug)
        if sys.platform == 'darwin':
            name_tool_fix_com_api(com_api)
        
        build_ext.build_extensions(self)
        
        if sys.platform == 'darwin':
            for item in self.get_outputs():
                install_name_tool(item)
        print "done!"

        
com_api, libaafintp, libaafpgapi = get_com_api()
package_data = [os.path.basename(com_api)]
for item in (libaafintp, libaafpgapi):
    package_data.append(os.path.join('aafext', os.path.basename(item)))
        
package_data = {'aaf':package_data}

setup(
    script_args=copy_args,
    name='PyAAF',
    version='0.7.1',
    description='Python Bindings for the Advanced Authoring Format (AAF)',
    
    author="Mark Reid",
    author_email="mindmark@gmail.com",
    
    url="https://github.com/markreidvfx/pyaaf",
    license='MIT',
    packages=['aaf'],
    ext_modules=ext_modules,
    cmdclass = {'build_ext':build_pyaaf_ext,
                'cythonize':cythonize_command},
    package_data=package_data

)