3

�T_,�@sZdd
lZddl
mZ
ddlmZ
ddlmZ
ejd�
ZGdd�de	�Z
Gdd	�d	e
�Zd
S)
�N)
�OrderedDict)
�DocStringParser)
�	ReSTStyleZbcdocsc@sneZ
dZddd�
Zdd�Zdd�Zdd	�Zd
d�Zdd
�Zdd�Z	dd�Z
dd�Zdd�Zdd�Z
dd�ZdS)�ReSTDocument�mancCsBt|�
|_
|
|_t|�
|_d
|_d|_i|_i|_g|_	d|_
dS)NTF)r�style�targetr�parser�	keep_dataZdo_translation�translation_map�hrefs�_writes�_last_doc_string)�selfr�r�/usr/lib/python3.6/restdoc.py�__init__s










zReSTDocument.__init__cCs|jr|
dk	r|j
j|
�

dS)
N)r
r
�append)r�
srrr�_write#s

zReSTDocument._writecCs|j|
�

d
S)z2
        Write content into the document.
        N)
r)r�contentrrr�write'szReSTDocument.writecCs|jd
|j
j�|
f�

dS)z-
        Write content on a newline.
        z%s%s
N)rrZspaces)rrrrr�writeln-szReSTDocument.writelnc

Cs
|jdS)zn
        Returns the last content written to the document without
        removing it from the stack.
        �
���)
r
)
rrrr�
peek_write3szReSTDocument.peek_writec


Cs
|jj
�S)
zL
        Removes and returns the last content written to the stack.
        )r
�pop)
rrrr�	pop_write:szReSTDocument.pop_writecCs|jj
|
�

d
S)z2
        Places new content on the stack.
        N)r
r)rrrrr�
push_write@szReSTDocument.push_writec
CsH|jr6|j
j�
x$|jj�D]\}
}|j
j|
|�
qWd
j|j�
jd�
S)zJ
        Returns the current content of the document as a string.
        �zutf-8)rr�
new_paragraph�items�link_target_definition�joinr
�encode)r�refname�linkrrr�getvalueFs





zReSTDocument.getvaluecs�f
d
d�|
D�
S)Nc
sg|]}
�jj
|
|
��qSr)r�get)�.0�
w)
rrr�
<listcomp>Qsz0ReSTDocument.translate_words.<locals>.<listcomp>r)rZwordsr)
rr�translate_wordsPs
zReSTDocument.translate_wordscCs|
r|jr|j
|
�

dS)
N)r
r)r�datarrr�handle_dataSs


zReSTDocument.handle_datacCsn|
rjy8t|j
�
}|jj|
�

|jj�
t|j
�
}||f|_Wn,tk
rh


tjd
dd�
tj|
�

YnXdS)NzError parsing doc stringT)
�exc_info)	�lenr
r	Zfeed�closer�	Exception�LOG�debug)rZ
doc_string�start�endrrr�include_doc_stringWs












zReSTDocument.include_doc_stringc
Cs$|jdk	r |j\}
}|j
|
|�=dS)
N)rr
)rr5r6rrr�remove_last_doc_stringcs



z#ReSTDocument.remove_last_doc_stringN)
r)�__name__�
__module__�__qualname__rrrrrrrr'r,r.r7r8rrrrrs

rcs�eZ
dZd�f
dd�	Zedd��
Zedd��
Zejd	d��
Zed
d��
Zedd
��
Z	dd�Z
ddd�
Zdd�Zdd�Z
dd�Zdd�Zdd�Zdd�Z�ZS) �DocumentStructureNrcsTtt
|�j|d
�

|
|_t�|_|jg
|_i|_|dk	r>||_|dk	rP|j|�

dS)a5Provides a Hierarichial structure to a ReSTDocument

        You can write to it similiar to as you can to a ReSTDocument but
        has an innate structure for more orginaztion and abstraction.

        :param name: The name of the document
        :param section_names: A list of sections to be included
            in the document.
        :param target: The target documentation of the Document structure
        :param context: A dictionary of data to store with the strucuture. These
            are only stored per section not the entire structure.
        )
rN)	�superr<r�_namer�
_structure�_path�_context�_generate_structure)r�name�
section_namesr�context)
�	__class__rrrks









zDocumentStructure.__init__c


Cs|jS)
z"The name of the document structure)
r>)
rrrrrC�szDocumentStructure.namec


Cs|jS)
zv
        A list of where to find a particular document structure in the
        overlying document structure.
        )
r@)
rrrr�path�szDocumentStructure.pathcCs
|
|_dS)
N)
r@)r�valuerrrrG�sc

Cs
t|j
�
S)
N)�listr?)
rrrr�available_sections�sz$DocumentStructure.available_sectionsc


Cs|jS)
N)
rA)
rrrrrE�szDocumentStructure.contextcCsx|
D]}|j|�

qWdS)
N)
�add_new_section)rrDZsection_namerrrrB�s


z%DocumentStructure._generate_structurecCsJ|j|
|j
|d
�}|j|
g
|_|jj|j_|j|_|j|_||j|
<|S)a�Adds a new section to the current document structure

        This document structure will be considered a section to the
        current document structure but will in itself be an entirely
        new document structure that can be written to and have sections
        as well

        :param name: The name of the section.
        :param context: A dictionary of data to store with the strucuture. These
            are only stored per section not the entire structure.
        :rtype: DocumentStructure
        :returns: A new document structure to add to but lives as a section
            to the document structure it was instantiated from.
        )rCrrE)rFrrGrZindentationrrr?)rrCrE�sectionrrrrK�s







z!DocumentStructure.add_new_sectioncCs
|j|
S)
zRetrieve a section)
r?)rrCrrr�get_section�szDocumentStructure.get_sectioncCs|j|
=d
S)zDelete a sectionN)
r?)rrCrrr�delete_section�sz DocumentStructure.delete_sectionc
Cstt|j
�
d
krD|jrD|jj�
x$|jj�D]\}
}|jj|
|�
q*W|j�}x"|jj�D]\}}||j	�7}qXW|S)z�Flushes a doc structure to a ReSTructed string

        The document is flushed out in a DFS style where sections and their
        subsections' values are added to the string as they are visited.
        r)
r0rGrrr r!r"r'r?�flush_structure)rr%r&rHrCrLrrrrO�s








z!DocumentStructure.flush_structurec

Csd
j|j
�
jd�
S)Nrzutf-8)r#r
r$)
rrrrr'�s
zDocumentStructure.getvaluec

Cst�|_
dS)
N)rr?)
rrrr�remove_all_sections�s
z%DocumentStructure.remove_all_sectionsc

Cs
g|_dS)
N)
r
)
rrrr�
clear_text�s
zDocumentStructure.clear_text)NrN)
N)r9r:r;r�propertyrCrG�setterrJrErBrKrMrNrOr'rPrQ�
__classcell__rr)
rFrr<js

r<)ZloggingZbotocore.compatrZawscli.bcdoc.docstringparserrZawscli.bcdoc.stylerZ	getLoggerr3�objectrr<rrrr�<module>
s


T