File: //lib/python3.9/site-packages/ipaclient/remote_plugins/__pycache__/schema.cpython-39.pyc
a
}�f�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 m Z m Z m Z d d
l!m"Z" d d l#m$Z$ d d l%m&Z& d d
l'm(Z( e�)e*�Z+dZ,ej-�re.Z/e&e(e/e0d�ee1e2e3e4e5e/e
j6d� Z7ej8ej9ej:ej;ej<ej=ej>ej?ej@ej ej6d� ZAdd� ZBG dd� de�ZCG dd� de�ZDG dd� de�ZEG dd� d�ZFG dd � d eF�ZGG d d � d eF�ZHG d d � d e�ZIG d!d"� d"eJ�ZKG d#d$� d$�ZLd%d&� ZMdS )'� )�Mapping�SequenceN)�x509)�
ClientCommand� ClientMethod)�errors�
parameters�plugable)�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| �d�S t�d S �N�utf-8)�
isinstancer �decode� TypeError)�obj� r( �C/usr/lib/python3.9/site-packages/ipaclient/remote_plugins/schema.py� json_defaultC s
r* c @ s e Zd ZdS )�_SchemaCommandN��__name__�
__module__� __qualname__r( r( r( r) r+ I 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_nameN s z_SchemaMethod.obj_namec C s | j j| j jS r1 )r2 r r3 �versionr5 r( r( r) � obj_versionR s z_SchemaMethod.obj_versionN)r- r. r/ �propertyr7 r9 r( r( r( r) r0 M s
r0 c @ s e Zd ZdS )�
_SchemaObjectNr, r( r( r( r) r; W s r; c @ sT e Zd ZdZdZdd� Zedd� �Zedd� �Zdd � Z d
d � Z
d d
� Z dd� Z dS )�
_SchemaPluginNc C s* |� d�\| _}| _|| _|| _d | _d S )N�/)� partitionr4 r8 � full_name�_schema�_class)r6 �schemar? �_slashr( r( r) �__init___ s z_SchemaPlugin.__init__c C sH | j d ur| j jS | j| j | j }z
|d W S tyB Y d S 0 d S )N�doc)rA rE r@ �
schema_keyr? �KeyError)r6 rB r( r( r) rE e s
z_SchemaPlugin.docc C sR | j d ur| j jS | j| j �| j�}z
|d W S tyL d| j Y S 0 d S )N�summaryz<%s>)rA rH r@ rF �get_helpr? rG �r6 �halpr( r( r) rH p s
z_SchemaPlugin.summaryc sV | j �� ���fdd���r*�fdd�}n �fdd�}d�| j��|_t|g��R � S )Nc s8 t t�| ��}� jjt��t��g|d�d }|���S )N)�params�kw�result)r �zipZCommandZcommand_defaultsr �get)�argsrM rN )r2 �cmd_name�keysr4 r( r) � get_default~ s ��z7_SchemaPlugin._create_default_from.<locals>.get_defaultc s � | � S r1 r( �rQ �rT r( r) �callback� s z4_SchemaPlugin._create_default_from.<locals>.callbackc s � � S r1 r( r( rV r( r) rW � s z{0}_{1}_default)r? �formatr4 r- r
)r6 r2 r4 rS rW r( )r2 rR rT rS r4 r) �_create_default_from{ s z"_SchemaPlugin._create_default_fromc
s� t |d �}t |d �}|�dd�}|dkr:|r:t}d}n\|dkrrd|v rr|d d d krr|�d
d�srt}|d= n$z t| }W n ty� t}Y n0 i }d }|�� D ]�\} }
| d v r�|
|| < q�| d v r�t |
�|| < q�| d
kr�|
|| < q�| dkr�|
}q�| dk�r&tdd� |
D ��} | � ||| �|d< q�| dv r�tdd� |
D ��|| < q�|d u�r�||fi t
|dd���� � j �r�t� fdd�|D ��}n� � |d �}||d< d|v �s�d|v �r�|�
d
d� |d< ||fi |��} |�r�t�| dd� | S )Nr4 �type� sensitiveFr! r �defaultr �False� alwaysask)r^ rE Zlabel�
multivalue�
no_convertZ option_group�required)Z cli_metavarZcli_nameZconfirmZdefault_from_paramc s s | ]}t |�V qd S r1 �r! )�.0�kr( r( r) � <genexpr>� � z._SchemaPlugin._create_param.<locals>.<genexpr>Z default_from)�excludeZincludec s s | ]}t |�V qd S r1 rb )rc �vr( r( r) re � rf )r` c 3 s | ]}� � |�V qd S r1 )�_convert_scalar)rc �d��tmpr( r) re � rf Zautofill�passwordT)r! rP r r �_PARAMSrG r �itemsr rY r r_ ri �pop�object� __setattr__)
r6 r2 rB r4 � type_namer[ �cls�kwargsr\ �key�valuerS Zparamr( rk r) �
_create_param� s` ��
�
�
z_SchemaPlugin._create_paramc s� i }t |d �|d< t |d �|d< t |d �|d< d|v rH|d |d< d|v rlt |d ��d�d |d< nd |d< t� �fd d
�|�d g �D ��|d < �j�j|fS )
Nr4 r8 r? rE � topic_topicr= r �topicc 3 s | ]}�� � |�V qd S r1 )rx �rc �s�r2 r6 r( r) re � s �z._SchemaPlugin._create_class.<locals>.<genexpr>rL � takes_params)r! r> r rP r4 �bases)r6 r2 rB �
class_dictr( r} r) �
_create_class� s
� z_SchemaPlugin._create_classc C sF | j d u r<| j| j | j }| �||�\}}}t|||�| _ | � |�S r1 )rA r@ rF r? r� rZ �r6 r2 rB r4 r r� r( r( r) �__call__� s
z_SchemaPlugin.__call__)
r- r. r/ r rF rD r: rE rH rY rx r� r� r( r( r( r) r< [ s
@r< c sF e Zd ZefZdZedd� �Zedd� �Zdd� Z � fdd �Z
� Z S )
�_SchemaCommandPlugin�commandsc C sX | j d ur| j jS | j| j �| j�}zt|d ��d�d W S tyR Y d S 0 d S )Nry r= r ) rA rz r@ rF rI r? r! r> rG rJ r( r( r) rz � s
z_SchemaCommandPlugin.topicc C s: | j d ur| j jS | j| j �| j�}d|�dg �v S d S )N�clirg )rA �NO_CLIr@ rF rI r? rP rJ r( r( r) r� � s
z_SchemaCommandPlugin.NO_CLIc C s� |� dd�r0ttf}|� dd�sp|td �f }n@zt|d }W n tyV d }Y n0 |� dd�sp|td �f}i }||d< d|v r�|d |d< |� dd�r�d|d < tt|d
�fi |��S ) Nr_ Fra TrZ rE �
no_display)r� �flagsr4 )rP r r rZ �_TYPESrG r r! )r6 r2 rB Z type_typeru r( r( r) �_create_output s"
z#_SchemaCommandPlugin._create_outputc s� t t���� |�\}}}d|v s(d|v r.tf}d|v rFt|d �|d< d|v r^t|d �|d< d|v rzd|d v rzd|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_namer3 rg r� Tr� c s s. | ]&}|� d |� dd��rt|d �V qdS )Z
positionalra Tr4 N)rP r! r{ r( r( r) re + s �z5_SchemaCommandPlugin._create_class.<locals>.<genexpr>rL c 3 s | ]}|j � v r|V qd S r1 �r4 �rc �prU r( r) re - s r~ Z
takes_argsc 3 s | ]}|j � vr|V qd S r1 r� r� rU r( r) re / s Z
takes_optionsc 3 s | ]}�� � |�V qd S r1 )r� r{ r} r( r) re 3 s �outputZ
has_output)�superr� r� r0 r! �setr r� �� __class__)r2 rQ r6 r) r� s, � � � � z"_SchemaCommandPlugin._create_class) r- r. r/ r+ r rF r: rz r� r� r� �
__classcell__r( r( r� r) r� � s
r� c @ s e Zd ZefZdZdS )�_SchemaObjectPlugin�classesN)r- r. r/ r; r rF r( r( r( r) r� 9 s r� c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �_SchemaNameSpacec C s || _ || _d S r1 )r4 r@ )r6 rB r4 r( r( r) rD @ s z_SchemaNameSpace.__init__c C s2 z| j �| j|�W S ty, t|��Y n0 d S r1 )r@ �read_namespace_memberr4 rG �r6 rv r( r( r) � __getitem__D s z _SchemaNameSpace.__getitem__c c s | j �| j�D ]
}|V qd S r1 )r@ �iter_namespacer4 r� r( r( r) �__iter__J s z_SchemaNameSpace.__iter__c C s t t| j�| j���S r1 )�lenr r@ r� r4 r5 r( r( r) �__len__N s z_SchemaNameSpace.__len__c C s2 z| j �| j|�W S ty, t|��Y n0 d S r1 )r@ rI r4 rG r� r( r( r) rI Q s z_SchemaNameSpace.get_helpN)r- r. r/ rD r� r� r� rI r( r( r( r) r� > s
r� c @ s e Zd ZdS )� NotAvailableNr, r( r( r( r) r� X s r� c @ sl e Zd ZdZh d�Zddd�Zddd �Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� 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� r� �topicsNr c
C s t j�|jjjdt�| _i | _i | _ d | _
| j D ] }i | j|< t | |�| j |< q0d}|d ur�z| �
|� W n8 ty� } z t�d|� d }d}W Y d }~n
d }~0 0 |d u �r| j||d�\}}| �| j�| _
z| �|� W n2 t�y } zt�d|� W Y d }~n
d }~0 0 || _|| _d S )NrB FzFailed to read schema: %sT)� ignore_cachezFailed to write schema: %s)�os�path�joinr2 �envZ cache_dir�FORMAT�_dir�_dict� _namespaces�_help�
namespacesr� � _read_schema� Exception�logger�warning�_fetch�_generate_help�
_write_schema� fingerprint�ttl)r6 �clientr� r� �nsZ read_failed�er( r( r) rD r s0
"zSchema.__init__Fc
C s( |� � s|jdd� g }|sJzdd� t�| j�D �}W n tyH Y n0 ddi}|r^||d< z|jdi |��d }W n tjy� t � �Y n0 zT|d
}|�
d �}|�
d� |� � D ]*\}} || j v r�d d
� | D �} | | j
|< q�W n8 t�y }
z t�d|
� t � �W Y d }
~
n
d }
~
0 0 ||fS )NF)�verbosec S s g | ] }t |��qS r( r )rc �fr( r( r) �
<listcomp>� rf z!Schema._fetch.<locals>.<listcomp>r8 z2.170Zknown_fingerprintsrB rN r� r� c S s i | ]}|d |�qS )r? r( )rc �mr( r( r) �
<dictcomp>� rf z!Schema._fetch.<locals>.<dictcomp>zFailed to fetch schema: %s)rB )Z isconnectedZconnectr� �listdirr� �EnvironmentErrorZforwardr Z CommandErrorr� rp ro r� r� rG r� r� ) r6 r� r� Zfpsru rB �fpr� rv rw r� r( r( r) r� � s6
z
Schema._fetchc C s� t j�| j|�}t�|d��b}|�� D ]H}|�d�\}}}|| jv rZ|� |�| j
| |< q&|dkr&|� |�| _ q&W d � n1 s�0 Y d S )N�rr= r� ) r� r� r� r� �zipfile�ZipFileZnamelistr> r� �readr� r� )r6 r� �filenamerB r4 r� rC rv r( r( r) r� � s
zSchema._read_schemac C s. z | j | W S ty( | j| Y S 0 d S r1 )r� rG r� r� r( r( r) r� � s zSchema.__getitem__c C s� i }dD ]�}i ||< || � � D ]t}|d }|| �|i �}|d |d< d|v rl|d �dd�d �� |d< d |v r�|d |d < d
|v r |d
|d
< q q|S ) N)r� r� r? r4 rE z
� r rH ry rg )�values�
setdefault�split�strip)r6 rB rK � namespaceZ
member_schemaZmember_full_namerz r( r( r) r� � s � zSchema._generate_helpc
C s� zt �| j� W n2 tyB } z|jtjkr.� W Y d }~n
d }~0 0 tjd|| jdd��l}z | �|� t �
|� |� � W n t y� t �
|j� � Y n 0 t �|jt j�| j|�� W d � n1 s�0 Y d S )N�wbF)�prefix�dir�delete)r� �makedirsr� r� �errnoZEEXIST�tempfileZNamedTemporaryFile�_write_schema_datar Z
flush_sync�closer� �unlinkr4 �renamer� r� )r6 r� r� r� r( r( r) r� � s" �
zSchema._write_schemac C s� t �|dt j���}| j�� D ]j\}}|| jv rn|}|D ]4}d�||�}tj|| t d�}|�
||� d�� q6q |�
|t�|�� d�� q |�
dtj| j t d�� d�� W d � n1 s�0 Y d S )N�wz{}/{})r\ r# r� )
r� r� � ZIP_DEFLATEDr� ro r� rX �json�dumpsr* Zwritestr�encoder� ) r6 ZfileobjrB rv rw r� �memberr� r| r( r( r) r� � s
�zSchema._write_schema_datac C s: | j | | }t|t�r6t�|�d��}|| j | |< |S r" )r� r$ r r� �loadsr% )r6 r� r� rw r( r( r) r� s
z Schema.read_namespace_memberc C s t | j| �S r1 )�iterr� )r6 r� r( r( r) r�
s zSchema.iter_namespacec C s. t | jt�r t�| j�d��| _| j| | S r" )r$ r� r r� r� r% )r6 r� r� r( r( r) rI
s
�zSchema.get_help)Nr )F)r- r. r/ �__doc__r� rD r� r� r� r� r� r� r� r� rI r( r( r( r) r� \ s
"
" r� c
C s� t � }|}| �� r | �d|�}|d ur�zZz ||u r<t|�}n
t||�}W n4 ty| } z t||j|j�}W Y d }~n
d }~0 0 W nH ty� d }d}Y n< ty� } z|j}|j}W Y d }~nd }~0 0 |j}|j}|| d< | �|� |d u r�t� �t |�}d�
t |�}d�
t j
�t�d |�}z tj| W S t�yB Y n0 t�|�} t j
�|d�| _dg| _| tj|< d�|df�}
t�|
�} t j
�|d�| _t�� | _ttfD ]2} || j D ] }
| |t |
��}| �� |�}�q��q�| tj|
< t � |d �D ]�\}
}t |d
�}d�||f�}
ztj|
} W n> t�yd t�|
� } tj|
< t j
�|d �
|��| _Y n0 |�d �| _ d
|v �r�|d
}t |t ��r�|�!d�}t |��"d�d | _#nd | _#�q�| S )Nr� i z{}${}r z __init__.pyZplugins�.z
plugins.pyr� r4 z{}.pyrE ry r# r= )$rq Zis_validrP r� r
r� r� r� Zupdate_validityr! rX r- r� r� �splitext�__file__�sys�modulesrG �types�
ModuleTyper� r ZRegistry�registerr� r� rF �sixZ iteritemsr� r$ r r% r> rz )Z server_infor� ZNO_FINGERPRINTr� rB r� r� Z package_nameZ package_dir�packageZ module_name�moduleZ
plugin_clsr? Zpluginrz r4 r| r( r( r) � get_package sx
*
r� )N�collections.abcr r r� r� Zloggingr� r� r� r� r� Z cryptographyr Z crypto_x509r� Zipaclient.frontendr r Zipalibr r r 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 rZ r r r r r r r� r ZDNParamZ DNSNameParamr ZBoolZBytesZDateTime�DictZIntrn r* r+ r0 r; r< r� r� r� r� r� r� r� r( r( r( r) �<module> s|
��
L ;