File: //lib/python3.6/site-packages/ipaclient/plugins/__pycache__/otptoken.cpython-36.pyc
3
�d[e� � @ s d dl mZ d dlZd dlmZ d dlmZmZmZm Z d dlm
Z
d dl m Z m
Z
d dlmZ d dlmZ d d lmZ d d
lmZ d dlZd dlZd dlZd d lmZ d dlZ d dl Z ej r�e Z e� Z!e!d d d
�G dd� de��Z"G dd� de j#j$�Z$e!� G dd� de��Z%dS )� )�print_functionN)�MethodOverride)�api�Str�Password�_)�errors)� add_message�ResultFormattingError)�Registry)�Local)�create_https_connection)� API_VERSION)�StringIOT)�overrideZno_failc s$ e Zd Zdd� Z� fdd�Z� ZS )� otptoken_addc
C s� t � }tj� }|j|� |j� |j|dd� ttjdd �}|d krNt j
d�}y|j � j |�}W n, t
k
r� t||ttd�d�d� d S X tjj� r�| jjjj� }t|j� d �} | |kr�t||ttd�d�d� |S )NF)�out�tty�encodingzuUnable to display QR code using the configured output encoding. Please use the token URI to configure your OTP device)�messager zRQR code width is greater than that of the output tty. Please resize your terminal.)r �qrcodeZQRCodeZadd_dataZmake� print_ascii�getattr�sys�stdout�locale�getpreferredencoding�getvalue�encode� UnicodeErrorr r
r �isattyr ZBackend�textuiZ
get_tty_width�len�
splitlines)
�self�output�uri�versionZ qr_output�qrr Zqr_codeZ output_widthZ
qr_code_width� r) � /usr/lib/python3.6/otptoken.py� _get_qrcode0 s8
zotptoken_add._get_qrcodec s� d|kr
n | j jjr d|d< nt|d< |d jdd �}|d k r`|jdd� r`| j|||d �}nd }tt| �j||f|�|�}|d k r�t d� |j
t j j
� d� t d� |S ) Nr' z2.0�resultr& Z no_qrcodeF�
)r )r �envZskip_version_checkr �getr+ �superr �output_for_cli�printr r r r )r$ r! r% �args�optionsr& r( �rv)� __class__r) r* r1 Z s
zotptoken_add.output_for_cli)�__name__�
__module__� __qualname__r+ r1 �
__classcell__r) r) )r6 r* r . s *r c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � HTTPSHandlerz=Opens SSL HTTPS connections that perform hostname validation.c K s || _ tjjj| � d S )N)�_HTTPSHandler__kwargs�urllib�requestr; �__init__)r$ �kwargsr) r) r* r? y s zHTTPSHandler.__init__c K s | j j� }|j|� t|f|�S )N)r< �copy�updater
)r$ �hostr@ Ztmpr) r) r* Z__inner s
zHTTPSHandler.__innerc C s | j | j|�S )N)Zdo_open�_HTTPSHandler__inner)r$ Zreqr) r) r* �
https_open� s zHTTPSHandler.https_openN)r7 r8 r9 �__doc__r? rD rE r) r) r) r* r; v s r; c @ s| e Zd Zed�ZdZeded�d�eded�dd �ed
ed �dd �ed ed
�dd �fZeded�d�fZ dd� Z
dd� Z dS )�
otptoken_synczSynchronize an OTP token.zX-IPA-TokenSync-Result�userzUser ID)�labelZpasswordr F)rI ZconfirmZ
first_codez
First CodeZ second_codez Second Codeztoken?zToken IDc
sT d| j dii}ttjj| jjj��}|d dks4t�|d j dd�|d< tjj
|�}dd � | j D �� � fd
d �|j � D �}|r�|d d k r�|d |d < tjj
|�}|jd
�}ttjjtjjtjjd�}tjj|�j||�}|j� dk�r|j� j| j d�|d | j < |j� |d | j dk�rPddd�j|d | j d�} tjt d�| d��|S )Nr, �unknownr Zhttps� z/xmlz/session/sync_tokenc S s h | ]
}|j �qS r) )�name)�.0�xr) r) r* � <setcomp>� s z(otptoken_sync.forward.<locals>.<setcomp>c s i | ]\}}|� kr||�qS r) r) )rM �k�v)r4 r) r* �
<dictcomp>� s z)otptoken_sync.forward.<locals>.<dictcomp>�tokenzutf-8)Zcafile�tls_version_min�tls_version_max�� �okzError contacting server!zInvalid Credentials!)�errorzinvalid-credentialszUnknown Error!z Unable to synchronize token: %s)r ) �header�listr= �parseZurlparser r. Z
xmlrpc_uri�AssertionError�replaceZ
urlunparse�
takes_options�itemsZ urlencoder r; Z tls_ca_certrT rU r> Z build_opener�openZgetcode�infor/ �closer ZExecutionErrorr )
r$ r3 r@ ZstatusZsegmentsZsync_uriZqueryZhandlerZrsp�msgr) )r4 r* �forward� s4
zotptoken_sync.forwardc O s |j d� d S )NzToken synchronized.)Z print_plain)r$ r! r, �keysr4 r) r) r* r1 � s z otptoken_sync.output_for_cliN) r7 r8 r9 r rF rY r r r^ Z
takes_argsrd r1 r) r) r) r* rG � s $rG )&Z
__future__r r Zipaclient.frontendr Zipalibr r r r r Zipalib.messagesr r
Zipalib.plugabler Zipalib.frontendr Z ipalib.utilr
Zipapython.versionr r r Zsix�ior Z urllib.parser= Zurllib.requestZPY3�strZunicode�registerr r> r; rG r) r) r) r* �<module> s. G