B
    Uvg                 @   sR   d dl Z d dlmZ d dlZd dlmZ ejjZG dd dZ	G dd de	Z
dS )    N)log)mapc               @   sT   e Zd ZdZdd Zdd Zdd ZdZd	Zd
d Z	dd Z
dd Zedd ZdS )	Installerz
-nspkg.pthc          	   C   s   |   }|sd S tj|  \}}|| j7 }| j| t	d| t
| j|}| jrdt| d S t|d}|| W d Q R X d S )NzInstalling %swt)_get_all_ns_packagesospathsplitext_get_target	nspkg_extoutputsappendr   infor   _gen_nspkg_linedry_runlistopen
writelines)selfnspfilenameextlinesf r   =/tmp/pip-install-r_9ig3yj/setuptools/setuptools/namespaces.pyinstall_namespaces   s    
zInstaller.install_namespacesc             C   sH   t j|  \}}|| j7 }t j|s.d S td| t | d S )NzRemoving %s)	r   r   r	   r
   r   existsr   r   remove)r   r   r   r   r   r   uninstall_namespaces!   s    
zInstaller.uninstall_namespacesc             C   s   | j S )N)target)r   r   r   r   r
   )   s    zInstaller._get_target)	zimport sys, types, osz#has_mfs = sys.version_info > (3, 5)z$p = os.path.join(%(root)s, *%(pth)r)z4importlib = has_mfs and __import__('importlib.util')z-has_mfs and __import__('importlib.machinery')zm = has_mfs and sys.modules.setdefault(%(pkg)r, importlib.util.module_from_spec(importlib.machinery.PathFinder.find_spec(%(pkg)r, [os.path.dirname(p)])))zCm = m or sys.modules.setdefault(%(pkg)r, types.ModuleType(%(pkg)r))z7mp = (m or []) and m.__dict__.setdefault('__path__',[])z(p not in mp) and mp.append(p))z4m and setattr(sys.modules[%(parent)r], %(child)r, m)c             C   s   dS )Nz$sys._getframe(1).f_locals['sitedir']r   )r   r   r   r   	_get_rootC   s    zInstaller._get_rootc             C   sV   t |}t|d}|  }| j}|d\}}}|rB|| j7 }d|t  d S )N.;
)	strtuplesplitr!   _nspkg_tmpl
rpartition_nspkg_tmpl_multijoinlocals)r   pkgpthrootZ
tmpl_linesparentsepchildr   r   r   r   F   s    
zInstaller._gen_nspkg_linec             C   s    | j jp
g }ttt| j|S )z,Return sorted list of all package namespaces)distributionZnamespace_packagessortedflattenr   
_pkg_names)r   pkgsr   r   r   r   Q   s    zInstaller._get_all_ns_packagesc             c   s,   |  d}x|r&d|V  |  qW dS )z
        Given a namespace package, yield the components of that
        package.

        >>> names = Installer._pkg_names('a.b.c')
        >>> set(names) == set(['a', 'a.b', 'a.b.c'])
        True
        r"   N)r'   r+   pop)r-   partsr   r   r   r6   V   s    

zInstaller._pkg_namesN)__name__
__module____qualname__r   r   r   r
   r(   r*   r!   r   r   staticmethodr6   r   r   r   r   r      s   r   c               @   s   e Zd Zdd Zdd ZdS )DevelopInstallerc             C   s   t t| jS )N)reprr%   Zegg_path)r   r   r   r   r!   g   s    zDevelopInstaller._get_rootc             C   s   | j S )N)egg_link)r   r   r   r   r
   j   s    zDevelopInstaller._get_targetN)r:   r;   r<   r!   r
   r   r   r   r   r>   f   s   r>   )r   	distutilsr   	itertoolsZsetuptools.extern.six.movesr   chainfrom_iterabler5   r   r>   r   r   r   r   <module>   s   [