Your IP : 216.73.216.213
B
�d{T�4 � @ sp d Z ddlmZ dZddlZddlZddlZddlmZ ddlm Z ddl
m Z m
ZmZmZmZmZmZmZmZmZ yddlmZ d ZW n ek
r� d
ZY nX dd lm Z m Z m Z m Z d d
� Z dd� Z!dd� Z"dd� Z#dd� Z$dd� Z%dd� Z&dd� Z'd d � Z(d d � Z)G d d!� d!e�Z
G d"d#� d#e �Z de_*d$d%� Z+d&d'� Z,e,e_,G d(d)� d)e�Z-d*ej.dfd+d,�Z/d-Z0dS ).a Add an abstraction level to transparently import optik classes from optparse
(python >= 2.3) or the optik package.
It also defines three new types for optik/optparse command line parser :
* regexp
argument of this type will be converted using re.compile
* csv
argument of this type will be converted using split(',')
* yn
argument of this type will be true if 'y' or 'yes', false if 'n' or 'no'
* named
argument of this type are in the form <NAME>=<VALUE> or <NAME>:<VALUE>
* password
argument of this type wont be converted but this is used by other tools
such as interactive prompt for configuration to double check value and
use an invisible field
* multiple_choice
same as default "choice" type but multiple choices allowed
* file
argument of this type wont be converted but checked that the given file exists
* color
argument of this type wont be converted but checked its either a
named color or a color specified using hexadecimal notation (preceded by a #)
* time
argument of this type will be converted to a float value in seconds
according to time units (ms, s, min, h, d)
* bytes
argument of this type will be converted to a float value in bytes
according to byte units (b, kb, mb, gb, tb)
� )�print_functionzrestructuredtext enN)�copy)�exists)
� OptionParser�Option� OptionGroup�OptionContainer�OptionValueError� OptionError�Values�
HelpFormatter�
NO_DEFAULT�
SUPPRESS_HELP)�DateTimeTF)�
splitstrip�
TIME_UNITS�
BYTE_UNITS� apply_unitsc C sB t |d�r|S y
t�|�S tk
r< td||f ��Y nX dS )zPcheck a regexp value by trying to compile it
return the compiled regexp
�patternz#option %s: invalid regexp value: %rN)�hasattr�re�compile�
ValueErrorr )�option�opt�value� r �I/opt/alt/python37/lib/python3.7/site-packages/logilab/common/optik_ext.py� check_regexpJ s
r c C sD t |ttf�r|S yt|�S tk
r> td||f ��Y nX dS )zTcheck a csv value by trying to split it
return the list of separated values
z option %s: invalid csv value: %rN)�
isinstance�list�tupler r r )r r r r r r � check_csvV s r" c C sB t |t�rt|�S |dkr dS |dkr*dS d}t|||f ��dS )z>check a yn value
return true for yes and false for no
)�y�yesT)�n�noFz<option %s: invalid yn value %r, should be in (y, yes, n, no)N)r �int�boolr )r r r �msgr r r �check_ynb s
r* c C s� t |t�r|S g }xVt| ||�D ]F}|�d�dkrF|�|�dd�� q |�d�dkr |�|�dd�� q W |rvt|�S d}t|||f ��dS )zVcheck a named value
return a dictionary containing (name, value) associations
�=���� �:zMoption %s: invalid named value %r, should be <NAME>=<VALUE> or <NAME>:<VALUE>N)r �dictr" �find�append�splitr )r r r �valuesr) r r r � check_namedo s
r4 c C s |S )z,check a password value (can't be empty)
r )r r r r r r �check_password� s r5 c C s$ t |�r |S d}t|||f ��dS )z/check a file value
return the filepath
z!option %s: file %r does not existN)r r )r r r r) r r r �
check_file� s r6 c C s4 y t �|d�S t jk
r. td| ��Y nX dS )z/check a file value
return the filepath
z%Y/%m/%dz#expected format of %s is yyyy/mm/ddN)r �strptime�Errorr )r r r r r r �
check_date� s
r9 c C s@ t �d|t j�r|S t �d|t j�r(|S d}t|||f ��dS )z�check a color value and returns it
/!\ does *not* check color labels (like 'red', 'green'), only
checks hexadecimal forms
z [a-z0-9 ]+$z #[a-f0-9]{6}zYoption %s: invalid color : %r, should be either hexadecimal value or predefined colorN)r �match�Ir )r r r r) r r r � check_color� s r< c C s t |tttf�r|S t|t�S )N)r r' �long�floatr r )r r r r r r �
check_time� s r? c C s t |d�r|S t|t�S )N�__int__)r r r )r r r r r r � check_bytes� s
rA c @ s� e Zd ZdZejd Zejddg Zeej�Ze ed< e
ed< e ed< e ed< e
ed < e
ed
< eed < eed < eed
< eed< er�ed7 Zeed< dd� Zdd� Zeejd< dd� ZdS )r z7override optik.Option to add some new option types
)
�regexp�csv�yn�named�password�multiple_choice�file�color�time�bytes�hide�levelrB rC rD rE rG rH rI rF rJ rK )�daterN c O s, t j| f|�|� t| d�r(| jr(t| _d S )NrL )�
BaseOption�__init__r rL r �help)�self�opts�attrsr r r rP � s zOption.__init__c C sr | j dkrT| jdkr td| ��qnt| jttf�sntdtt | j���d�d | ��n| jdk rntd| j | ��dS )z3FIXME: need to override this due to optik misdesign)�choicerG Nz/must supply a list of choices for type 'choice'z1choices must be a list of strings ('%s' supplied)�'r- z#must not supply choices for type %r)�type�choicesr
r r! r �strr2 )rR r r r �
_check_choice� s
zOption._check_choice� c C sL | � ||�}| jdkr4t|| j�}|r4|�|� |}| �| j| j||||�S )NrE )Z
convert_valuerW �getattr�dest�updateZ take_action�action)rR r r r3 �parserZexistantr r r �process� s
zOption.processN)�__name__�
__module__� __qualname__�__doc__rO ZTYPESZATTRSr Z TYPE_CHECKERr r" r* r4 r6 r< r5 r? rA �HAS_MX_DATETIMEr9 rP rZ Z
CHECK_METHODSra r r r r r � s*
r c @ s& e Zd ZdZefdd�Zddd�ZdS )r z8override optik.OptionParser to use our Option class
c O s t j| f|�dti|�� d S )N� option_class)�
BaseParserrP r )rR rg �args�kwargsr r r rP � s zOptionParser.__init__Nc C s� |d kr| j }t|dd�}|�| � g }|�|�d�� |�� | jrb|�t�| |�� |�d� xB| j D ]8}|j
|krj|j s�t ||�rj|�|�
|�� |�d� qjW |�� d�|d d� �S )N� output_levelr ZOptions�
� r, )� formatterr\ Zstore_option_stringsr1 �format_heading�indent� option_listr �format_option_helpZ
option_groupsrM � description�
level_options� format_help�dedent�join)rR rn � outputlevel�result�groupr r r rr � s"
z OptionParser.format_option_help)N)rb rc rd re r rP rr r r r r r � s r c s � fdd�| j D �S )Nc s. g | ]&}t |d d�pd� kr|jtk r|�qS )rM r )r\ rQ r )�.0r )rx r r �
<listcomp> s z!level_options.<locals>.<listcomp>)rq )rz rx r )rx r rt s rt c C sB g }t |dd�pd}x"t| |�D ]}|�|�|�� q W d�|�S )Nrk r rm )r\ rt r1 �
format_optionrw )rR rn ry rx r r r r rr s
rr c @ sd e Zd ZdZddd�Zdd� Zd d
� Zd d � Zd dd�Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dS ) �ManHelpFormatterz'Format help using man pages ROFF formatr � �O c C s t �| ||||� d S )N)r rP )rR Zindent_incrementZmax_help_position�widthZ short_firstr r r rP s zManHelpFormatter.__init__c C s d|� � S )Nz.SH %s
)�upper)rR Zheadingr r r ro ' s z ManHelpFormatter.format_headingc C s |S )Nr )rR rs r r r �format_description* s z#ManHelpFormatter.format_descriptionc C sd y
|j }W n tk
r( | �|�}Y nX |jrT| �|�}d�dd� |�� D ��}nd}d||f S )N� c S s g | ] }|� � �qS r )�strip)r{ �lr r r r| 4 s z2ManHelpFormatter.format_option.<locals>.<listcomp>rm z .IP "%s"
%s
)Zoption_strings�AttributeErrorZformat_option_stringsrQ Zexpand_defaultrw �
splitlines)rR r � optstringZ help_textrQ r r r r} - s
z ManHelpFormatter.format_optionr- c C sr d}y |� � }W n tk
r, |�� }Y nX | �||j�}t|d�rT| �||j�}d| �||�|| � |�|f S )Nrm � long_descz %s
%s
%s
%s)
Z_get_prog_namer� Z
get_prog_name�format_short_descriptionrs r �format_long_descriptionr� � format_title�format_synopsis)rR � optparser�pkginfo�sectionr� �pgm�
short_descr r r � format_head; s
z ManHelpFormatter.format_headc C s0 d� dd� t�� d d� D ��}d||||f S )N�-c S s g | ] }t |��qS r )rY )r{ �numr r r r| J s z1ManHelpFormatter.format_title.<locals>.<listcomp>� z.TH %s %s "%s" %s)rw rJ � localtime)rR r� r� rN r r r r� I s z ManHelpFormatter.format_titlec C s d||� � f S )Nz.SH NAME
.B %s
\- %s
)r� )rR r� r� r r r r� M s z)ManHelpFormatter.format_short_descriptionc C s d| S )Nz6.SH SYNOPSIS
.B %s
[
.I OPTIONS
] [
.I <arguments>
]
r )rR r� r r r r� S s z ManHelpFormatter.format_synopsisc C sR d� dd� |�� D ��}|�dd�}|�� �|�rB|t|�d � }d||�� f S )Nrl c S s g | ] }|� � �qS r )�lstrip)r{ �liner r r r| ^ s z<ManHelpFormatter.format_long_description.<locals>.<listcomp>z
.
z
z.SH DESCRIPTION
.B %s
%s
)rw r� �replace�lower�
startswith�lenr� )rR r� r� r r r r� ] s
z(ManHelpFormatter.format_long_descriptionc C s< dt |d|j�|j|j|jf }t|d�r8|d|j 7 }|S )Nz}.SH SEE ALSO
/usr/share/doc/pythonX.Y-%s/
.SH BUGS
Please report bugs on the project's mailing list:
%s
.SH AUTHOR
%s <%s>
Z debian_name� copyrightz
.SH COPYRIGHT
%s
)r\ �modname� mailinglist�author� author_emailr r� )rR r� �tailr r r � format_tailh s
z ManHelpFormatter.format_tailN)r r r� r )r- )rb rc rd re rP ro r� r} r� r� r� r� r� r� r r r r r~ s
r~ r- c C sP t � }||_| |_t|�| ||�|d� t| �|�|d� t|�|�|d� dS )z(generate a man page from an optik parser)rH N)r~ rk r` �printr� rr r� )r� r� r� �streamrM rn r r r �generate_manpage} s r� )r r r r r )1re �
__future__r �
__docformat__r �sysrJ r �os.pathr �optparser rh r rO r r r r
r r r
r Zmxr rf � ImportErrorZlogilab.common.textutilsr r r r r r" r* r4 r5 r6 r9 r< r? rA rM rt rr r~ �stdoutr� �__all__r r r r �<module>0 sB 0
;a