U
�V|# �
@ s d dl mZ d dlZd dlZd dlZd dlZd dlZz d dlZW n ek
r\ d dl ZY nX z4d dl
m Z d dl
m Z d dl
m
Z
d dl
mZ W n. ek
r� Z zeee���W 5 dZ[X Y nX d dlmZ d dlmZ d Zd
d � ZG d d
� d
ej�ZG dd� de�ZdS )� )�logN)�apidoc)� application)�config)�
setup_command)�git)�optionszi%(heading)s
%(underline)s
.. automodule:: %(module)s
:members:
:undoc-members:
:show-inheritance:
c C s@ |D ]6}|� d�r|dkrd| d|�dd�|d d� f < qd S )Nz.pyz __init__.pyTz%s.%s�/�.���)�endswith�replace)�arg�dirname�files�filename� r � /builddoc.py�
_find_modules< s �
�r c @ sR e Zd ZdZddgZdd� Zddd�Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dS )�
LocalBuildDoc� build_sphinxZhtml�manc C sJ | j �d�}d|kr,tj�|d d d�}nd}tj�|�sFt�|� |S )Nr �
source_dir� Zapizdoc/source/api)� distribution�get_option_dict�os�path�join�exists�makedirs)�self� option_dictr r r r �_get_source_dirH s
z LocalBuildDoc._get_source_dirNc s2 t �dtj�tj� � i }| �� }| jjD ].}d|kr,t� |�D ]\}}}t
|||� qBq,� fdd��t �fdd�|� � D ��}tj�
|d�} t| d���}
|
�d � |D ]x} tj�
|d
| �} d | }
d t|
� }t| |
|d
�}t �d| � t| d��}|�t| � W 5 Q R X |
�d| � q�W 5 Q R X d S )Nz [pbr] Autodocumenting from %sr
c s t � fdd��D �� S )Nc 3 s | ]}t � � |�V qd S �N)�fnmatch)�.0Zpat��moduler r � <genexpr>] s �zDLocalBuildDoc.generate_autoindex.<locals>.include.<locals>.<genexpr>)�anyr' )�excluded_modulesr' r �include\ s �z1LocalBuildDoc.generate_autoindex.<locals>.includec 3 s | ]}� |�r|V qd S r$ r )r& �mod)r, r r r) ` s z3LocalBuildDoc.generate_autoindex.<locals>.<genexpr>z
autoindex.rst�wz .. toctree::
:maxdepth: 1
z%s.rstzThe :mod:`%s` Module�=)r( �heading� underlinez[pbr] Generating %sz
%s.rst
)r �infor r �abspath�curdirr# r Zpackages�walkr �sorted�keysr �open�write�len�dict�
_rst_template)r! r+ �modulesr Zpkg�dirpathZdirnamesr Z module_listZautoindex_filenameZ autoindexr( Zoutput_filenamer0 r1 �valuesZ output_filer )r+ r, r �generate_autoindexR s: �
� �� z LocalBuildDoc.generate_autoindexc C s, | � � }ddddd|g}t�|| j � d S )Nr r
z-HZModulesz-o)r# r �main�autodoc_tree_excludes)r! r �cmdr r r � _sphinx_treeu s zLocalBuildDoc._sphinx_treec
C s� | j st�� }ntj}i }| jr*| j|d< | jr:| j|d< | jrJ| j|d< t� | j
di g �}t � d�j}t �
|�t �
d�kr�|�tj� n|�� | jdkr�t|j�dkr�d S tj| j| j
| j| j| j||| jd d
� }z|j| jd � W nt t k
�r\ } zTdd l m } t ||j ��rJtj!�"d
� tj#�"|j$d �%dd�� tj#�"d� n� W 5 d }~X Y nX | j&�r�|jj'|jj( }|j�)d�} t*�+|| � d S )N�version�release�todayzconf.py�sphinxz1.3.1r r F)ZfreshenvZwarningiserror)Z force_all)�utilszreST markup error:
�ascii�backslashreplace�
�index),�verbose� cStringIO�StringIO�sys�stdoutrE rF rG r ZConfigZ
config_dir�
pkg_resourcesZget_distributionZ
parse_versionZ init_values�warnings�warn�builderr: Z man_pagesr ZSphinxr Zbuilder_target_dirZ doctree_dirZ fresh_envZbuildZ all_files� ExceptionZdocutilsrI �
isinstanceZ
SystemMessageZstderr9 �stderr�args�encodeZ
link_indexZ
master_docZ
out_suffixZget_outfilenamer �symlink)
r! Z
status_streamZ
confoverridesZ
sphinx_configZ
sphinx_verZapp�errrI �srcZdstr r r � _sphinx_runz sZ
� � � zLocalBuildDoc._sphinx_runc C s� | j �d�}t�� r,tj|d� tj|d� t�|dd�}t�|dd�}t� d�s�|r^| �
� |r�| � t |�
dd d g�d
�� �� | jD ]Z}|| _| �� | j �� | _| j �� | _| j �� | _t�|d d �r�| �� q�tj�| � q�d S )
N�pbr)r" Zautodoc_tree_index_modulesZAUTODOC_TREE_INDEX_MODULESZautodoc_index_modulesZAUTODOC_INDEX_MODULESZ SPHINX_DEBUGZautodoc_exclude_modules� r Z
warnerrorsZ
WARNERRORS)r r r Z_git_is_installedZwrite_git_changelogZgenerate_authorsr Zget_boolean_optionr �getenvrD r@ �set�get�split�buildersrV �finalize_optionsZget_nameZprojectZ get_versionrE rF r_ r �BuildDoc�run)r! r" Z
tree_indexZ
auto_indexrV r r r ri � sH ��
���
�
zLocalBuildDoc.runc C s t j�| � dg| _d S )Nzsetup.py)r rh �initialize_optionsrB )r! r r r rj � s z LocalBuildDoc.initialize_optionsc C s� t j�| � | j�d�}d|�dg g�d kr<|d d | _t| jt�s\| jr\| j� d�| _d}| j�d�}||kr�|| d | _
| � |� d S ) Nr z command linerV r r �,rB r` ) r rh rg r r rd rf rX �listre rB Zensure_string_list)r! r" �optr r r rg � s z LocalBuildDoc.finalize_options)N) �__name__�
__module__� __qualname__� command_namerf r# r@ rD r_ ri rj rg r r r r r C s
#,"r c @ s e Zd ZdgZdZdS )�LocalBuildLatexZlatexZbuild_sphinx_latexN)rn ro rp rf rq r r r r rr � s rr )Z distutilsr r% r rS rQ rT rO � ImportError�iorH r r r r rW �e�strr` r r r< r rh r rr r r r r �<module> s.
"