U
    ڲgD*                  	   @   s  d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl	Z	ddl
mZ ddlmZmZmZ eeZdaG dd deZG dd	 d	eZG d
d deZG dd deZG dd deZG dd deZedee	jeiZzPzd dlZW n e k
r   d dl!ZY nX eeej"< eeej#< eeej$< [W n e e%fk
rV   Y nX dd Z&i Z'dd Z(e)e*dZ+dd Z,dS )    )unicode_literalsN   )DistlibException)cached_propertyget_cache_baseCachec                       s.   e Zd Zd fdd	Zdd Zdd Z  ZS )	ResourceCacheNc                    s0   |d krt jt td}tt| | d S )Nzresource-cache)ospathjoinr   strsuperr   __init__)selfbase	__class__ 5/tmp/pip-unpacked-wheel-mq592sfu/distlib/resources.pyr      s    zResourceCache.__init__c                 C   s   dS )z
        Is the cache stale for the given resource?

        :param resource: The :class:`Resource` being cached.
        :param path: The path of the resource in the cache.
        :return: True if the cache is stale.
        Tr   r   resourcer
   r   r   r   is_stale"   s    	zResourceCache.is_stalec              	   C   s   |j |\}}|dkr|}n~tj| j| ||}tj|}tj|sXt	| tj
|sjd}n| ||}|rt|d}||j W 5 Q R X |S )z
        Get a resource into the cache,

        :param resource: A :class:`Resource` instance.
        :return: The pathname of the resource in the cache.
        NTwb)finderget_cache_infor	   r
   r   r   prefix_to_dirdirnameisdirmakedirsexistsr   openwritebytes)r   r   prefixr
   resultr   stalefr   r   r   get-   s    
zResourceCache.get)N)__name__
__module____qualname__r   r   r'   __classcell__r   r   r   r   r      s   r   c                   @   s   e Zd Zdd ZdS )ResourceBasec                 C   s   || _ || _d S N)r   name)r   r   r.   r   r   r   r   H   s    zResourceBase.__init__N)r(   r)   r*   r   r   r   r   r   r,   G   s   r,   c                   @   s@   e Zd ZdZdZdd Zedd Zedd Zed	d
 Z	dS )Resourcez
    A class representing an in-package resource, such as a data file. This is
    not normally instantiated by user code, but rather by a
    :class:`ResourceFinder` which manages the resource.
    Fc                 C   s   | j | S )z
        Get the resource as a stream.

        This is not a property to make it obvious that it returns a new stream
        each time.
        )r   
get_streamr   r   r   r   	as_streamU   s    zResource.as_streamc                 C   s   t d krt a t | S r-   )cacher   r'   r1   r   r   r   	file_path^   s    zResource.file_pathc                 C   s   | j | S r-   )r   	get_bytesr1   r   r   r   r"   e   s    zResource.bytesc                 C   s   | j | S r-   )r   get_sizer1   r   r   r   sizei   s    zResource.sizeN)
r(   r)   r*   __doc__is_containerr2   r   r4   r"   r7   r   r   r   r   r/   M   s   	

r/   c                   @   s   e Zd ZdZedd ZdS )ResourceContainerTc                 C   s   | j | S r-   )r   get_resourcesr1   r   r   r   	resourcesq   s    zResourceContainer.resourcesN)r(   r)   r*   r9   r   r<   r   r   r   r   r:   n   s   r:   c                   @   s   e Zd ZdZejdrdZndZdd Zdd Z	d	d
 Z
dd Zdd Zdd Zdd Zdd Zdd Zdd Zdd ZeejjZdd ZdS )ResourceFinderz4
    Resource finder for file system resources.
    java).pyc.pyoz.class)r?   r@   c                 C   s.   || _ t|dd | _tjt|dd| _d S )N
__loader____file__ )modulegetattrloaderr	   r
   r   r   )r   rD   r   r   r   r      s    zResourceFinder.__init__c                 C   s   t j|S r-   )r	   r
   realpathr   r
   r   r   r   _adjust_path   s    zResourceFinder._adjust_pathc                 C   sB   t |trd}nd}||}|d| j tjj| }| |S )N   //r   )	
isinstancer"   splitinsertr   r	   r
   r   rI   )r   resource_nameseppartsr$   r   r   r   
_make_path   s    

zResourceFinder._make_pathc                 C   s   t j|S r-   )r	   r
   r   rH   r   r   r   _find   s    zResourceFinder._findc                 C   s
   d |j fS r-   )r
   r   r   r   r   r   r      s    zResourceFinder.get_cache_infoc                 C   sD   |  |}| |sd }n&| |r0t| |}n
t| |}||_|S r-   )rR   rS   _is_directoryr:   r/   r
   )r   rO   r
   r$   r   r   r   find   s    



zResourceFinder.findc                 C   s   t |jdS Nrb)r    r
   rT   r   r   r   r0      s    zResourceFinder.get_streamc              
   C   s,   t |jd}| W  5 Q R  S Q R X d S rW   )r    r
   read)r   r   r&   r   r   r   r5      s    zResourceFinder.get_bytesc                 C   s   t j|jS r-   )r	   r
   getsizerT   r   r   r   r6      s    zResourceFinder.get_sizec                    s*   fdd t  fddt|jD S )Nc                    s   | dko|   j S )N__pycache__)endswithskipped_extensions)r&   r1   r   r   allowed   s    
