3
e                 @   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   | j  }|sd S tjj| j \}}|| j7 }| jj| tj	d| t
| j|}| jrdt| d S t|d}|j| 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-build-argcdawy/setuptools/setuptools/namespaces.pyinstall_namespaces   s    
zInstaller.install_namespacesc             C   sH   t jj| j \}}|| j7 }t jj|s.d S tjd| t j| 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import sys, types, os#has_mfs = sys.version_info > (3, 5)$p = os.path.join(%(root)s, *%(pth)r)4importlib = has_mfs and __import__('importlib.util')-has_mfs and __import__('importlib.machinery')m = has_mfs and sys.modules.setdefault(%(pkg)r, importlib.util.module_from_spec(importlib.machinery.PathFinder.find_spec(%(pkg)r, [os.path.dirname(p)])))Cm = m or sys.modules.setdefault(%(pkg)r, types.ModuleType(%(pkg)r))7mp = (m or []) and m.__dict__.setdefault('__path__',[])(p not in mp) and mp.append(p)4m 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|jd}| j }| j}|jd\}}}|rB|| j7 }dj|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,   | j d}x|r&dj|V  |j  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)r1   r5   pop)r7   partsr   r   r   r@   V   s    

zInstaller._pkg_namesN)	r!   r"   r#   r$   r%   r&   r'   r(   r)   )r*   )__name__
__module____qualname__r   r   r   r
   r2   r4   r+   r   r   staticmethodr@   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)rD   rE   rF   r+   r
   r   r   r   r   rH   f   s   rH   )r   	distutilsr   	itertoolsZsetuptools.extern.six.movesr   chainfrom_iterabler?   r   rH   r   r   r   r   <module>   s   [