a
    6Rh                     @  s   d Z ddlmZ ddlZddlmZ ddlmZ ddlm	Z	 ddl
mZ ddlmZ dd	lmZ d
dlmZ d
dlmZ edZeedf ZddddddZdddddZddddddZdd Zdd Zdd  Zd!d" Zd#d$ Zd%d& ZdS )'aS  Helper code used to generate ``requires.txt`` files in the egg-info directory.

The ``requires.txt`` file has an specific format:
    - Environment markers need to be part of the section headers and
      should not be part of the requirement spec itself.

See https://setuptools.pypa.io/en/latest/deprecated/python_eggs.html#requires-txt
    )annotationsN)defaultdict)Mapping)filterfalse)TypeVar)yield_lines)Requirement   )_reqs)
_StrOrIter_Tr   zMapping[str, _StrOrIter]z&tuple[list[str], dict[str, list[str]]])install_requiresextras_requirereturnc                 C  s   t |}t| |S )zGiven values for ``install_requires`` and ``extras_require``
    create modified versions in a way that can be written in ``requires.txt``
    )_convert_extras_requirements"_move_install_requirements_markers)r   r   extras r   U/opt/python-3.9.24/usr/lib/python3.9/site-packages/setuptools/command/_requirestxt.py_prepare   s    r   z'defaultdict[str, _Ordered[Requirement]])r   r   c                 C  sX   t ttt f t}|  D ]6\}}||  t|D ]}||t|  	| q6q|S )z
    Convert requirements in `extras_require` of the form
    `"extra": ["barbazquux; {marker}"]` to
    `"extra:{marker}": ["barbazquux"]`.
    )
r   str_Orderedr   dictitemsr
   parse_suffix_for
setdefault)r   outputsectionvrr   r   r   r   '   s    r   z#Mapping[str, _Ordered[Requirement]]c                 C  sp   t t| }tt|}tt|}t tt|}|D ]}|dt|j  	| q4t
dd | D }||fS )a+  
    The ``requires.txt`` file has an specific format:
        - Environment markers need to be part of the section headers and
          should not be part of the requirement spec itself.

    Move requirements in ``install_requires`` that are using environment
    markers ``extras_require``.
    :c              	   s  s4   | ],\}}|t td d tt|D fV  qdS )c                 s  s   | ]}t |V  qd S N)r   ).0r    r   r   r   	<genexpr>R       z?_move_install_requirements_markers.<locals>.<genexpr>.<genexpr>N)listr   fromkeysmap
_clean_req)r#   kr   r   r   r   r$   P   s   z5_move_install_requirements_markers.<locals>.<genexpr>)r&   r
   r   filter
_no_markerr   r(   r   markerr   r   r   )r   r   Z	inst_reqsZsimple_reqsZcomplex_reqsZsimple_install_requiresr    Zexpanded_extrasr   r   r   r   9   s    

r   c                 C  s   | j rdt| j  S dS )z;Return the 'extras_require' suffix for a given requirement.r!    )r-   r   reqr   r   r   r   Y   s    r   c                 C  s   t t| }d|_|S )z=Given a Requirement, remove environment markers and return itN)r   r   r-   )r0   r    r   r   r   r)   ^   s    r)   c                 C  s   | j  S r"   )r-   r/   r   r   r   r,   e   s    r,   c                 C  s,   t |pd}dd }t||}| | d S )Nr   c                 S  s   | d S )N
r   )liner   r   r   	append_crl   s    z&_write_requirements.<locals>.append_cr)r   r(   
writelines)streamreqslinesr3   r   r   r   _write_requirementsi   s    
r8   c                 C  s|   | j }t }t|jpd|jp i \}}t|| t|D ]*}|dj	f i t
  t|||  q:| d||  d S )Nr   z
[{extra}]
requirements)distributionioStringIOr   r   r   r8   sortedwriteformatvarswrite_or_delete_filegetvalue)cmdbasenamefilenamedistdatar   r   extrar   r   r   write_requirementss   s    
rI   c                 C  s,   t  }t|| jj | d||  d S )Nzsetup-requirements)r;   r<   r8   r:   Zsetup_requiresrA   rB   )rC   rD   rE   rG   r   r   r   write_setup_requirements   s    rJ   )__doc__
__future__r   r;   collectionsr   collections.abcr   	itertoolsr   typingr   Zjaraco.textr   Zpackaging.requirementsr   r.   r
   r   r   r   r   r   r   r   r   r)   r,   r8   rI   rJ   r   r   r   r   <module>   s*   	
 
