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
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

  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"""
+     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
## Code Before:
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
## Instruction:
Fix stray zmq import in egg_info
## Code After:
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