z-ResourceFinder.get_resources.<locals>.allowedc                    s   g | ]} |r|qS r   r   ).0r&   )r^   r   r   
<listcomp>   s      z0ResourceFinder.get_resources.<locals>.<listcomp>)setr	   listdirr
   rT   r   )r^   r   r   r;      s    zResourceFinder.get_resourcesc                 C   s   |  |jS r-   )rU   r
   rT   r   r   r   r9      s    zResourceFinder.is_containerc                 c   s   |  |}|d k	r|g}|r|d}|V  |jr|j}|jD ]>}|sL|}nd||g}|  |}|jrv|| q>|V  q>qd S )Nr   rK   )rV   popr9   r.   r<   r   append)r   rO   r   todornamer.   new_namechildr   r   r   iterator   s     



zResourceFinder.iteratorN)r(   r)   r*   r8   sysplatform
startswithr]   r   rI   rR   rS   r   rV   r0   r5   r6   r;   r9   staticmethodr	   r
   r   rU   ri   r   r   r   r   r=   v   s"   r=   c                       s`   e Zd ZdZ fddZdd Zdd Zdd	 Zd
d Zdd Z	dd Z
dd Zdd Z  ZS )ZipResourceFinderz6
    Resource finder for resources in .zip files.
    c                    sZ   t t| | | jj}dt| | _t| jdr>| jj| _nt	j
| | _t| j| _d S )Nr   _files)r   rn   r   rF   archivelen
prefix_lenhasattrro   	zipimport_zip_directory_cachesortedindex)r   rD   rp   r   r   r   r      s    zZipResourceFinder.__init__c                 C   s   |S r-   r   rH   r   r   r   rI      s    zZipResourceFinder._adjust_pathc                 C   s   || j d  }|| jkrd}nX|r:|d tjkr:|tj }t| j|}z| j| |}W n tk
rt   d}Y nX |st	d|| j
j nt	d|| j
j |S )NTFz_find failed: %r %rz_find worked: %r %r)rr   ro   r	   rP   bisectrw   rl   
IndexErrorloggerdebugrF   r#   )r   r
   r$   ir   r   r   rS      s    


zZipResourceFinder._findc                 C   s&   | j j}|jdt| d  }||fS )Nr   )rF   rp   r
   rq   )r   r   r#   r
   r   r   r   r      s    z ZipResourceFinder.get_cache_infoc                 C   s   | j |jS r-   )rF   get_datar
   rT   r   r   r   r5      s    zZipResourceFinder.get_bytesc                 C   s   t | |S r-   )ioBytesIOr5   rT   r   r   r   r0      s    zZipResourceFinder.get_streamc                 C   s   |j | jd  }| j| d S )N   )r
   rr   ro   r   r   r   r   r6      s    zZipResourceFinder.get_sizec                 C   s   |j | jd  }|r,|d tjkr,|tj7 }t|}t }t| j|}|t| jk r| j| |shq| j| |d  }|	|
tjdd  |d7 }qH|S )Nrx   r   r   )r
   rr   r	   rP   rq   ra   ry   rw   rl   addrM   )r   r   r
   plenr$   r}   sr   r   r   r;     s    

zZipResourceFinder.get_resourcesc                 C   sj   || j d  }|r*|d tjkr*|tj7 }t| j|}z| j| |}W n tk
rd   d}Y nX |S )Nrx   F)rr   r	   rP   ry   rw   rl   rz   )r   r
   r}   r$   r   r   r   rU     s    

zZipResourceFinder._is_directory)r(   r)   r*   r8   r   rI   rS   r   r5   r0   r6   r;   rU   r+   r   r   r   r   rn      s   rn   c                 C   s   |t t| < d S r-   )_finder_registrytype)rF   finder_makerr   r   r   register_finder2  s    r   c                 C   s   | t krt |  }nv| tjkr$t|  tj|  }t|dd}|dkrJtdt|dd}tt|}|dkrxtd|  ||}|t | < |S )z
    Return a resource finder for a package.
    :param package: The name of the package.
    :return: A :class:`ResourceFinder` instance for the package.
    __path__Nz8You cannot get a finder for a module, only for a packagerA   zUnable to locate finder for %r)	_finder_cacherj   modules
__import__rE   r   r   r'   r   )packager$   rD   r
   rF   r   r   r   r   r   9  s    


r   	__dummy__c                 C   sR   d}t |  tj| }tt|}|rNt}tj	
| d|_||_||}|S )z
    Return a resource finder for a path, which should represent a container.

    :param path: The path.
    :return: A :class:`ResourceFinder` instance for the path.
    NrC   )pkgutilget_importerrj   path_importer_cacher'   r   r   _dummy_moduler	   r
   r   rB   rA   )r
   r$   rF   r   rD   r   r   r   finder_for_pathU  s    
r   )-
__future__r   ry   r   loggingr	   r   rj   typesrt   rC   r   utilr   r   r   	getLoggerr(   r{   r3   r   objectr,   r/   r:   r=   rn   r   zipimporterr   _frozen_importlib_externalZ_fiImportError_frozen_importlibSourceFileLoader
FileFinderSourcelessFileLoaderAttributeErrorr   r   r   
ModuleTyper   r   r   r   r   r   r   <module>   sN   
,!ZO  


