U
    d                     @   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 5 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   9/tmp/pip-unpacked-wheel-3cscu3zg/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}|r$d|V  |  q
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   r8   V   s    

zInstaller._pkg_namesN)__name__
__module____qualname__r   r   r   r
   r*   r,   r#   r   r   staticmethodr8   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 r    )reprr'   Zegg_pathr"   r   r   r   r#   g   s    zDevelopInstaller._get_rootc                 C   s   | j S r    )egg_linkr"   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_iterabler7   r   r@   r   r   r   r   <module>   s   [