# -*- coding: utf-8 -*-
from __future__ import absolute_import

import glob
import os
import copy

import zmq
from distutils.core import Extension

def get_extensions(**kwargs):
    """Get the Cython extensions"""
    this_directory = os.path.dirname(__file__)
    this_name = __name__.split(".")[:-1]
    
    extension_args = {
        'include_dirs' : ['numpy'] + zmq.get_includes(),
        'libraries' : [],
        'sources' : []
    }
    extension_args.update(kwargs)
    
    extensions = []
    
    for component in glob.iglob(os.path.join(this_directory, "*.pyx")):
        # Component name and full module name.
        this_extension_args = copy.deepcopy(extension_args)
        cname = os.path.splitext(os.path.basename(component))[0]
        if cname.startswith("_"):
            cname = cname[1:]
            name = ".".join(this_name + ["_{0:s}".format(cname)])
        else:
            name = ".".join(this_name + [cname])
        this_extension_args['sources'].append(component)
        
        # Extension object.
        extension = Extension(name, **this_extension_args)
        extensions.append(extension)
    
    return extensions
# -*- coding: utf-8 -*-
from __future__ import absolute_import

import glob
import os
import copy

from distutils.core import Extension

def get_extensions(**kwargs):
    """Get the Cython extensions"""
    import zmq
    
    this_directory = os.path.dirname(__file__)
    this_name = __name__.split(".")[:-1]
    
    extension_args = {
        'include_dirs' : ['numpy'] + zmq.get_includes(),
        'libraries' : [],
        'sources' : []
    }
    extension_args.update(kwargs)
    
    extensions = []
    
    for component in glob.iglob(os.path.join(this_directory, "*.pyx")):
        # Component name and full module name.
        this_extension_args = copy.deepcopy(extension_args)
        cname = os.path.splitext(os.path.basename(component))[0]
        if cname.startswith("_"):
            cname = cname[1:]
            name = ".".join(this_name + ["_{0:s}".format(cname)])
        else:
            name = ".".join(this_name + [cname])
        this_extension_args['sources'].append(component)
        
        # Extension object.
        extension = Extension(name, **this_extension_args)
        extensions.append(extension)
    
    return extensions
Fix stray zmq import in egg_info
Fix stray zmq import in egg_info
