3 Ävh ã @ sR d dl Z d dlmZ d dlZd dlmZ ejjZG dd„ dƒZ G dd„ de ƒZ dS )é N)Úlog)Úmapc @ sT e Zd ZdZdd„ Zdd„ Zdd„ ZdZdZdd„ Z dd„ Z dd„ Zedd„ ƒZ dS )Ú Installerz -nspkg.pthc C sˆ | j ƒ }|sd S tjj| jƒ ƒ\}}|| j7 }| jj|ƒ tj d|ƒ t | j|ƒ}| jrdt |ƒ d S t|dƒ�}|j|ƒ W d Q R X d S )Nz Installing %sZwt)Ú_get_all_ns_packagesÚosÚpathÚsplitextÚ_get_targetÚ nspkg_extZoutputsÚappendr Úinfor Ú_gen_nspkg_lineZdry_runÚlistÚopenÚ writelines)ÚselfZnspÚfilenameÚextÚlinesÚf© r ú /usr/lib/python3.6/namespaces.pyÚinstall_namespaces s zInstaller.install_namespacesc C sH t jj| jƒ ƒ\}}|| j7 }t jj|ƒs.d S tjd|ƒ t j|ƒ d S )NzRemoving %s) r r r r r Úexistsr r Úremove)r r r r r r Úuninstall_namespaces! s zInstaller.uninstall_namespacesc C s | j S )N)Útarget)r r r r r ) s zInstaller._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 dS )Nz$sys._getframe(1).f_locals['sitedir']r )r r r r Ú _get_rootC s zInstaller._get_rootc C sV t |ƒ}t|jdƒƒ}| jƒ }| j}|jdƒ\}}}|rB|| j7 }dj|ƒtƒ d S )NÚ.ú;Ú ) ÚstrÚtupleÚsplitr' Ú_nspkg_tmplÚ rpartitionÚ_nspkg_tmpl_multiÚjoinÚlocals)r ÚpkgZpthÚrootZ tmpl_linesÚparentÚsepZchildr r r r F s zInstaller._gen_nspkg_linec C s | j jp g }ttt| j|ƒƒƒS )z,Return sorted list of all package namespaces)ZdistributionZnamespace_packagesÚsortedÚflattenr Ú _pkg_names)r Zpkgsr r r r Q s zInstaller._get_all_ns_packagesc c s, | j dƒ}x|r&dj|ƒV |jƒ qW dS )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- r1 Úpop)r3 Úpartsr r r r9 V s zInstaller._pkg_namesN) r r r r r! r" r# r$ r% )r&