U
    d                     @   s  d dl Z d dlmZmZmZmZmZ ddlmZm	Z	 ddl
mZmZ eed eeef f ZedeZG dd	 d	eZG d
d deZe dZe dZeedddZddeeef eedddZeeeeeee f dddZeeeef dddZdS )    N)	FrozenSetNewTypeTupleUnioncast   )Tag	parse_tag)InvalidVersionVersion NormalizedNamec                   @   s   e Zd ZdZdS )InvalidWheelFilenamezM
    An invalid wheel filename was found, users should refer to PEP 427.
    N__name__
__module____qualname____doc__r   r   r   3/tmp/pip-unpacked-wheel-d55rnsd_/packaging/utils.pyr      s   r   c                   @   s   e Zd ZdZdS )InvalidSdistFilenamez^
    An invalid sdist filename was found, users should refer to the packaging user guide.
    Nr   r   r   r   r   r      s   r   z[-_.]+z	(\d+)(.*))namereturnc                 C   s   t d|  }tt|S )N-)_canonicalize_regexsublowerr   r   )r   valuer   r   r   canonicalize_name    s    r   T)strip_trailing_zero)versionr   r   c                C   s  t | tr2zt| }W q6 tk
r.   |  Y S X n| }g }|jdkrV||j d ddd |jD }|r~t	dd|}|| |j
dk	r|dd	d |j
D  |jdk	r|d
|j  |jdk	r|d|j  |jdk	r|d|j  d|S )z
    This is very similar to Version.__str__, but has one subtle difference
    with the way it handles the release segment.
    r   !.c                 s   s   | ]}t |V  qd S Nstr.0xr   r   r   	<genexpr>=   s     z'canonicalize_version.<locals>.<genexpr>z(\.0)+$ Nc                 s   s   | ]}t |V  qd S r"   r#   r%   r   r   r   r(   E   s     z.postz.dev+)
isinstancer$   r   r
   epochappendjoinreleaserer   prepostdevlocal)r   r   parsedpartsZrelease_segmentr   r   r   canonicalize_version&   s,    





r7   )filenamer   c           
      C   s  |  dstd|  | d d } | d}|dkrDtd|  | d|d }|d }d	|ksxtd
|tjd krtd|  t|}t|d }|dkr|d }t	|}|d krtd| d|  dt
tt|d|df}nd}t|d }	||||	fS )Nz.whlz3Invalid wheel filename (extension must be '.whl'): r   )      z0Invalid wheel filename (wrong number of parts):    r   __z^[\w\d._]*$zInvalid project name: r   r;   zInvalid build number: z in ''r   )endswithr   countsplitr0   matchUNICODEr   r   _build_tag_regexr   BuildTagintgroupr	   )
r8   Zdashesr6   	name_partr   r   Z
build_partZbuild_matchbuildtagsr   r   r   parse_wheel_filenameV   s6    


 rL   c                 C   s   |  dr| d td  }n,|  dr<| d td  }ntd|  |d\}}}|sltd|  t|}t|}||fS )Nz.tar.gzz.zipz@Invalid sdist filename (extension must be '.tar.gz' or '.zip'): r   zInvalid sdist filename: )r@   lenr   
rpartitionr   r   )r8   Z	file_stemrI   sepZversion_partr   r   r   r   r   parse_sdist_filenamez   s    

rP   )r0   typingr   r   r   r   r   rK   r   r	   r   r
   r   rG   r$   rF   r   
ValueErrorr   r   compiler   rE   r   boolr7   rL   rP   r   r   r   r   <module>   s(   



 1$