obitools3/distutils.ext/obidistutils/core.py

'''
Created on 20 oct. 2012

@author: coissac
'''

from os import path
import os.path
import glob
import sys


from distutils.core import setup as ori_setup
from distutils.extension import Extension

from obidistutils.serenity.checkpackage import  install_requirements,\
                                                check_requirements, \
                                                RequirementError
                                                
from obidistutils.serenity.rerun import enforce_good_python
from obidistutils.serenity.rerun import rerun_with_anothe_python
from distutils import log

from obidistutils.dist import Distribution
 
from obidistutils.command.build import build
from obidistutils.command.littlebigman import littlebigman
from obidistutils.command.build_cexe import build_cexe
from obidistutils.command.build_sphinx import build_sphinx
from obidistutils.command import build_ext
from obidistutils.command.build_ctools import build_ctools
from obidistutils.command.build_files import build_files
from obidistutils.command.build_scripts import build_scripts
from obidistutils.command.install_scripts import install_scripts
from obidistutils.command.install_sphinx import install_sphinx
from obidistutils.command.install import install
from obidistutils.command.pidname import pidname
from obidistutils.command.sdist import sdist
 


def findPackage(root,base=None):
    modules=[]
    if base is None:
        base=[]
    for module in (path.basename(path.dirname(x)) 
                   for x in glob.glob(path.join(root,'*','__init__.py'))):
        modules.append('.'.join(base+[module]))
        modules.extend(findPackage(path.join(root,module),base+[module]))
    return modules
    
def findCython(root,base=None,pyrexs=None):
    setupdir = os.path.dirname(sys.argv[0])
    pyrexs=[]

    if base is None:
        base=[]
    for module in (path.basename(path.dirname(x)) 
                   for x in glob.glob(path.join(root,'*','__init__.py'))):
                       
                
        for pyrex in glob.glob(path.join(root,module,'*.pyx')):
            pyrexs.append(Extension('.'.join(base+[module,path.splitext(path.basename(pyrex))[0]]),
                                    [pyrex]
                                    )
                          )
            try:
                cfiles = os.path.splitext(pyrex)[0]+".cfiles"
                cfilesdir = os.path.dirname(cfiles)
                cfiles = open(cfiles)
                cfiles = [os.path.relpath(os.path.join(cfilesdir,y),setupdir).strip() 
                          if y[0] !='@' else y.strip()
                          for y in cfiles]
                
                log.info("Cython module : %s",cfiles)   
                incdir = set(os.path.dirname(x) for x in cfiles if x[-2:]==".h")
                cfiles = [x for x in cfiles if x[-2:]==".c"]                
                pyrexs[-1].sources.extend(cfiles)
                pyrexs[-1].include_dirs.extend(incdir)
                pyrexs[-1].extra_compile_args.extend(['-msse2'])
                
            except IOError:
                pass
            
        pyrexs.extend(findCython(path.join(root,module),base+[module]))
    return pyrexs
    

def rootname(x):
    return os.path.splitext(x.sources[0])[0]

COMMANDS = {'build':build,
            'littlebigman':littlebigman,
            'pidname':pidname,
            'build_ctools':build_ctools, 
            'build_files':build_files,
            'build_cexe':build_cexe, 
            'build_ext': build_ext,
            'build_scripts':build_scripts, 
            'build_sphinx':build_sphinx, 
            'install_scripts':install_scripts,
            'install_sphinx':install_sphinx,
            'install':install,
            'sdist':sdist}


CTOOLS =[]
CEXES  =[]
FILES  =[]

def setup(**attrs):
    
    minversion      = attrs.get("PYTHONMIN",'3.4')
    maxversion      = attrs.get('PYTHONMAX',None)    
    fork            = attrs.get('FORK',False)
    requirementfile = attrs.get('REQUIREMENTS','requirements.txt')
    
    enforce_good_python(minversion, maxversion, fork)
    
    if (install_requirements(requirementfile)):
        rerun_with_anothe_python(sys.executable,minversion,maxversion,fork)
        
    try:
        check_requirements(requirementfile)
    except RequirementError as e :
        log.error(e)                                   
        sys.exit(1)
        
    if 'distclass' not in attrs:
        attrs['distclass']=Distribution

    if 'python_src' not in attrs:
        SRC = 'python'
    else:
        SRC = attrs['python_src']
        del(attrs['python_src'])
    
    if 'scripts' not in attrs:
        attrs['scripts'] = glob.glob('%s/*.py' % SRC)

    if 'package_dir' not in attrs:
        attrs['package_dir'] = {'': SRC}

    if 'packages' not in attrs:
        attrs['packages'] = findPackage(SRC)
    
    if 'cmdclass' not in attrs:
        attrs['cmdclass'] = COMMANDS

    if 'ctools' not in attrs:
        attrs['ctools'] = CTOOLS
    
    if 'executables' not in attrs:
        attrs['executables'] = CEXES
        
    if 'files' not in attrs:
        attrs['files'] = FILES
        
    if 'sse' not in attrs:
        attrs['sse']=None
        
    if 'serenity' not in attrs:
        attrs['serenity']=False
    
    EXTENTION=findCython(SRC)
    
    if 'ext_modules' not in attrs:
        attrs['ext_modules'] = EXTENTION
    
    ori_setup(**attrs)