File: //lib/python3.6/site-packages/glances/plugins/__pycache__/glances_docker.cpython-36.pyc
3
Eo�c�| � @ s& d Z ddlZddlZddlZddlmZ ddlmZmZm Z m
Z
ddl m Z ddl
mZ ddlmZmZ ddlmZ y ddlZdd lmZmZ W n6 ek
r� Z zd
Ze j d j e�� W Y ddZ[X nX d Zd
ddd�gZ ddddgZ ddddddd�Z G d d � d e�Z!G d d � d ej"�Z#d d!� Z$dS )"zDocker plugin.� N)�deepcopy)�iterkeys�
itervalues� nativestr� pretty_date)�logger)�
GlancesPlugin)�
sort_stats�glances_processes)�getTimeSinceLastUpdate)�parser�tzTz=Error loading Docker deps Lib. Docker plugin is disabled ({})F� cpu_percentz Container CPU consumption in %�%)�name� descriptionZy_unit�cpu�io�memory�networkzdisk IOzCPU consumptionzmemory consumptionZuptimezcontainer name�None)� io_countersr � memory_usage� cpu_timesr Nc s� e Zd ZdZd%� fdd� Z� fdd�Zdd� Zd d
� Zd d � Zd
d� Z e
j e
j dd� ��Z
dd� Zdd� Zdd� Zdd� Zdd� Zdd � Z� fd d �Zd&d d �Zd!d"� Zd#d$� Z� ZS )'�Pluginz\Glances Docker plugin.
stats is a dict: {'version': {...}, 'containers': [{}, {}]}
Nc sb t t| �j||td� || _|| _d| _| j� | _i | _ i | _
i | _ d| _ | j
� | jjd� dS )zInit the plugin.)�args�config�items_history_listTNr )�superr �__init__r r r Z
display_curse�connect�
docker_client� thread_list� network_old�io_old�sort_key�updateZ
refresh_timer�set)�selfr r )� __class__� �$/usr/lib/python3.6/glances_docker.pyr N s
zPlugin.__init__c s. xt | j�D ] }|j� q W tt| �j� dS )z+Overwrite the exit method to close threads.N)r r" �stopr r �exit)r( �t)r) r* r+ r- t s z Plugin.exitc C s dS )zReturn the key of the list.r r* )r( r* r* r+ �get_key{ s zPlugin.get_keyc C st yt | jd �}W n6 tk
rH } ztjdj|�� g }W Y dd}~X nX x$|D ] }xtD ]}|j|� qZW qPW |S )z~Overwrite the default export method.
- Only exports containers
- The key is the first container name
�
containersz&docker plugin - Docker export error {}N)r �stats�KeyErrorr �debug�format�export_exclude_list�pop)r( �ret�e� container�ir* r* r+ �
get_export s
zPlugin.get_exportc C sH y t j� }W n6 tk
rB } ztjdj|�� d}W Y dd}~X nX |S )z Connect to the Docker server.z.docker plugin - Can not connect to Docker ({})N)�dockerZfrom_env� Exceptionr �errorr4 )r( r7 r8 r* r* r+ r � s zPlugin.connectc C s. | j d�}t|�dkrdS |d j� dkS dS )z�Return the all tag of the Glances/Docker configuration file.
# By default, Glances only display running containers
# Set the following key to True to display all containers
all=True
�allr F�trueN)Zget_conf_value�len�lower)r( Zall_tagr* r* r+ �_all_tag� s
zPlugin._all_tagc # C sr | j � }ts| jdkr | jS | jdk�rPy| jj� |d< W nR tk
r� } z6tjdj | j
|�� d| jkrr| jd= g | jd< | jS d}~X nX y | jj j | j
� d�p�g }W n@ tk
r� } z$tjdj | j
|�� g | jd< | jS d}~X nX xR|D ]J}|j| jkr�tjdj | j
|jdd � �� t|�}|| j|j< |j� q�W tt| j��td
d � |D �� }x@|D ]8}tjd j | j
|dd � �� | j| j� | j|= �qfW g |d< �x�|D �]�}| jt|j���sΐq�i }| j� |d
< t|j�|d< |j|d< |jj|d< |j d d |d< g |d< |j d j dd��rL|d j |j d j dg �� |j d j dd��r||d j |j d j dg �� |d �s�d|d< |d d/k�r�| j |j| j|j j�|d< |d j dd�|d< | j |j| j|j j�|d < |d j d d�|d < |d j d d�dk �r*|d |d d 8 < | j!|j| j|j j�|d < |d j d!d�|d"< |d j d#d�|d$< | j"|j| j|j j�|d%< |d% j d&d�|d'< |d% j d(d�|d)< t#t$j%|j d d* �j&t'j(� �j)dd+��|d,< nXi |d< d|d< i |d < d|d-< i |d < d|d"< d|d$< i |d%< d|d'< d|d)< d|d,< |d j*|� �q�W n | jd.k�r\t+|�\| _,| _| jS )0z+Update Docker stats using the input method.NZlocal�versionz*{} plugin - Cannot get Docker version ({})r0 )r? z+{} plugin - Cannot get containers list ({})z*{} plugin - Create thread for container {}� c S s g | ]
}|j �qS r* )�id)�.0�cr* r* r+ �
<listcomp>� s z!Plugin.update.<locals>.<listcomp>z,{} plugin - Stop thread for old container {}�keyr ZIdZImageZState�Status�CommandZConfigZ
EntrypointZCmd�running�pausedr �totalr r �usager �cacher �iorZio_r�iowZio_wr �rxZ
network_rx�txZ
network_txZ StartedAt)Ztzinfo�Uptime�memory_percentZsnmp)rM rN )-Zget_init_value�import_error_tagr! r1 Z input_methodrD r= r r> r4 Z plugin_namer0 �listrC rF r" r3 �ThreadDockerGrabber�startr' r r, Z
is_displayr r r/ ZimageZtagsZattrs�get�extend�get_docker_cpu�get_docker_memory�
get_docker_io�get_docker_networkr r �parseZ
astimezoner
Ztzlocal�replace�append�sort_docker_statsr% ) r( r1 r8 r0 r9 r. Zabsent_containers� container_idZcontainer_statsr* r* r+ r& � s�
z
Plugin.updatec C sF ddi}y||d d |d d d d�}|d d |d d d d�}|d j d d
�|d < |d d
kr�t|d d d pzg �|d < W n> tk
r� } z"tjd
j||�� tj|� W Y d
d
}~X n�X y<|d |d }|d |d }|| |d d |d< W n@ tk
�r@ } z"tjdj||�� tj|� W Y d
d
}~X nX |S )z�Return the container CPU usage.
Input: id is the full container id
all_stats is the output of the stats method of the Docker API
Output: a dict {'total': 1.49}
rO g � cpu_statsZsystem_cpu_usageZ cpu_usageZ total_usage)�systemrO Z precpu_statsZ online_cpusNZnb_coreZ percpu_usagez;docker plugin - Cannot grab CPU usage for container {} ({})rh g Y@z>docker plugin - Cannot compute CPU usage for container {} ({}))r\ rA r2 r r3 r4 � TypeError) r( rf � all_statsrg r Zprecpur8 Z cpu_deltaZsystem_cpu_deltar* r* r+ r^ C s(
zPlugin.get_docker_cpuc C s� i }y�|d d |d< |d d |d< d|d d krL|d d d |d< n.d|d d krr|d d d |d< nd|d< |d d j dd�|d< |d j d d�|d < W nB ttfk
r� } z"tjd
j||�� tj|� W Y dd}~X nX |S ) z�Return the container MEMORY.
Input: id is the full container id
all_stats is the output of the stats method of the Docker API
Output: a dict {'rss': 1015808, 'cache': 356352, 'usage': ..., 'max_usage': ...}
� memory_statsrP �limitZrssr1 Z total_rssNrQ Z max_usagez;docker plugin - Cannot grab MEM usage for container {} ({}))r\ r2 ri r r3 r4 )r( rf rj rk r8 r* r* r+ r_ l s zPlugin.get_docker_memoryc C s i }y |d }W n< t k
rL } z tjdj||�� tj|� |S d}~X nX y$|d d |d< |d d |d< W n> t k
r� } z"tjd j||�� tj|� W Y dd}~X n^X td
j|��|d < || jk�r|d | j| d |d < |d | j| d |d
< || j|< |S )az Return the container network usage using the Docker API (v1.0 or higher).
Input: id is the full container id
Output: a dict {'time_since_update': 3000, 'rx': 10, 'tx': 65}.
with:
time_since_update: number of seconds elapsed between the latest grab
rx: Number of bytes received
tx: Number of bytes transmitted
Znetworksz;docker plugin - Cannot grab NET usage for container {} ({})NZeth0Zrx_bytesZ
cumulative_rxZtx_bytesZ
cumulative_txzIdocker plugin - Cannot grab network interface usage for container {} ({})z
docker_net_{}�time_since_updaterT rU )r2 r r3 r4 r r# )r( rf rj Z network_newZ net_statsr8 r* r* r+ ra � s(
zPlugin.get_docker_networkc C s, i }y |d }W n< t k
rL } z tjdj||�� tj|� |S d}~X nX y@|d }dd� |D �d d |d < d
d� |D �d d |d < W n< ttt tfk
r� } ztjdj||�� W Y dd}~X n^X td j|��|d
< || jk�r |d | j| d |d< |d | j| d |d< || j|< |S )aq Return the container IO usage using the Docker API (v1.0 or higher).
Input: id is the full container id
Output: a dict {'time_since_update': 3000, 'ior': 10, 'iow': 65}.
with:
time_since_update: number of seconds elapsed between the latest grab
ior: Number of bytes read
iow: Number of bytes written
Z blkio_statsz@docker plugin - Cannot grab block IO usage for container {} ({})N�io_service_bytes_recursivec S s g | ]}|d j � dkr|�qS )�op�read)rB )rG r: r* r* r+ rI � s z(Plugin.get_docker_io.<locals>.<listcomp>r �valueZcumulative_iorc S s g | ]}|d j � dkr|�qS )ro �write)rB )rG r: r* r* r+ rI � s Zcumulative_iowz docker_io_{}rm rR rS ) r2 r r3 r4 ri �
IndexError�AttributeErrorr r$ )r( rf rj Zio_newZio_statsr8 rn r* r* r+ r` � s&
$
zPlugin.get_docker_ioc C s t jt jd �S )z:Return the user ticks by reading the environment variable.�
SC_CLK_TCK)�os�sysconf�
sysconf_names)r( r* r* r+ �get_user_ticks� s zPlugin.get_user_ticksc C s
| j d S )zZReturn stats for the action.
Docker will return self.stats['containers']
r0 )r1 )r( r* r* r+ �get_stats_action� s zPlugin.get_stats_actionc sF t t| �j� d| jkr dS �x"| jd D �]}i i d�| j|| j� < d|kr�d|d kr�| j|d d |d d |d d�}|d kr�| j|d d dd
�}|| j|| j� d d < d |ko�d
|d kr*| j|d d
|d d |d d |d d�}|d k�r$| j|d d
|d d dd�}|| j|| j� d d < q*W dS )zUpdate stats views.r0 F)r �memr rO r Z_cpu)�header�
action_key�DEFAULT)r| �
decorationr rP rl Z_mem)�maximumr| r} r{ )r� r| T)r r � update_viewsr1 Zviewsr/ Z get_alert)r( r: Zalert)r) r* r+ r� � s"
$*
zPlugin.update_viewsc > C s g }| j s0d| j ks0t| j d �dks0| j� r4|S djd�}|j| j|d�� djt| j d ��}|j| j|�� djt| j �}|j| j|�� |j| j� � |j| j� � t | j
dk r�| j
j d d
d d �nd tt | j d d
d� d�d ��}djd|d�}|j| j|| jdk�rdnd�� djd�}|j| j|�� djd�}|j| j|�� djd�}|j| j|| jdk�rxdnd�� d jd �}|j| j|| jd k�r�dnd�� d jd �}|j| j|�� d jd!�}|j| j|�� d"jd#�}|j| j|�� d jd$�}|j| j|�� d"jd%�}|j| j|�� d&jd'�}|j| j|�� �x�| j d D �]�}|j| j� � |j| j| j
||d(��� | j|d �}dj|d dd)� �}|j| j||�� |d �r�dj|d �}n
djd*�}|j| j||�� yd+j|d, d- �}W n tk
�r& djd*�}Y nX |j| j|| j|d d,d.d/��� y d j| j|d0 d1 ��}W n tk
�r� d jd*�}Y nX |j| j|| j|d d2d.d/��� y d j| j|d0 d3 ��}W n tk
�r� d jd*�}Y nX |j| j|�� d4}y4| jt|d5 d6 |d5 d7 ��| } d j| �}W n tk
�rP d jd*�}Y nX |j| j|�� y4| jt|d5 d8 |d5 d7 ��| } d"j| �}W n tk
�r� d"jd*�}Y nX |j| j|�� |j�r�d9}
d:}nd;}
d<}y8| jt|d= d> |d= d7 |
��| } d j| �}W n tk
�r: d jd*�}Y nX |j| j|�� y8| jt|d= d? |d= d7 |
��| } d"j| �}W n tk
�r� d"jd*�}Y nX |j| j|�� |d' dk �r�djd@j|d' ��}n
djd*�}|j| j|dAdB�� �qVW |S )Cz2Return the dict to display in the curse interface.r0 r z{}Z
CONTAINERSZTITLEz {}z
sorted by {}Nr< Z
max_name_size� )�defaultc S s t | d �S )Nr )rA )�xr* r* r+ �<lambda>0 s z"Plugin.msg_curse.<locals>.<lambda>)rJ r z {:{width}}�Name)�widthZSORTr~ z{:>10}rK rV z{:>6}zCPU%r z{:>7}ZMEMr z/{:<7}�MAXzIOR/sz {:<7}zIOW/szRx/szTx/sz {:8}rL )r9 � max_width�
�_z{:>6.1f}r rO r )�itemrJ Zoptionr rP r{ rl �Br rR rm rS � � � �br rT rU � T)Z
splittable)r1 rA Z is_disabledr4 rd Zcurse_add_line�sort_for_humanr% Zcurse_new_line�minr Z
get_int_value�max� _msg_name�container_alertr2 Z get_viewsZ auto_unit�intZbyte�join) r( r r� r7 �msgZname_max_widthr9 �statusZunitrq Zto_bitr* r* r+ � msg_curse s� ,
"
"
"
" " &&
zPlugin.msg_cursec C s |d d|� }dj ||d�S )zBuild the container name.r Nz {:{width}})r� )r4 )r( r9 r� r r* r* r+ r� � s zPlugin._msg_namec C s, |dkr dS |dkrdS |dkr$dS dS dS ) z Analyse the container status.rM ZOKZexitedZWARNINGZdeadZCRITICALZCAREFULNr* )r( r� r* r* r+ r� � s zPlugin.container_alert)NN)NN)�__name__�
__module__� __qualname__�__doc__r r- r/ r; r rC r Z_check_decoratorZ_log_result_decoratorr& r^ r_ ra r` ry rz r� r� r� r� �
__classcell__r* r* )r) r+ r H s* &
) /0 "
r c sT e Zd ZdZ� fdd�Zdd� Zedd� �Zejdd� �Zdd
d �Z d d
� Z
� Z S )rZ zD
Specific thread to grab docker stats.
stats is a dict
c sJ t t| �j� tj� | _|| _|jdd�| _i | _ t
j dj | jj
�� dS )zLInit the class.
container: instance of Docker-py Container
T)�decodez.docker plugin - Create thread for container {}N)r rZ r � threadingZEvent�_stopper�
_containerr1 �
_stats_stream�_statsr r3 r4 r )r( r9 )r) r* r+ r � s
z ThreadDockerGrabber.__init__c C sn y.x(| j D ] }|| _tjd� | j� r
P q
W W n: tk
rh } z tjdj|�� | j � W Y dd}~X nX dS )z_Grab the stats.
Infinite loop, should be stopped by calling the stop() method
g�������?z0docker plugin - Exception thrown during run ({})N)
r� r� �timeZsleep�stoppedr= r r3 r4 r, )r( r: r8 r* r* r+ �run� s
zThreadDockerGrabber.runc C s | j S )z
Stats getter.)r� )r( r* r* r+ r1 � s zThreadDockerGrabber.statsc C s
|| _ dS )z
Stats setter.N)r� )r( rq r* r* r+ r1 � s Nc C s" t jdj| jj�� | jj� dS )zStop the thread.z-docker plugin - Close thread for container {}N)r r3 r4 r� r r� r' )r( Ztimeoutr* r* r+ r, � s zThreadDockerGrabber.stopc C s
| j j� S )z%Return True is the thread is stopped.)r� Zis_set)r( r* r* r+ r� � s zThreadDockerGrabber.stopped)N) r� r� r� r� r r� �propertyr1 �setterr, r� r� r* r* )r) r+ rZ � s
rZ c C sJ t j}d}|dkr d}d}n |d
kr(d}t| d ||t jdkd � || fS ) Nr rW r �usernamer r r0 r )Z sorted_byZsorted_by_secondary�reverse)r� r r )r
r% �sort_stats_processes)r1 Zsort_byZsort_by_secondaryr* r* r+ re � s re )%r� rv r� r� �copyr Zglances.compatr r r r Zglances.loggerr Z glances.plugins.glances_pluginr Zglances.processesr r� r
Z
glances.timerr r< Zdateutilr r
r= r8 rX Zwarningr4 r r5 r� r ZThreadrZ re r* r* r* r+ �<module>
s> " k9