File: //usr/lib/python3.9/site-packages/ipaclient/plugins/__pycache__/otptoken.cpython-39.opt-1.pyc
a
}�f� � @ 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 � }t�� }|�|� |�� |j|dd� ttjdd �}|d u rNt �
d�}z|� � � |�}W n, t
y� t||ttd�d�d� Y d S 0 tj�� r�| jjj�� }t|�� 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.9/site-packages/ipaclient/plugins/otptoken.py� _get_qrcode0 sB
��
���
zotptoken_add._get_qrcodec s� d|v r
n | j jjr d|d< nt|d< |d �dd �}|d ur^|�dd�s^| �|||d �}nd }tt| �j||g|�R i |��}|d ur�t d� |j
t 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+ r2 Z s*
���zotptoken_add.output_for_cli)�__name__�
__module__� __qualname__r, r2 �
__classcell__r* r* r7 r+ r . s *r c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � HTTPSHandlerz=Opens SSL HTTPS connections that perform hostname validation.c K s || _ tjj�| � d S �N)�_HTTPSHandler__kwargs�urllib�requestr= �__init__)r% �kwargsr* r* r+ rB y s zHTTPSHandler.__init__c K s$ | j �� }|�|� t|fi |��S r> )r? �copy�updater
)r% �hostrC �tmpr* r* r+ Z__inner s
zHTTPSHandler.__innerc C s | � | j|�S r> )Zdo_open�_HTTPSHandler__inner)r% Zreqr* r* r+ �
https_open� s zHTTPSHandler.https_openN)r9 r: r; �__doc__rB rH rI 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)�label�passwordr F)rM ZconfirmZ
first_codez
First CodeZ second_codez Second Codeztoken?zToken IDc
sB d| j dii}ttj�| jjj��}|d �dd�|d< tj� |�}dd� | j
D �� � fdd �|� � D �}|r�|d
d ur�|d
|d < tj� |�}|�
d �}ttjjtjjtjjd
�}tj�|��||�}|�� dkr�|�� �| j d�|d | j < |�� |d | j dk�r>ddd��|d | j d�} tjtd�| d��|S )Nr- �unknown� z/xmlz/session/sync_tokenc S s h | ]
}|j �qS r* )�name)�.0�xr* r* r+ � <setcomp>� � z(otptoken_sync.forward.<locals>.<setcomp>c s i | ]\}}|� v r||�qS r* r* )rR �k�v�r5 r* r+ �
<dictcomp>� rU z)otptoken_sync.forward.<locals>.<dictcomp>r �tokenzutf-8)Zcafile�tls_version_min�tls_version_max�� �okzError contacting server!zInvalid Credentials!)�errorzinvalid-credentialszUnknown Error!z Unable to synchronize token: %sr ) �header�listr@ �parse�urlparser r/ Z
xmlrpc_uri�replace�
urlunparse�
takes_options�items� urlencoder r= Z tls_ca_certr[ r\ rA Z build_opener�openZgetcode�infor0 �closer ZExecutionErrorr )
r% r4 rC �status�segmentsZsync_uri�queryZhandlerZrsp�msgr* rX r+ �forward� s:
� ��
�zotptoken_sync.forwardc O s |� d� d S )NzToken synchronized.)Z print_plain)r% r" r- �keysr5 r* r* r+ r2 � s z otptoken_sync.output_for_cliN) r9 r: r; r rJ r` r r rf Z
takes_argsrp r2 r* r* r* r+ rK � s ��$rK )&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 � urllib.parser@ Zurllib.requestZPY3�strZunicode�registerr rA r= rK r* r* r* r+ �<module> s.
G