File: //lib/python3.6/site-packages/ipaclient/remote_plugins/__pycache__/schema.cpython-36.pyc
3
�d[e�I �
@ s d dl mZmZ d dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl m Z
d dlZd dlmZmZ d dlmZmZmZ d dlmZ d dlmZ d dlmZ d d l m Z d d
l m Z m Z m!Z!m"Z" d d l#m$Z$ d d l%m&Z& d d
l'm(Z( d dl)m*Z* ej+e,�Z-dZ.ej/�re0Z1e(e*e1e2d�ee3e4e5e6e7e1e
j8d� Z9ej:ej;ej<ej=ej>ej?ej@ejAejBej"ej8d� ZCdd� ZDG dd� de�ZEG dd� de�ZFG dd� de�ZGG dd� d�ZHG d d � d eH�ZIG d d � d eH�ZJG d d!� d!e�ZKG d"d#� d#eL�ZMG d$d%� d%�ZNd&d'� ZOdS )(� )�Mapping�SequenceN)�x509)�
ClientCommand� ClientMethod)�errors�
parameters�plugable)�USER_CACHE_PATH)�SchemaUpToDate)�Object)�Output)� DefaultFrom�Flag�Password�Str)�ipautil)�fsdecode)�DN)�DNSName�1) r r � PrincipalZNoneTyper �bool�dict�int�list�tuple�unicode� Certificate) �Decimalr r r r �bytesZdatetimer r �strr c C s t | t�r| jd�S t�d S )Nzutf-8)�
isinstancer �decode� TypeError)�obj� r& � /usr/lib/python3.6/schema.py� json_defaultD s
r( c @ s e Zd ZdS )�_SchemaCommandN)�__name__�
__module__� __qualname__r& r& r& r' r) J s r) c @ s$ e Zd Zedd� �Zedd� �ZdS )�
_SchemaMethodc C s | j j| j jS )N)�apir �
obj_full_name�name)�selfr& r& r' �obj_nameO s z_SchemaMethod.obj_namec C s | j j| j jS )N)r. r r/ �version)r1 r& r& r' � obj_versionS s z_SchemaMethod.obj_versionN)r* r+ r, �propertyr2 r4 r& r& r& r' r- N s r- c @ s e Zd ZdS )�
_SchemaObjectN)r* r+ r, r& r& r& r' r6 X s r6 c @ sT e Zd ZdZdZdd� Zedd� �Zedd� �Zdd � Z d
d � Z
d d
� Z dd� Z dS )�
_SchemaPluginNc C s* |j d�\| _}| _|| _|| _d | _d S )N�/)� partitionr0 r3 � full_name�_schema�_class)r1 �schemar: �_slashr& r& r' �__init__` s z_SchemaPlugin.__init__c C sF | j d k r| j jS | j| j | j }y|d S tk
r@ d S X d S )N�doc)r< r@ r; �
schema_keyr: �KeyError)r1 r= r& r& r' r@ f s
z_SchemaPlugin.docc C sN | j d k r| j jS | j| j j| j�}y|d S tk
rH d| j S X d S )N�summaryz<%s>)r< rC r; rA �get_helpr: rB )r1 �halpr& r& r' rC q s
z_SchemaPlugin.summaryc sT | j �� ���fdd���r*�fdd�}n �fdd�}dj| j��|_t|f��� S )Nc s8 t t�| ��}� jjt��t��g|d�d }|j��S )N)�params�kw�result)r �zipZCommandZcommand_defaultsr �get)�argsrG rH )r. �cmd_name�keysr0 r& r' � get_default s z7_SchemaPlugin._create_default_from.<locals>.get_defaultc s � | � S )Nr& )rK )rN r& r' �callback� s z4_SchemaPlugin._create_default_from.<locals>.callbackc s � � S )Nr& r& )rN r& r' rO � s z{0}_{1}_default)r: �formatr0 r* r )r1 r. r0 rM rO r& )r. rL rN rM r0 r' �_create_default_from| s z"_SchemaPlugin._create_default_fromc
s� t |d �}t |d �}|jdd�}|dkr:|r:t}d}n`|dkrtd|krt|d d d krt|jd
d� rtt}|d= n&y t| }W n tk
r� t}Y nX i }d }x�|j� D ]�\} }
| d kr�|
|| < q�| d!kr�t |
�|| < q�| dk� r�|
|| < q�| dk�r |
}q�| dk�r0tdd� |
D ��} | j ||| �|d< q�| d"kr�tdd� |
D ��|| < q�W |d k �r�||ft
|dd��� � j �r�t� fd d�|D ��}n� j |d �}||d< d|k�s�d|k�r�|j
d
d� |d < ||f|�} |�r�tj| d d � | S )#Nr0 �type� sensitiveFr! r �defaultr �False� alwaysaskr@ �label�
multivalue�
no_convert� option_group�required� cli_metavar�cli_nameZconfirmZdefault_from_paramc s s | ]}t |�V qd S )N)r! )�.0�kr& r& r' � <genexpr>� s z._SchemaPlugin._create_param.<locals>.<genexpr>Z default_from�exclude�includec s s | ]}t |�V qd S )N)r! )r^ �vr& r& r' r` � s )rY c 3 s | ]}� j |�V qd S )N)�_convert_scalar)r^ �d)�tmpr& r' r` � s ZautofillZpasswordT)rV r@ rW rX rY rZ r[ )r\ r] )ra rb )r! rJ r r �_PARAMSrB r �itemsr rQ r rX rd �pop�object� __setattr__)
r1 r. r= r0 Z type_namerS �cls�kwargsrT �key�valuerM Zparamr& )rf r' �
_create_param� sf
z_SchemaPlugin._create_paramc s� i }t |d �|d< t |d �|d< t |d �|d< d|krH|d |d< d|krlt |d �jd�d |d< nd |d< t� �fd d
�|jd g �D ��|d < �j�j|fS )
Nr0 r3 r: r@ � topic_topicr8 r �topicc 3 s | ]}�j � |�V qd S )N)rp )r^ �s)r. r1 r& r' r` � s z._SchemaPlugin._create_class.<locals>.<genexpr>rF � takes_params)r! r9 r rJ r0 �bases)r1 r. r= �
class_dictr& )r. r1 r' �
_create_class� s z_SchemaPlugin._create_classc C sF | j d kr<| j| j | j }| j||�\}}}t|||�| _ | j |�S )N)r< r; rA r: rw rR )r1 r. r= r0 ru rv r& r& r' �__call__� s
z_SchemaPlugin.__call__)
r* r+ r, ru rA r? r5 r@ rC rQ rp rw rx r& r& r& r' r7 \ s @r7 c sF e Zd ZefZdZedd� �Zedd� �Zdd� Z � fdd �Z
� Z S )
�_SchemaCommandPlugin�commandsc C sV | j d k r| j jS | j| j j| j�}yt|d �jd�d S tk
rP d S X d S )Nrq r8 r ) r< rr r; rA rD r: r! r9 rB )r1 rE r& r& r' rr � s
z_SchemaCommandPlugin.topicc C s: | j d k r| j jS | j| j j| j�}d|jdg �kS d S )N�clira )r< �NO_CLIr; rA rD r: rJ )r1 rE r& r& r' r| � s
z_SchemaCommandPlugin.NO_CLIc C s� |j dd�r0ttf}|j dd�sr|td �f }nByt|d }W n tk
rX d }Y nX |j dd�sr|td �f}i }||d< d|kr�|d |d< |j dd�r�d
|d< tt|d �f|�S ) NrX Fr[ TrR r@ �
no_display�flagsr0 )r} )rJ r r rR �_TYPESrB r
r! )r1 r. r= Z type_typerm r& r& r' �_create_output s"
z#_SchemaCommandPlugin._create_outputc s� t t��j� |�\}}}d|ks(d|kr.tf}d|krFt|d �|d< d|kr^t|d �|d< d|krzd|d krzd|d< tdd � |d
D ���t�fd d �|d D ��|d
< t�fdd �|d D ��|d< |d = t� �fdd �|d D ��|d< |||fS )NZ obj_classZ attr_namer/ ra r{ Tr| c s s. | ]&}|j d |j dd��rt|d �V qdS )Z
positionalr[ Tr0 N)rJ r! )r^ rs r& r& r' r` , s z5_SchemaCommandPlugin._create_class.<locals>.<genexpr>rF c 3 s | ]}|j � kr|V qd S )N)r0 )r^ �p)rK r& r' r` / s rt Z
takes_argsc 3 s | ]}|j � kr|V qd S )N)r0 )r^ r� )rK r& r' r` 1 s Z
takes_optionsc 3 s | ]}�j � |�V qd S )N)r� )r^ rs )r. r1 r& r' r` 5 s �outputZ
has_output)�superry rw r- r! �setr )r1 r. r= r0 ru rv )� __class__)r. rK r1 r' rw s$ z"_SchemaCommandPlugin._create_class) r* r+ r, r) ru rA r5 rr r| r� rw �
__classcell__r& r& )r� r' ry � s ry c @ s e Zd ZefZdZdS )�_SchemaObjectPlugin�classesN)r* r+ r, r6 ru rA r& r& r& r' r� : s r� c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �_SchemaNameSpacec C s || _ || _d S )N)r0 r; )r1 r= r0 r& r& r' r? A s z_SchemaNameSpace.__init__c C s2 y| j j| j|�S tk
r, t|��Y nX d S )N)r; �read_namespace_memberr0 rB )r1 rn r& r& r' � __getitem__E s z _SchemaNameSpace.__getitem__c c s" x | j j| j�D ]
}|V qW d S )N)r; �iter_namespacer0 )r1 rn r& r& r' �__iter__K s z_SchemaNameSpace.__iter__c C s t t| jj| j���S )N)�lenr r; r� r0 )r1 r& r& r' �__len__O s z_SchemaNameSpace.__len__c C s2 y| j j| j|�S tk
r, t|��Y nX d S )N)r; rD r0 rB )r1 rn r& r& r' rD R s z_SchemaNameSpace.get_helpN)r* r+ r, r? r� r� r� rD r& r& r& r' r� ? s
r� c @ s e Zd ZdS )� NotAvailableN)r* r+ r, r& r& r& r' r� Y s r� c @ s� e Zd ZdZdddhZejjedde �Z
d d d
�Z d d d
�Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � ZdS ) �Schemaa�
Store and provide schema for commands and topics
Create api instance
>>> from ipalib import api
>>> api.bootstrap(context='cli')
>>> api.finalize()
Get schema object
>>> m = Schema(api)
From now on we can access schema for commands stored in cache
>>> m['commands'][u'ping'][u'doc']
u'Ping a remote server.'
>>> m['topics'][u'ping'][u'doc']
u'Ping the remote IPA server to ...'
r� rz �topicsZipar= Nr c C s i | _ i | _d | _x(| jD ] }i | j |< t| |�| j|< qW d}|d k r�y| j|� W n6 tk
r� } ztjd|� d }d}W Y d d }~X nX |d kr�| j ||d�\}}| j
| j �| _y| j |� W n. tk
r� } ztjd|� W Y d d }~X nX || _ || _
d S )NFzFailed to read schema: %sT)� ignore_cachezFailed to write schema: %s)�_dict� _namespaces�_help�
namespacesr� � _read_schema� Exception�logger�warning�_fetch�_generate_help�
_write_schema� fingerprint�ttl)r1 �clientr� r� �nsZ read_failed�er& r& r' r? t s.
zSchema.__init__Fc C s* |j � s|jdd� g }|sLydd� tj| j�D �}W n tk
rJ Y nX ddi}|r`||d< y|jd|�d }W n tjk
r� t � �Y nX yX|d
}|j
d �}|j
d� x6|j � D ]*\}} || j kr�d d
� | D �} | | j
|< q�W W n6 tk
�r }
ztjd|
� t � �W Y d d }
~
X nX ||fS )NF)�verbosec S s g | ] }t |��qS r& )r )r^ �fr& r& r' �
<listcomp>� s z!Schema._fetch.<locals>.<listcomp>r3 z2.170Zknown_fingerprintsr= rH r� r� c S s i | ]}||d �qS )r: r& )r^ �mr& r& r' �
<dictcomp>� s z!Schema._fetch.<locals>.<dictcomp>zFailed to fetch schema: %s)r= )Z isconnectedZconnect�os�listdir�_DIR�EnvironmentErrorZforwardr Z CommandErrorr� ri rh r� r� rB r� r� ) r1 r� r� Zfpsrm r= �fpr� rn ro r� r& r& r' r� � s6
z
Schema._fetchc C s� t jj| j|�}tj|d��\}xT|j� D ]H}|jd�\}}}|| jkr\|j |�| j
| |< q(|dkr(|j |�| _ q(W W d Q R X d S )N�rr8 r� ) r� �path�joinr� �zipfile�ZipFileZnamelistr9 r� �readr� r� )r1 r� �filenamer= r0 r� r> rn r& r& r' r� � s
zSchema._read_schemac C s* y
| j | S tk
r$ | j| S X d S )N)r� rB r� )r1 rn r& r& r' r� � s
zSchema.__getitem__c C s� i }x�d D ]�}i ||< x�|| j � D ]t}|d }|| j|i �}|d |d< d|krp|d jdd�d j� |d < d
|kr�|d
|d
< d |kr$|d |d < q$W q
W |S )
Nrz r� r: r0 r@ z
� r rC rq ra )rz r� )�values�
setdefault�split�strip)r1 r= rE � namespaceZ
member_schemaZmember_full_namerr r& r& r' r� � s
zSchema._generate_helpc " C s� yt j| j� W n0 tk
r@ } z|jtjkr0� W Y d d }~X nX tjd|| jdd��d}y | j|� t j
|� |j � W n" t k
r� t j
|j� � Y n X t j|jt jj| j|�� W d Q R X d S )N�wbF)�prefix�dir�delete)r� �makedirsr� r� �errnoZEEXIST�tempfileZNamedTemporaryFile�_write_schema_datar Z
flush_sync�closer� �unlinkr0 �renamer� r� )r1 r� r� r� r& r& r' r� � s
zSchema._write_schemac
C s� t j|dt j���}x|| jj� D ]n\}}|| jkrt|}xV|D ]4}dj||�}tj|| t d�}|j
||j d�� q:W q |j
|tj|�j d�� q W |j
dtj| j t d�j d�� W d Q R X d S )N�wz{}/{})rT zutf-8r� )
r� r� Z ZIP_DEFLATEDr� rh r� rP �json�dumpsr( Zwritestr�encoder� ) r1 Zfileobjr= rn ro r� �memberr� rs r& r& r' r� � s
zSchema._write_schema_datac C s: | j | | }t|t�r6tj|jd��}|| j | |< |S )Nzutf-8)r� r" r r� �loadsr# )r1 r� r� ro r& r& r' r� s
z Schema.read_namespace_memberc C s t | j| �S )N)�iterr� )r1 r� r& r& r' r� s zSchema.iter_namespacec C s. t | jt�r tj| jjd��| _| j| | S )Nzutf-8)r" r� r r� r� r# )r1 r� r� r& r& r' rD s zSchema.get_help)Nr )F)r* r+ r, �__doc__r� r� r� r� r
�FORMATr� r? r� r� r� r� r� r� r� r� rD r& r& r& r' r� ] s
!
" r� c ( C s� t � }|}| j� r | jd|�}|d k r�yXy ||kr<t|�}n
t||�}W n2 tk
rz } zt||j|j�}W Y d d }~X nX W nH tk
r� d }d}Y n: tk
r� } z|j}|j}W Y d d }~X nX |j}|j}|| d< | j|� |d kr�t� �t |�}dj
t |�}dj
t j
jt�d |�}y
tj| S tk
�r@ Y nX tj|�} t j
j|d�| _dg| _| tj|< dj|df�}
tj|
�} t j
j|d�| _tj� | _xBttfD ]6} x.|| j D ] }
| |t |
��}| j� |�}�q�W �q�W | tj|
< x�t j |d �D ]�\}
}t |d
�}dj||f�}
ytj|
} W n@ tk
�rn tj|
� } tj|
< t j
j|d j
|��| _Y nX |jd �| _ d
|k�r�|d
}t |t ��r�|j!d�}t |�j"d�d | _#nd | _#�q�W | S )Nr� i z{}${}r z __init__.pyZplugins�.z
plugins.pyr� r0 z{}.pyr@ rq zutf-8r8 )$rj Zis_validrJ r� r r� r� r� Zupdate_validityr! rP r* r� r� �splitext�__file__�sys�modulesrB �types�
ModuleTyper� r ZRegistry�registerry r� rA �sixZ iteritemsr� r" r r# r9 rr )Z server_infor� ZNO_FINGERPRINTr� r= r� r� Z package_nameZ package_dir�packageZ module_name�moduleZ
plugin_clsr: Zpluginrr r0 rs r& r& r' � get_package sx
&
r� )P�collections.abcr r r� r� Zloggingr� r� r� r� r� Z cryptographyr Z crypto_x509r� Zipaclient.frontendr r Zipalibr r r Zipalib.constantsr
Z
ipalib.errorsr Zipalib.frontendr Z
ipalib.outputr
Zipalib.parametersr r r r Z ipapythonr Zipapython.ipautilr Z ipapython.dnr Zipapython.dnsutilr Z getLoggerr* r� r� ZPY3r! r rR r r r r r r r r ZDNParamZ DNSNameParamr ZBoolZBytesZDateTimeZDictZIntrg r( r) r- r6 r7 ry r� r� r� r� r� r� r& r& r& r' �<module> sz
L ;