EVOLUTION-NINJA
Edit File: ipc_client.cpython-36.pyc
3 q��i� � @ s: d dl Z d dlZd dlZG dd� d�ZG dd� de�ZdS )� Nc @ s e Zd Zddd�Zddd�ZdS )� IPCClientNc C s> || _ | j d krd| _ || _| jd kr.td��tjd�| _d S )Nz /tmp/com.fortinet.fortimonitorz Missing port� ipc_client)�ipc_path�port� Exception�logging� getLogger�log)�selfr r � r �'/usr/lib/fm-agent/library/ipc_client.py�__init__ s zIPCClient.__init__c C s dd l }|rdj||�}|d7 }y�|j |j|j���}|jd� |jtjj| j | j �� |j d�}|j |� t � }xH|jd�}| jjtj�r�| jjdjt|��� dt|�kr�P ||7 }qrW |jd�}d|kr�d S |S Q R X W n4 tk �r } z| jjd j| �� d S d } ~ X nX d S ) Nr z{}:{}� g $@zutf-8i z received {} bytesZnackzSend/recv failure: {})�socket�format�AF_UNIX� SOCK_STREAM� settimeout�connect�os�path�joinr r �encode�sendall�bytes�recvr � isEnabledForr �DEBUG�info�len�decoder �error) r �command�payloadr �sockZtoSendZ receivedBytes�rZreceived�er r r � send_receive s2 zIPCClient.send_receive)N)N)�__name__� __module__� __qualname__r r' r r r r r s r c s> e Zd Zd � fdd� Zdd� Zdd� Zdd � Zd d � Z� ZS ) � DEMClientNc s" t t| �j||� tjd�| _d S )N� dem_client)�superr+ r r r r )r �dem_portr )� __class__r r r 1 s zDEMClient.__init__c C sV | j d�}y|std��tj|�S tk rP } ztjd|� �� d S d }~X nX d S )Nz wifi-infozNo payload recordedz#Unable to communicate with service )r' � ValueError�json�loadsr r �warning)r �response�errr r r �get_dem_wifi_info5 s zDEMClient.get_dem_wifi_infoc C s$ i }| j d�}|d kr|S tj|�S )N�collect)r' r1 r2 )r �rvr4 r r r �get_dem_results? s zDEMClient.get_dem_resultsc C sV y| j dtj|�d� W n8 tk rP } z | jjdjt|��� W Y d d }~X nX d S )N� initSchedules)r# zinit_schedules error: {})r' r1 �dumpsr r r! r �str)r � schedules�exr r r �init_dem_schedulesG s (z DEMClient.init_dem_schedulesc C s | j dtj|�d�}d S )Nzupdate-schedules)r# )r' r1 r; )r � newSchedules�_r r r �update_schedulesV s zDEMClient.update_schedules)N) r( r) r* r r6 r9 r? rB � __classcell__r r )r/ r r+ 0 s r+ )r1 r r r r+ r r r r �<module> s *