a
    !0f                     @  s*  d Z 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	m
Z
 ddlmZ ddlmZmZmZmZ ddlmZ d	d
dddZejdddddZeed	ddddZdddddZd	d	dddZd	d	ddd Zd!d	d"d#d$Zd%dd&d'd(Zd)d*d+d,d-Zd	d	d.d/d0d1Z dS )2z2
Tools for converting old- to new-style metadata.
    )annotationsN)Message)Parser)	GeneratorIterableIteratorLiteral   )Requirementstrzbool | Literal[''])r   returnc                 C  s   | o|  d S )N#)
startswith)r    r   -lib/python3.9/site-packages/wheel/metadata.py	_nonblank   s    r   zIterable[str]zIterator[str])iterabler   c                 C  s   t jtt| S )a  
    Yield valid lines of a string or iterable.
    >>> list(yield_lines(''))
    []
    >>> list(yield_lines(['foo', 'bar']))
    ['foo', 'bar']
    >>> list(yield_lines('foo\nbar'))
    ['foo', 'bar']
    >>> list(yield_lines('\nfoo\n#bar\nbaz #comment'))
    ['foo', 'baz #comment']
    >>> list(yield_lines(['foo\nbar', 'baz', 'bing\n\n\n']))
    ['foo', 'bar', 'baz', 'bing']
    )	itertoolschainfrom_iterablemapyield_lines)r   r   r   r   r      s    r   )textr   c                 C  s   t tttj|  S N)filterr   r   r   strip
splitlines)r   r   r   r   _)   s    r   zstr | Iterator[str]z3Generator[tuple[str | None, list[str]], None, None])sr   c                 c  sv   d}g }t | D ]V}|dr\|drP|s0|r:||fV  |dd  }g }qftd|q|| q||fV  dS )ar  Split a string or iterable thereof into (section, content) pairs
    Each ``section`` is a stripped version of the section header ("[section]")
    and each ``content`` is a list of stripped lines excluding blank lines and
    comment-only lines.  If there are any such lines before the first section
    header, they're returned in a first ``section`` of ``None``.
    N[]r	   zInvalid section heading)r   r   endswithr   
ValueErrorappend)r   ZsectionZcontentliner   r   r   split_sections.   s    	


r&   )extrar   c                 C  s   t dd|  S )zConvert an arbitrary string to a standard 'extra' name
    Any runs of non-alphanumeric characters are replaced with a single '_',
    and the result is always lowercased.
    z[^A-Za-z0-9.-]+r   )resublower)r'   r   r   r   
safe_extraI   s    r+   )namer   c                 C  s   t dd| S )zConvert an arbitrary string to a standard distribution name
    Any runs of non-alphanumeric/. characters are replaced with a single '-'.
    z[^A-Za-z0-9.]+-)r(   r)   )r,   r   r   r   	safe_nameQ   s    r.   r
   )requirementr   c                 C  sP   | j rd| j  S g }| jD ]}||j|j  q|rHddt| S dS dS )zFReturn the version specifier for a requirement in PEP 345/566 fashion.z @  , N)ZurlZ	specifierr$   operatorversionjoinsorted)r/   Zrequires_distspecr   r   r   requires_to_requires_distX   s    

r8   z	list[str])requirementsr   c                 c  s\   | D ]R}t |}t|}dtdd |jD }|rBd| d}t|j| | V  qdS )z=Yield Requires-Dist: strings for parsed requirements strings.r1   c                 s  s   | ]}t |V  qd S r   )r+   ).0er   r   r   	<genexpr>l       z'convert_requirements.<locals>.<genexpr>r   r    N)r
   r8   r5   r6   extrasr.   r,   )r9   ZreqZparsed_requirementr7   r>   r   r   r   convert_requirementsg   s    r?   zdict[str | None, list[str]]zIterator[tuple[str, str]])extras_requirer   c                 c  s   |   D ]\}}d}|pd}d|v r4|dd\}}t|}|rjd|fV  |rZd| d }|d| d7 }|rvd	| }t|D ]}tt|| }d
|fV  q~qdS )a,  
    Convert requirements from a setup()-style dictionary to
    ('Requires-Dist', 'requirement') and ('Provides-Extra', 'extra') tuples.

    extras_require is a dictionary of {extra: [requirements]} as passed to setup(),
    using the empty extra {'': [requirements]} to hold install_requires.
    r2   :r	   Provides-Extra(z) and z
extra == ''z ; Requires-DistN)itemssplitr+   r?   r   r
   )r@   r'   ZdependsZ	conditionZnew_reqZcanonical_reqr   r   r   generate_requirementss   s     

rH   r   )egg_info_pathpkginfo_pathr   c              
   C  sH  t |dd}t |}W d   n1 s.0    Y  |dd |d= |d= tj| d}tj|rt |dd}| }W d   n1 s0    Y  t	t
|d	d
 d}|D ]6\}}	t||	iD ] \}
}|
|f| vr|||
< qq|d }|rD|d  }d|d  td|dd df}|| |d= |S )zN
    Convert .egg-info directory with PKG-INFO to the Metadata 2.1 format
    zutf-8)encodingNzMetadata-Versionz2.1rB   rE   zrequires.txtc                 S  s   | d p
dS )Nr   r2   r   )xr   r   r   <lambda>   r=   z%pkginfo_to_metadata.<locals>.<lambda>)keyZDescription
r   r	   )openr   parseZreplace_headerospathr5   existsreadr6   r&   rH   rF   r   lstriptextwrapdedentZset_payload)rI   rJ   ZheadersZpkg_infoZrequires_pathZrequires_fileZrequiresZparsed_requirementsr'   ZreqsrN   valueZdescriptionZdescription_linesZdedented_descriptionr   r   r   pkginfo_to_metadata   s4    *&
	
rZ   )!__doc__Z
__future__r   	functoolsr   os.pathrR   r(   rW   Zemail.messager   Zemail.parserr   typingr   r   r   r   Zvendored.packaging.requirementsr
   r   singledispatchr   registerr   r   r&   r+   r.   r8   r?   rH   rZ   r   r   r   r   <module>   s,   