File: //usr/local/lib/python3.8/dist-packages/__pycache__/zabbix_api.cpython-38.pyc
U
�o�e�3 � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZz d dlZW n ek
rd d dl m
Z Y nX d dl Z d dl m
Z
zd dlmZ dZW n ek
r� dZY nX e�ej�Ze�d�Ze�e� e�dd� zd dlZe�d d
� W n* ek
�r d dlZe�d d � Y nX d d
� Zdd� Zdd� ZG dd� de �Z G dd� de �Z G dd� de �Z G dd� de �Z G dd� de!�Z"G d d � d e"�Z#dS ) � N)�deque)�_create_unverified_contextTFZ
zabbix_api�
zStarting logging� zUsing simplejson libraryzUsing native json libraryc s � fdd�}|S )z; Decorator to check authentication of the decorated method c s | � � � | |�S �N)�
__checkauth__)�self�args��fn� �./tmp/pip-unpacked-wheel-y8_snd7v/zabbix_api.py�retD s zcheckauth.<locals>.retr )r r r r
r
� checkauthA s r c s � fdd�}|S )Nc s6 | j �tjdt� �t|�f � | �| �||��d S )Nz'Going to do_request for %s with opts %s�result)�logger�log�logging�DEBUG�repr�
do_request�json_obj)r �method�optsr
r r
�wrapperK s
��zdojson.<locals>.wrapperr )r r r r
r
�dojsonJ s r c C s~ t | �d�|�d��D ]b\}}|�� r\|�� r\t|�t|�krD dS t|�t|�k rx dS q| |krj dS | |k r dS qdS )zG
The result is 0 if v1 == v2, -1 if v1 < v2, and +1 if v1 > v2
�.� ���r )�zip�split� isdecimal�int)�v1Zv2Zv1_partZv2_partr r r
�version_compareS s r$ c @ s e Zd ZdZdS )�ZabbixAPIExceptionz� generic zabbix api exception
code list:
-32602 - Invalid params (eg already exists)
-32500 - no permissions
N��__name__�
__module__� __qualname__�__doc__r r r r
r% e s r% c @ s e Zd ZdS )�Already_ExistsN�r' r( r) r r r r
r+ o s r+ c @ s e Zd ZdZdS )�InvalidProtoErrorz Recived an invalid proto Nr& r r r r
r- s s r- c @ s e Zd ZdS )�
APITimeoutNr, r r r r
r. y s r. c @ s� e Zd ZdZdZdZdZdZdZdZ dZ
dZ dZ dZ
dZde e ejdddfd d
�Zd d � Zd
d� Zdd� Zd%dd�Zi dfdd�Zd&dd�Zdd� Zdd� Zdd � Zd d � Zd d � Z d!d"� Z d#d$� Z dS )'� ZabbixAPI� F�/api_jsonrpc.phpN�httpr zhttp://localhost/zabbixTc K s~ | � � | �|� || _|d | _| j�d�d | _|| _|| _|| _|| _ d| _
t g |d�| _ || _
| �tjd| j � dS )z Create an API object. r1 z://r )�maxlenzurl: N)�
_setuplogging�
set_log_level�server�urlr �proto�httpuser�
httppasswd�timeout�kwargs�idr �r_query�validate_certs�debugr �INFO) r r6 �user�passwdZ log_levelr; Z r_query_lenr? r< r r r
�__init__� s
zZabbixAPI.__init__c C s t �d| jj �| _d S )Nz
zabbix_api.%s)r � getLogger� __class__r' r �r r r r
r4 � s zZabbixAPI._setuploggingc C s" | � tjd| � | j�|� d S )NzSet logging level to %d)r@ r rA r �setLevel)r �levelr r r
r5 � s zZabbixAPI.set_log_levelc C s
t | j�S )z%
return recent query
)�listr> rG r r r
� recent_query� s zZabbixAPI.recent_queryc C sB t |�d }|r |t |� }|dkr0|t |� }| j�||� d S )Nz: r0 )�strr r )r rI �var�msg�strvalr r r
r@ � s zZabbixAPI.debugc C s> d||| j | jd�}|s |d= | �tjdt|� � t�|�S )Nz2.0)Zjsonrpcr �params�authr= rQ z
json_obj: )rQ r= r@ r r rL �json�dumps)r r rP rQ �objr r r
r � s �zZabbixAPI.json_objc C s� |d k r&| � tjd� || _d| _d S |dkrH|}|}|rh|| _|| _n | jdkr`| j}| j}ntd��dt� |�
d��� � d }| � tjdt |�t |�f � t
| �� d �d
kr�||d �}n
||d �}| jd
|dd�} | �| �}
|
d | _d S )NzUsing API Token for authTr0 �(No authentication information available.zmd5(�utf-8�)zTrying to login with %s:%sz5.4r )�username�password)rB rY z
user.loginF�rQ r )r@ r r rQ �
__tokenauth__� __username__� __password__r% �hashlib�md5�encode� hexdigestr r$ � api_versionr r ) r rB rY �saveZ api_tokenZl_userZ
l_passwordZhashed_pw_stringZ login_argrT r r r r
�login� s2
�
zZabbixAPI.loginc C s� | j r0| �tjd� d| _d| _d| _d| _ d S | jdkrBtd��| �tjd| j � | jddd�}| � |�}|d r�d| _d| _d| _d S )
Nz1Clearing auth information due to use of API Tokenr0 FrU zTrying to logout user: %s.z user.logoutTrZ r )
r[ r@ r r rQ r\ r] r% r r �r rT r r r r
�logout� s
zZabbixAPI.logoutc C sD | j dkr<| �dd| j i�}| �|�}|d s8d| _ dS dS dS d S )Nr0 zuser.checkAuthenticationZ sessionidr FT)rQ r r re r r r
�
test_login s
zZabbixAPI.test_loginc
C sF ddd�}| j rP| �tjd� | j d | j �d�}dt�|��d� }||d< | j �
t |�� | �tjd t |� � | �tj d
t |� � t
j| j|�d �|d �}| jd
kr�tr�| js�t
jdt� d�}n t
jdd�}t
�|�}n2| jdk�rt
jdd�}t
�|�}ntd| j ��t
�|� z|j|| jd�} W n� tj k
�rv }
z"t |
d��rZ|
j }
td|
��W 5 d }
~
X Y n� t jk
�r� }
z t d��W 5 d }
~
X Y n` t
j!k
�r }
z>t |
d��r�|
j �r�|
j }
nt |
d��r�|
j"}
td|
��W 5 d }
~
X Y nX | �tjdt | j#� � | j#dk�r:td| j$| j"f ��| �%� } t&| �dk�rXtd ��zt'�(| �d ��} W n: t)k
�r� }
zt*d | � t+�,d � W 5 d }
~
X Y nX | �tj d t | � � | j-d 7 _-d!| k�rBd"| d! d# | d! d | d! d$ t |�f }
t.�/d%| d! d$ t.j0��r0t1|
| d! d# ��nt|
| d! d# ��| S )&Nzapplication/json-rpczpython/zabbix_api)z Content-Typez
User-AgentzHTTP Auth enabled�:�asciizBasic �
Authorizationz Sending: zSending headers: rV )r7 �data�headers�httpsr )�
debuglevel�context)rn r2 zUnknow protocol %s)r; �messagezssl.SSLError - %szHTTP read timeout�reasonzurllib2.URLError - %szResponse Code: �� zHTTP ERROR %s: %szReceived zero answerz%unable to decode. returned string: %sr zResponse Body: r �errorz!Error %s: %s, %s while sending %s�coderk z.*already\sexists.*)2r9 r@ r rA r: r` �base64� b64encode�decoder> �appendrL r �urllib2�Requestr7 r8 �HAS_SSLCONTEXTr? � HTTPSHandlerr � build_opener� HTTPHandlerr% �install_opener�openr; �ssl�SSLError�hasattrrp �socketr. �URLErrorrq rt �status�read�lenrR �loads�
ValueError�print�sys�exitr= �re�search�Ir+ )r r rl � credentialsrQ �request�
https_handler�openerZ http_handler�response�eZreadsZjobjrN r r r
r sz �
�
�zZabbixAPI.do_requestc C s | j dkrdS dS )Nr0 TFrZ rG r r r
� logged_inV s
zZabbixAPI.logged_inc K s | � | jd|dd��}|d S )Nzapiinfo.versionFrZ r )r r )r �optionsrT r r r
rb [ s zZabbixAPI.api_versionc C s | � � std��d S )NzNot logged in.)r� r% rG r r r
r _ s zZabbixAPI.__checkauth__c C s t | td|if| j��S )N�prefix)�ZabbixAPISubClass�dictr< )r �namer r r
� __getattr__c s zZabbixAPI.__getattr__)r0 N)r0 r0 TN) r' r( r) r\ r] r[ rQ r7 rP r r8 r9 r: r; r? r �WARNINGrD r4 r5 rK r@ r rd rf rg r r� rb r r� r r r r
r/ } s> �
# Er/ c @ sP e Zd ZdZdZdZdd� Zdd� Zdd� Zd d
� Z d d � Z
e e d
d� ��Z
dS )r� z@ wrapper class to ensure all calls go through the parent object Nc K sj | � � | �tjd| jj � || _|| _|�� D ]2\}}t | ||� | �tj
dt |�t |�f � q2d S )Nz Creating %sz Set %s:%s) r4 r@ r rA rF r' rk �parent�items�setattrr� r )r r� rk r< �key�valr r r
rD m s zZabbixAPISubClass.__init__c s, �j d dkr� dkrd� � �fdd�}|S )Nr� �
configuration�import_�importc s �� d�jd � f | d �S )Nz%s.%sr� r )� universalrk )r �r� r r r
r | s z-ZabbixAPISubClass.__getattr__.<locals>.method)rk )r r� r r r� r
r� x s z ZabbixAPISubClass.__getattr__c C s | j �� d S r )r� r rG r r r
r � s z ZabbixAPISubClass.__checkauth__c C s | j �|�S r )r� r )r �reqr r r
r � s z ZabbixAPISubClass.do_requestc C s | j �||�S r )r� r )r r �paramr r r
r � s zZabbixAPISubClass.json_objc K s |S r r )r r r r r
r� � s zZabbixAPISubClass.universal)r' r( r) r* r� rk rD r� r r r r r r� r r r r
r� g s r� )$ru r^ r �stringr� r� r� ry � ImportError�urllib.requestr� r� � collectionsr r r{ �
StreamHandler�stdoutZdefault_log_handlerrE Z__logger�
addHandlerr �
simplejsonrR r r r$ � Exceptionr% r+ r- r. �objectr/ r� r r r r
�<module> sJ
k