????
Your IP : 3.144.40.238
3
Eo�cl' � @ s� d Z ddlZddlZddlZddlmZmZmZ ddlm Z m
Z
ddl m Z m
Z
ddlmZ ddlmZ ddlmZ dd lmZ G d
d � d e�ZdS ) z;Manage the Glances client browser (list of Glances server).� N)�Fault�
ProtocolError� ServerProxy)�
GlancesClient�GlancesClientTransport)�logger� LOG_FILENAME)�GlancesPasswordList)�GlancesStaticServer)�GlancesAutoDiscoverServer)�GlancesCursesBrowserc @ sb e Zd ZdZddd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dS )�GlancesClientBrowserzEThis class creates and manages the TCP client browser (servers list).Nc C sJ || _ || _d | _d | _| j� | j js2t� | _nd | _t| j d�| _ d S )N)�args)
r �config�
static_server�password�loadZdisable_autodiscoverr �autodiscover_serverr �screen)�selfr r � r �$/usr/lib/python3.6/client_browser.py�__init__ s
z GlancesClientBrowser.__init__c C s t | jd�| _t| jd�| _dS )z:Load server and password list from the configuration file.)r N)r
r r �GlancesPasswordr )r r r r r 0 s zGlancesClientBrowser.loadc C s8 g }| j jr4| jj� }| jdk r4| jj� | jj� }|S )zmReturn the current server list (list of dict).
Merge of static + autodiscover servers list.
N)r Zbrowserr �get_servers_listr )r �retr r r r 8 s
z%GlancesClientBrowser.get_servers_listc C sx |d dkr`|d dkr@| j j|d �}|dk r@| j j|�|d< dj|d |d |d |d
�S d j|d |d
�S dS ) z)Return the URI for the given server dict.r � �status� PROTECTED�nameNzhttp://{}:{}@{}:{}�username�ip�portz http://{}:{})r � get_password� sha256_hash�format)r �server�clear_passwordr r r Z __get_uriF s z GlancesClientBrowser.__get_uric C s� | j |�}t� }|jd� yt||d�}W n6 tk
r` } ztjdj||�� W Y dd}~X �n`X yTdtj |j
� �d }dj|�|d< tj |j � �d |d
< tj |j � �d |d < W n� t
jttfk
r� } z tjd j||�� d
|d< W Y dd}~X n� tk
�rZ } zD|jdk�r*d|d< d|d< nd
|d< tjdj||j|j�� W Y dd}~X nfX d|d< y$tj |j� �d }dj|�|d< W n6 tk
�r� } ztjd j||�� W Y dd}~X nX |S )z?Update stats for the given server (picked from the server list)� )Z transportz,Client browser couldn't create socket {}: {}N�d Zidlez{:.1f}� cpu_percentZpercentZ mem_percentZhr_namez&Error while grabbing stats form {}: {}�OFFLINEr i� r r z!Cannot grab stats from {} ({} {})�ONLINEZmin5z{:.2f}� load_min5)� _GlancesClientBrowser__get_urir Z set_timeoutr � Exceptionr Zwarningr% �ujson�loadsZgetCpuZgetMemZ getSystem�socket�errorr �KeyError�debugr Zerrcode�errmsgZgetLoad)r r& Zuri�t�s�er* r- r r r Z__update_statsS s8
&
*$z#GlancesClientBrowser.__update_statsc C s� t jdj|�� | jjdj|d |d �dd� |d dkr�| jj|d �}|dksl| j� | jj d d
kr�| jjd j|d �d d
�}|dk r�| j d| jj
|�� t j dj|d �� | j }|d |_
|d |_|d |_|d |_t| j|d d�}|j� �s$| jjdj|d t�� | j d d� n\|j� }yt jdj|d �� W n tk
�rZ Y n&X |dk�rt| j d d� n | j d d� d| j_dS )z$Connect and display the given serverzSelected server {}zConnect to {}:{}r r" � )Zdurationr Nr r zPassword needed for {}: T)Zis_inputz'Connect Glances client to the {} server�keyr! r )r r Zreturn_to_browserz7Sorry, cannot connect to '{}'
See '{}' for more detailsr+ z,Disconnect Glances client from the {} serverZsnmpZSNMPr, )r r5 r% r Z
display_popupr r# r �
active_server�set_in_selectedr$ �infor �clientr" r r r Zloginr �
serve_forever�
IndexError)r r& r' Z args_serverr? Zconnection_typer r r Z__display_server� s:
z%GlancesClientBrowser.__display_serverc C s� i }x�| j js�tjdj| j� �� xZ| j� D ]N}|d }|j|d�}|dksX|j� dkr,tj | j
|gd�}|||< |j � q,W | j j dkr�| j j
| j� � q| j| j� | j j � qW x|j� D ] }|j� q�W dS )zMain client loop.z*Iter through the following server list: {}r; NF)�targetr )r Zis_endr r5 r% r �getZis_alive� threadingZThread�#_GlancesClientBrowser__update_stats�startr< �update�%_GlancesClientBrowser__display_server�values�join)r Z thread_list�vr; Zthreadr r r Z__serve_forever� s
z$GlancesClientBrowser.__serve_foreverc
C s z| j � S | j� X dS )z�Wrapper to the serve_forever function.
This function will restore the terminal to a sane state
before re-raising the exception and generating a traceback.
N)�$_GlancesClientBrowser__serve_forever�end)r r r r r@ � s z"GlancesClientBrowser.serve_foreverc C sR | j jt| jj� �kr:| jj| j jt| jj� � ||� n| jj| j j||� dS )z9Set the (key, value) for the selected server in the list.N)r r< �lenr r r Z
set_server)r r; �valuer r r r= � s z$GlancesClientBrowser.set_in_selectedc C s | j j� dS )z"End of the client browser session.N)r rM )r r r r rM � s zGlancesClientBrowser.end)NN)�__name__�
__module__� __qualname__�__doc__r r r r. rE rH rL r@ r= rM r r r r r
s
2@
r
)rS r0 r2 rD Zglances.compatr r r Zglances.clientr r Zglances.loggerr r Zglances.password_listr r Zglances.static_listr
Zglances.autodiscoverr Z&glances.outputs.glances_curses_browserr �objectr
r r r r �<module>
s