File: //usr/lib/python3.6/site-packages/glances/__pycache__/client.cpython-36.pyc
3
Eo�c%# � @ s� d Z ddlZddlZddlZddlZddlmZ ddlmZm Z m
Z
m Z ddl m
Z
ddlmZ ddlmZ ddlmZ G d d
� d
e �ZG d d � d e�ZdS )
zManage the Glances client.� N)� __version__)�Fault�
ProtocolError� ServerProxy� Transport)�logger)�GlancesStatsClient)�GlancesCursesClient)�Counterc @ s e Zd ZdZdd� ZdS )�GlancesClientTransportzFThis class overwrite the default XML-RPC transport and manage timeout.c C s
|| _ d S )N)�timeout)�selfr � r � /usr/lib/python3.6/client.py� set_timeout s z"GlancesClientTransport.set_timeoutN)�__name__�
__module__� __qualname__�__doc__r r r r r r s r c @ s� e Zd ZdZd dd�Zedd� �Zd d
d �Zed d
� �Zej dd
� �Zdd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � ZdS )!�
GlancesClientz.This class creates and manages the TCP client.N� Fc C s� || _ || _|j| _|j| _d| _|| _|jdkrNdj |j
|j|j |j �| _
ndj |j |j �| _
tjdj | j
�� t� }|j|� yt| j
|d�| _ W n6 tk
r� } z| jdj | j
|�� W Y d d }~X nX d S )N�glances� zhttp://{}:{}@{}:{}z http://{}:{}zTry to connect to {})� transportz$Client couldn't create socket {}: {})�args�config�quiet�_quiet�time� refresh_time� _client_mode�return_to_browserZpassword�formatZusername�clientZport�urir �debugr r r � Exception� log_and_exit)r
r r r r! r �er r r �__init__% s
zGlancesClient.__init__c C s | j S )N)r )r
r r r r C s zGlancesClient.quietr c C s* | j s tj|� tjd� n
tj|� dS )z
Log and exit.� N)r! r �critical�sys�exit�error)r
�msgr r r r' G s
zGlancesClient.log_and_exitc C s | j S )zGet the client mode.)r )r
r r r � client_modeO s zGlancesClient.client_modec C s
|| _ dS )zpSet the client mode.
- 'glances' = Glances server (default)
- 'snmp' = SNMP (fallback)
N)r )r
�moder r r r0 T s c C sX d}y| j j� }W n� tjk
rx } zHd| _tjdj|j|j�� dj| j �}| j
s^t |� n
tj |� W Y dd}~X n^ t
k
r� } zBdj| j �}|jdkr�|d7 }n|dj|j|j�7 }| j|� d S d}~X nX | jd
k�rTtjd �d |jd �d k�r>t| j| jd
�| _| jjtj| j j� �� tjdjt|�� n| jdjt|�� d S dS )zLogin to a Glances serverN�snmpz+Connection to Glances server failed ({} {})z9No Glances server found on {}. Trying fallback to SNMP...z Connection to server {} failedi� z (Bad username/password)z ({} {})Fr �.r )r r z'Client version: {} / Server version: {}zIClient and server not compatible: Client version: {} / Server version: {}T) r# Zinit�socketr. r0 r r" �errno�strerrorr$ r! �print�infor Zerrcode�errmsgr' r �splitr r r �statsZ set_plugins�ujson�loadsZ
getAllPluginsr% )r
Zclient_version�errZ
fall_back_msgr/ r r r �_login_glances] s6
z GlancesClient._login_glancesc C sD t jd� ddlm} || j| jd�| _| jj� s@| jd� dS dS )zLogin to a SNMP serverz Trying to grab stats by SNMP...r )�GlancesStatsClientSNMP)r r z Connection to SNMP server failedFT) r r8 Zglances.stats_client_snmpr@ r r r; Z
check_snmpr' )r
r@ r r r � _login_snmp� s
zGlancesClient._login_snmpc C sr | j jrd| _n | j� s dS | jdkr2| j� s2dS tjd� | jj| j � | j
r\tj d� nt | j | j d�| _
dS )zLogon to the server.r2 Fz.Load limits from the client configuration filez+Quiet mode is ON: Nothing will be displayed)r r T)r Z
snmp_forcer0 r? rA r r% r; Z load_limitsr r r8 r �screen)r
r r r �login� s
zGlancesClient.loginc C sL | j dkr| j� S | j dkr$| j� S | j� tjdj| j �� tjd� dS )z&Update stats from Glances/SNMP server.r r2 zUnknown server mode: {}r* N) r0 �update_glances� update_snmp�endr r+ r" r, r- )r
r r r �update� s
zGlancesClient.updatec C sR yt j| jj� �}W n( tjk
r* dS tk
r< dS X | jj|� dS dS )z�Get stats from Glances server.
Return the client/server connection status:
- Connected: Connection OK
- Disconnected: Connection NOK
� DisconnectedZ ConnectedN) r< r= r# ZgetAllr4 r. r r; rG )r
Z server_statsr r r rD � s z GlancesClient.update_glancesc C s, y| j j� W n tk
r" dS X dS dS )z�Get stats from SNMP server.
Return the client/server connection status:
- SNMP: Connection with SNMP server OK
- Disconnected: Connection NOK
rH ZSNMPN)r; rG r& )r
r r r rE � s
zGlancesClient.update_snmpc C s | j � s tjd� | j� | jS d}y�x�dr�| r�t� }| j� }tjdj|j � �� t� }| j
j | j
� tjdj|j � �� | j |j � }|dkr�|nd}| j
s�| jj| j
||| jd�}q(tj|� q(W W n6 tk
�r } ztj|� | j� W Y dd}~X nX | jS ) zMain client loop.z4The server version is not compatible with the clientFTz"Stats updated duration: {} secondsz#Stats exported duration: {} secondsr )Zduration� cs_statusr! N)rC r r+ rF r0 r
rG r% r" �getr; Zexportr r rB r! r Zsleepr& )r
Zexit_keyZcounterrI Zcounter_exportZadapted_refreshr( r r r �
serve_forever� s4
zGlancesClient.serve_foreverc C s | j s| jj� dS )zEnd of the client session.N)r rB rF )r
r r r rF s zGlancesClient.end)NNr F)r )r r r r r) �propertyr r' r0 �setterr? rA rC rG rD rE rK rF r r r r r ! s
* +r )r r< r4 r, r r r Zglances.compatr r r r Zglances.loggerr Zglances.stats_clientr Z glances.outputs.glances_cursesr Z
glances.timerr
r �objectr r r r r �<module>
s