File: //opt/alt/python37/lib/python3.7/site-packages/pyroute2/cli/__pycache__/server.cpython-37.pyc
B
tP�e� � @ s� d dl Z d dlmZ d dlmZ y d dlmZ d dlmZ W n, ek
rh d dl mZ d dl mZ Y nX G dd� de
�Z G dd � d e�Z G d
d � d e�Z
dS ) � N)�Session)�NDB)�BaseHTTPRequestHandler)�
HTTPServerc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� ProxyEncoderc C s
|| _ d S )N)�wfile)�selfr � r �D/opt/alt/python37/lib/python3.7/site-packages/pyroute2/cli/server.py�__init__ s zProxyEncoder.__init__c C s | j �|�d�� d S )Nzutf-8)r �write�encode)r �datar r r
r s zProxyEncoder.writec C s | j �� d S )N)r �flush)r r r r
r s zProxyEncoder.flushN)�__name__�
__module__� __qualname__r r r r r r r
r s r c @ s e Zd Zdd� Zdd� ZdS )�Handlerc C s | � ||� | �� d S )N)Z
send_error� end_headers)r �code�reasonr r r
�do_error s zHandler.do_errorc
C s� | j dkr| �dd�S d| jkr,| �dd�S d| jkrB| �dd�S t| jd �}| jd }| j�|�}|dkr�yt�|�}W q� tk
r� | �dd �S X n*|d
kr�d |� d ��
d
�i}n | �dd� d| jk�r8| j j �
| jd �}|d kr�| �dd�S y|| j�}W n tk
�r& | �dd�S X | j j�|�}n | j j�rN| �dd�S | j j}t|t| j�dd�}| �d� | �� x |d D ]} |�| � �q�W d S )Nz/v1/i� z
url not foundzContent-Lengthi� z Content-Typei� zapplication/jsonzIncorrect JSON inputz
text/plain�commandszutf-8�;zIncorrect content typez X-Auth-Mechi� z"Authentication mechanism not foundi� zAuthentication failedzAuthentication required)Zls�.z..�version)�ndb�stdout�builtins�� )�pathr �headers�int�rfile�read�json�loads�
ValueError�decode�split�server� auth_plugins�get� Exceptionr Z
auth_proxy� auth_strictr r r Z
send_responser �handle)
r �content_length� content_typer �requestZ auth_plugin�amr �session�cmdr r r
�do_POST sL
zHandler.do_POSTN)r r r r r6 r r r r
r s r c @ s e Zd Zddd�ZdS )�Server� localhost� NFc C s^ i | _ || _|pi | _|d k r&|| _nt||d�| _| jj�ddd�� t�| ||ft � d S )N)�sources�logr% �true)Z show_formatZrecordset_pipe)
�sessionsr. r+ r r �config�updater r r )r �address�portr: r r; r. r+ r r r
r X s
zServer.__init__)r8 r9 NNNFN)r r r r r r r r
r7 W s r7 )r% Zpyroute2.cli.sessionr Zpyroute2.ndb.mainr �BaseHTTPServerr r � ImportErrorZ http.server�objectr r r7 r r r r
�<module> s >