File: //usr/lib/python3/dist-packages/sphinx/environment/collectors/__pycache__/toctree.cpython-310.pyc
o
�$�a�3 � @ s� d Z ddlmZmZmZmZmZmZmZm Z ddl
m Z ddl m
Z
mZ ddlmZ ddlmZ ddlmZ ddlmZ dd lmZ dd
lmZ dd lm Z dd l m Z m Z ed
�Z e �!e"�Z#G dd� de�Z$dedee%ef fdd�Z&dS )z�
sphinx.environment.collectors.toctree
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Toctree collector for sphinx.environment.
:copyright: Copyright 2007-2021 by the Sphinx team, see AUTHORS.
:license: BSD, see LICENSE for details.
� )�Any�Dict�List�Set�Tuple�Type�TypeVar�cast)�nodes)�Element�Node)�addnodes)�Sphinx)�BuildEnvironment)�TocTree)�EnvironmentCollector)�__)�SphinxContentsFilter)�logging�url_re�Nc
@ s� e Zd Zdedededdfdd�Zdededee d eddf
d
d �Zded e j
ddfd
d�Z dedede e fdd�Z
dede e fdd�Zdede e fdd�ZdS )�TocTreeCollector�app�env�docname�returnNc C s� |j �|d � |j�|d � |j�|d � |j�|d � |j�|d � |j�|� |j�|� t |j
� � �D ]\}}|�|� |sE|j
|= q6d S �N) �tocs�pop�toc_secnumbers�toc_fignumbers�toc_num_entries�toctree_includes�
glob_toctrees�discard�numbered_toctrees�list�files_to_rebuild�items)�selfr r r �subfn�fnset� r, �G/usr/lib/python3/dist-packages/sphinx/environment/collectors/toctree.py� clear_doc s
��zTocTreeCollector.clear_doc�docnames�otherc C s� |D ]5}|j | |j |< |j| |j|< ||jv r!|j| |j|< ||jv r,|j�|� ||jv r7|j�|� q|j�� D ]\}}|j�|t � ��
|t |�@ � q=d S r ) r r! r" r# �addr% r' r( �
setdefault�set�update)r) r r r/ r0 r r* r+ r, r, r- � merge_other- s
� �z TocTreeCollector.merge_other�doctreec s� � j j�dg�dtdtt dtt f�fdd� �d
dtdtdtjf� �����fd d
�
����}|r9|� j j �< n t�d �� j j �< �d � j j
�< d S )z;Build a TOC from the doctree and store it in the inventory.r �node�clsr c sR g }t | |�r |�| � | jD ]}t |tj�rqt |tj�r&|�� ||�� q|S )z2Like traverse(), but stay within the same section.)�
isinstance�append�childrenr
�sectionr �extend)r7 r8 �result�child)�traverse_in_sectionr, r- r@ A s
�z9TocTreeCollector.process_doc.<locals>.traverse_in_section� �depthc sd g }| D ]�}t |tj�rc|d }t��}|�|� |�� }�d s$d}nd|d d }�d d7 < tjddg|�R d�|d��}t�dd|�} t� d| �}
�||d �} | r]|
| 7 }
|�
|
� qt |tj �r�tj |d d �} �||�}
|
r�| |
j 7 } |�
| � qt |tj
�r��|tj�D ]}|�� }
|�
|
� t� j���|� q�q|r�tjdg|�R � S d S )
Nr � �#�idsrA T)�internal�refuri�
anchorname�expr)rI )r9 r
r< r � walkabout�get_entry_text� referencer
�compact_paragraph� list_itemr: �onlyr; r �toctree�copyr r �note� bullet_list)r7 rB �entries� sectionnode�title�visitor�nodetextrH rL �para�item�sub_item�onlynode�blist� toctreenode�r � build_tocr r6 �
numentriesr@ r, r- r` M sV
���
� �
�z/TocTreeCollector.process_doc.<locals>.build_tocrC N)rA ) r r r r r r �intr
rS r r! )r) r r6 �tocr, r_ r- � process_doc< s "( /z TocTreeCollector.process_docc C s | � |�| �|� S r )�assign_section_numbers�assign_figure_numbers)r) r r r, r, r- �get_updated_docs� s z!TocTreeCollector.get_updated_docsc s� g �t � ��j�i �_ddtdtdtdtjddf
� ��fdd�
� d tjdtddf� ����fd
d � ��j D ]%}��
|� �� |�}|� tj�D ]}|�
d d
�}|rYd
g��||� qGq5�S )zAAssign a section number to each heading under a numbered toctree.Nr7 �secnumsrB � titlenoder c s | j D ]�}t|tj�r ��d� � |||d |� ��� d }qt|tj�r0� ||||� d }qt|tj�r@� ||||� d }qt|tj �r|�d d7 < t
tj |d �}|dkrgt ��}t
��||d < nd }d ||d < ||d< |r{||d< d }qt|tj�r��||� qd S )Nr rA ���rH � secnumber)r; r9 r
rS r: r rN r
rO rM r rL r& �tuplerP )r7 rh rB ri �subnoderL �number)� _walk_toc�
_walk_toctree�numstackr, r- ro � s:
�
��z:TocTreeCollector.assign_section_numbers.<locals>._walk_tocr^ c s� |dkrd S | d D ]J\}}t �|�s|dkrq
|�v r)tjtd�|| ddd� q
|�jv rTi }|�j|< ��|� � �j| ||�j� |�� |�� |�krT��
|� q
d S )Nr rT r) zA%s is already assigned section numbers (nested numbered toctree?)rc �secnum)�location�type�subtype) r �match�logger�warningr r r r1 �titles�getr: )r^ rB rV �refrh )ro �assignedr �old_secnumbers�rewrite_neededr, r- rp � s&
�
��z>TocTreeCollector.assign_section_numbers.<locals>._walk_toctree�numberedr r )r3 r r r rb r
rV r
rP r% r1 � get_doctree�traverserz )r) r r r6 r^ rB r, )ro rp r| r rq r} r~ r- re � s" *$"
��z'TocTreeCollector.assign_section_numbersc
sH g }t � ��j}i �_i �dtdtf�fdd� �dtdtjdttdf f�fdd � �d
td ttdf dttdf f��fd d
� �dtd ttdf d
tdtddf
��fdd� �dtdtd ttdf ddf� ����fdd� �dtd ttdf ddf���fdd� � �j j
r�� �j j t � � �j�
� D ]\}}||�|�kr�|�|� q�|S )z?Assign a figure number to each figure under a numbered toctree.r7 r c s@ � j �� D ]}|�| �}|jdkr|�| �sq|r | S qd S )N�std)�domains�values�get_enumerable_node_type�name�get_numfig_title)r7 �domain�figtype�r r, r- � get_figtype� s
�z;TocTreeCollector.assign_figure_numbers.<locals>.get_figtyper r< .c sF d|d d }� j �| i �}||v r|�|�}n|�d�}|p"t� S )NrD rE r rC )r rz rl )r r< rH �
secnumbersrr r� r, r- �get_section_number� s
zBTocTreeCollector.assign_figure_numbers.<locals>.get_section_numberr� rr c s> �� | i �}|d � jj� }|�|d�d ||< ||| f S )Nr rA )r2 �config�numfig_secnum_depthrz )r� rr �counter)r �fignum_counterr, r- �get_next_fignumber� s zBTocTreeCollector.assign_figure_numbers.<locals>.get_next_fignumber�fignodeNc s> � j �| i � � j | �|i �}|d d }�||�||< d S )NrE r )r r2 )r rr r� r� �
fignumbers� figure_id)r r� r, r- �register_fignumber� s zBTocTreeCollector.assign_figure_numbers.<locals>.register_fignumberr6 c s� |j D ]Y}t|tj�r �| |�}|r�| ||� q�| ||� qt|tj�r?|d D ]\}}t�|�s7|dkr8q*� ||� q*qt|tj�r\�|�}|rV|d rV�| |||� �| ||� qd S )NrT r) rE ) r; r9 r
r< r
rP r rv r )r r6 rr rm � next_secnumrV �
subdocnamer� )� _walk_doc�
_walk_doctreer� r� r� r, r- r� � s&
� ��z=TocTreeCollector.assign_figure_numbers.<locals>._walk_doctreec s0 | �vr�� | � ��| �}� | ||� d S d S r )r1 r� )r rr r6 )r� r| r r, r- r� s
�z9TocTreeCollector.assign_figure_numbers.<locals>._walk_doc)r3 r r �strr
r< r rb r r� �numfig�root_docrl r( rz r: )r) r r~ �old_fignumbersr �fignumsr, ) r� r� r| r r� r� r� r� r� r- rf � s0 $ ,
���.&
�z&TocTreeCollector.assign_figure_numbers)�__name__�
__module__� __qualname__r r r� r. r r5 r
�documentrd r rg re rf r, r, r, r- r s �
�GJr r r c C s | � t� dddd�S )N�builtinT)�version�parallel_read_safe�parallel_write_safe)�add_env_collectorr )r r, r, r- �setup$ s
�r� N)'�__doc__�typingr r r r r r r r �docutilsr
�docutils.nodesr r �sphinxr
�sphinx.applicationr �sphinx.environmentr �#sphinx.environment.adapters.toctreer � sphinx.environment.collectorsr �
sphinx.localer �sphinx.transformsr � sphinx.utilr r r � getLoggerr� rw r r� r� r, r, r, r- �<module> s$ (