File: //lib/python3/dist-packages/sphinx/util/__pycache__/inventory.cpython-310.pyc
o
�$�a � @ s� d Z ddlZddlZddlZddlmZmZmZmZ ddl m
Z
ddl m Z dZ
e
�e�Zer;ddlmZ ddlmZ G d d
� d
�ZG d d � d �ZdS )
z�
sphinx.util.inventory
~~~~~~~~~~~~~~~~~~~~~
Inventory utility functions for Sphinx.
:copyright: Copyright 2007-2021 by the Sphinx team, see AUTHORS.
:license: BSD, see LICENSE for details.
� N)�IO�
TYPE_CHECKING�Callable�Iterator)�logging)� Inventoryi @ )�Builder)�BuildEnvironmentc @ sp e Zd ZdZdeddfdd�Zddd�Zdefd d
�Zde e fd d �Z
de e fd
d�Z de e fdd�Z
dS )�InventoryFileReaderzjA file reader for an inventory file.
This reader supports mixture of texts and compressed texts.
�stream�returnNc C s || _ d| _d| _d S )N� F)r �buffer�eof)�selfr � r �7/usr/lib/python3/dist-packages/sphinx/util/inventory.py�__init__ s
z InventoryFileReader.__init__c C s, | j �t�}|dkr
d| _| j|7 _d S )Nr
T)r �read�BUFSIZEr r )r �chunkr r r � read_buffer% s z InventoryFileReader.read_bufferc C sl | j �d�}|dkr | j d |� �� }| j |d d � | _ |S | jr,| j �� }d| _ |S | �� | �� }|S )N�
���� r
)r �find�decoder r �readline)r �pos�liner r r r + s �
�z InventoryFileReader.readlinec c s( � | j s| �� }|r
|V | j rd S d S )N)r r )r r r r r � readlines9 s ��z InventoryFileReader.readlinesc c s@ � t �� }| js| �� |�| j�V d| _| jr|�� V d S )Nr
)�zlib�
decompressobjr r �
decompressr �flush)r � decompressorr r r �read_compressed_chunks? s ��z*InventoryFileReader.read_compressed_chunksc c sf � d}| � � D ])}||7 }|�d�}|dkr0|d |� �� V ||d d � }|�d�}|dksqd S )Nr
r r r )r& r r )r �bufr r r r r �read_compressed_linesG s �
���z)InventoryFileReader.read_compressed_lines)r N)�__name__�
__module__� __qualname__�__doc__r r r �strr r r �bytesr& r( r r r r r
s
r
c @ s� e Zd Zededededefdd��Zede dededefdd ��Z
ede dededefd
d ��Z ed ed
dddddfdd��Z dS )�
InventoryFiler �uri�joinfuncr c C sL t |�}|�� �� }|dkr| �|||�S |dkr | �|||�S td| ��)Nz # Sphinx inventory version 1z # Sphinx inventory version 2z invalid inventory header: %s)r
r �rstrip�load_v1�load_v2�
ValueError)�clsr r0 r1 �readerr r r r �loadS s zInventoryFile.load�joinc C s� i }|� � �� dd � }|� � �� dd � }|�� D ]5}|�� �d d�\}} }
|||
�}
| dkr9d} |
d| 7 }
n
d| } |
d| 7 }
|||
df|�| i �|< q|S ) N� � �mod� py:modulez#module-zpy:�#�-)r r2 r �split�
setdefault) r6 r r0 r9 �invdata�projname�versionr �name�type�locationr r r r3 ^ s
zInventoryFile.load_v1c C s� i }|� � �� dd � }|� � �� dd � }|� � }d|vr$td| ��|�� D ]H}t�d|�� �}|s5q(|�� \} }
} } }
d|
vrCq(|
dkrR|
|v rR| ||
v rRq(| �d�r_| d d� | } ||| �} ||| |
f|�|
i �| < q(|S ) Nr: r! z-invalid inventory header (not compressed): %sz+(?x)(.+?)\s+(\S+)\s+(-?\d+)\s+?(\S*)\s+(.*)�:r= �$r ) r r2 r5 r( �re�match�groups�endswithrA )r6 r r0 r9 rB rC rD r �mrE rF �priorG �dispnamer r r r4 p s0 �
�zInventoryFile.load_v2�filename�envr �builderr Nc
C s, dt dt fdd�}ttj�|�d��z}|�d||jj�||jj�f � � � t
� d�}t |j
�� �D ]J\}}t |�� �D ]?\} }
} } }
}|
�| �rT|
d t| � � d }
|�| �}|
ra|d |
7 }|
| krgd
}
d | || |||
f }|�|�|� � �� q<q2|�|�� � W d � d S 1 s�w Y d S ) N�stringr c S s t �dd| �S )Nz\s+� )rJ �sub)rT r r r �escape� s z"InventoryFile.dump.<locals>.escape�wbzp# Sphinx inventory version 2
# Project: %s
# Version: %s
# The remainder of this file is compressed using zlib.
� rI r> r? z%s %s:%s %s %s %s
)r- �open�os�pathr9 �write�config�projectrD �encoder! � compressobj�sorted�domains�items� get_objectsrM �len�get_target_uri�compressr$ )r6 rQ rR rS rW �f�
compressor�
domainname�domainrE rP �typ�docname�anchorrO r0 �entryr r r �dump� s6
���
�
��
"�zInventoryFile.dump)
r) r* r+ � classmethodr r- r r r8 r
r3 r4 rq r r r r r/ R s
! r/ )r, r[ rJ r! �typingr r r r � sphinx.utilr �sphinx.util.typingr r � getLoggerr) �logger�sphinx.buildersr �sphinx.environmentr r
r/ r r r r �<module> s
8