File: //lib/python3.9/site-packages/cockpit/__pycache__/samples.cpython-39.pyc
a
�/�h�C � @ sv d dl Z d dlZd dlZd dlZd dlmZmZmZmZm Z m
Z
m Z d dl m
Z
e�ejd �Zded krnend ZdZeeef Ze�e�Zd"eee
e ee
e d �d
d �ZG d d
� d
e �ZG dd� d�Z G dd� de �Z G 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#G d d � d e �Z$G d d!� d!e �Z%e%e!e"e e e e e$e#g Z&dS )#� N)�Any� DefaultDict�Iterable�List�
NamedTuple�Optional�Tuple)�Handle�
SC_CLK_TCK�� �d z/sys/class/hwmon� )�rootfd�statfile�default�key�returnc
C s� zt j|t j| d�}W n ty* Y d S 0 zlzt �|d�}W nL ty� } z4|jtjkrht� d||� W Y d }~W t �
|� d S d }~0 0 W t �
|� n t �
|� 0 |r�|� |�t |� }|� d|�}|||� }zt
|�p�|W S t� y� Y d S 0 d S )N��dir_fd� zFailed to read %s: %s�
)�os�open�O_RDONLY�FileNotFoundError�read�OSError�errnoZENODEV�loggerZwarning�close�index�len�int�
ValueError) r r r r �fd�data�e�start�end� r) �3/usr/lib/python3.9/site-packages/cockpit/samples.py�
read_int_file% s*
� r+ c @ s. e Zd ZU eed< eed< eed< eed< dS )�SampleDescription�nameZunitsZ semantics� instancedN)�__name__�
__module__� __qualname__�str�__annotations__�boolr) r) r) r* r, C s
r, c @ s* e Zd ZU ee ed< edd�dd�ZdS )�Sampler� descriptionsN��samplesr c C s t �d S )N)�NotImplementedError)�selfr8 r) r) r* �sampleM s zSampler.sample)r/ r0 r1 r r, r3 �Samplesr; r) r) r) r* r5 J s
r5 c
@ s� e Zd Zeddddd�eddddd�eddddd�eddddd�ed ddd
d�ed ddd
d�ed ddd
d�ed
ddd
d�gZedd�dd�ZdS )�
CPUSamplerzcpu.basic.nice�millisec�counterF�r. zcpu.basic.userzcpu.basic.systemzcpu.basic.iowaitz
cpu.core.niceTz
cpu.core.userzcpu.core.systemzcpu.core.iowaitNr7 c C sB t d���"}|D �]
}|�d�s"q|�� d d� \}}}}}} |dd � pLd }
|
r�d} t|�t || � d� |
< t|�t || � d� |
< t|�t || � d� |
< t| �t || � d � |
< qd
} t|�t || � d�< t|�t || � d�< t|�t || � d�< t| �t || � d �< qW d � n1 �s40 Y d S ) Nz
/proc/stat�cpu� � zcpu.corez.nicez.userz.systemz.iowaitz cpu.basic)r �
startswith�splitr" � MS_PER_JIFFY) r: r8 �stat�linerA �user�nice�systemZ_idleZiowaitZcore�prefixr) r) r* r; ^ s"
zCPUSampler.sample�r/ r0 r1 r, r6 r< r; r) r) r) r* r= Q s � r= c @ sX e Zd Zeddddd�eddddd�eddddd�eddddd�gZed d
�d d �Zd S )
�
MemorySampler� memory.free�bytes�instantFr@ � memory.used�
memory.cached�memory.swap-usedNr7 c C s� t d�� }dd� |D �}W d � n1 s,0 Y d|d |d< d|d |d |d < d|d
|d |d < d|d
|d |d< d S )Nz
/proc/meminfoc S s, i | ]$}|� d d�\}}|t|�d���qS )�:� z kB
)rE r" �strip)�.0rH �k�vr) r) r* �
<dictcomp>} r
z(MemorySampler.sample.<locals>.<dictcomp>r ZMemFreerO ZMemTotalZ MemAvailablerR ZBuffersZCachedrS Z SwapTotalZSwapFreerT )r )r: r8 Zmeminfo�itemsr) r) r* r; { s
,zMemorySampler.samplerM r) r) r) r* rN s s �rN c @ sr e Zd ZU dZeee ed< eddddd�gZ e
e e e d�d d
��Z
e
e e d �d d
��Zedd�dd�ZdS )�CPUTemperatureSamplerN�sensors�cpu.temperatureZcelsiusrQ Tr@ )r r c c sJ zJt jdtj| d��$}t�|d��� �� }W d � n1 s>0 Y W n ty^ Y d S 0 |dkrrdd� }n.|dkr�d }n |dkr�d }n|d v r�d }nd S t�| �D ]�}|� d
�s�q�|�r>zX|�
d
d �}t j|tj| d��$}t�|d��� �� }W d � n1 �s 0 Y W n t�y0 Y q�Y n0 ||��s>q�|V q�d S ) Nr- r r Zatk0110c S s | dkS )NzCPU Temperaturer) )�labelr) r) r* �<lambda>� r
z:CPUTemperatureSampler.detect_cpu_sensors.<locals>.<lambda>Z cpu_thermalZcoretemp)Zk8tempZk10tempZ_inputZ_label) r r r r r �decoderW r �listdir�endswith�replace)r r$ r- � predicateZinput_filenameZlabel_filenamer` r) r) r* �detect_cpu_sensors� s6 6
8
z(CPUTemperatureSampler.detect_cpu_sensors)r c
c s� zt �ttjtjB �} W n ty, Y d S 0 | �~ t�| �D ]b}t j|tjtjB | d��6}t� |�D ]}t� d|� d|� �V qfW d � q>1 s�0 Y q>W d � n1 s�0 Y d S )Nr �/)
r r �
HWMON_PATHr r � O_DIRECTORYr rc r] rg )Ztop_fdZ
hwmon_nameZ subdir_fd�sensorr) r) r* � scan_sensors� s z"CPUTemperatureSampler.scan_sensorsr7 c C s� | j d u rtt�� �| _ | j D ]f}t|��:}t|�� �� �}|dkrVW d � d S W d � n1 sj0 Y |d |d |< q d S )Nr r r_ )r^ �listr] rl r r" r rW )r: r8 Z sensor_pathrk Z temperaturer) r) r* r; � s
0z CPUTemperatureSampler.sample)r/ r0 r1 r^ r r r2 r3 r, r6 � staticmethodr" r rg rl r< r; r) r) r) r* r] � s
�* r] c @ sX e Zd Zeddddd�eddddd�eddddd�ed dddd�gZed
d �d d
�Zd
S )� DiskSampler�
disk.all.readrP r? Fr@ �disk.all.written�
disk.dev.readT�disk.dev.writtenNr7 c
C s t d���}d}d}|D ]�}|�� �� }|d }|d }|d } |d }
|dkrPq|�d�r\q|d d� dv rz|d �� rzq|�d
�r�d |v r�qt| �d } t|
�d } || 7 }|| 7 }| |d
|< | |d |< q||d< ||d< W d � n1 s�0 Y d S )N�/proc/diskstatsr � � � �9zdm-)�sdZhdZvd���Znvme�p� rr rs rp rq )r rW rE rD �isdigitr" )
r: r8 � diskstatsZall_read_bytesZall_written_bytesrH �fieldsZ dev_major�dev_nameZnum_sectors_readZnum_sectors_written�
read_bytesZ
written_bytesr) r) r* r; � s2
zDiskSampler.samplerM r) r) r) r* ro � s �ro c @ s� e Zd ZU eddddd�eddddd�eddddd�eddddd�ed d
d dd�ed d
ddd�gZdZee ed< e dd�dd�Z
dS )�
CGroupSampler�cgroup.memory.usagerP rQ Tr@ �cgroup.memory.limit�cgroup.memory.sw-usage�cgroup.memory.sw-limit�cgroup.cpu.usager> r? �cgroup.cpu.shares�countN�
cgroups_v2r7 c C s� | j d u rtj�d�| _ | j r�d}t�|�D ]�\}}}}|�|d�}|sJq,t|dd�|d |< t|d�|d |< t|d d�|d
|< t|d �|d |< t|d
�|d |< t|dddd�}|r,|d |d |< q,n�d}t�|�D ]j\}}}}|�|d�}|s�q�t|dd�|d |< t|d�|d |< t|dd�|d
|< t|d�|d |< q�d} t�| �D ]V\}}}}|�| d�}|�sz�qXt|d�|d |< t|d�}
|
�rX|
d |d |< �qXd S ) Nz!/sys/fs/cgroup/cgroup.controllersz/sys/fs/cgroup/� zmemory.currentr r� z
memory.maxr� zmemory.swap.currentr� zmemory.swap.maxr� z
cpu.weightr� zcpu.stats
usage_usec)r r r� z/sys/fs/cgroup/memory/zmemory.usage_in_byteszmemory.limit_in_byteszmemory.memsw.usage_in_byteszmemory.memsw.limit_in_bytesz/sys/fs/cgroup/cpu/z
cpu.sharesz
cpuacct.usagei@B )r� r �path�exists�fwalkre r+ ) r: r8 Zcgroups_v2_pathr� �_r �cgroupZ
usage_usecZ memory_pathZcpu_pathZ
usage_nsecr) r) r* r; sD
zCGroupSampler.sample) r/ r0 r1 r, r6 r� r r4 r3 r< r; r) r) r) r* r� s
� r� c @ sz e Zd Zejdejd�Zeddddd�eddddd�gZe e
e d �d
d ��Z e e
e
e
e
f d �d d
��Zed�dd�ZdS )� CGroupDiskIOs: \bread_bytes: (?P<read>\d+).*\nwrite_bytes: (?P<write>\d+))�flags�disk.cgroup.readrP r? Tr@ �disk.cgroup.written)r$ r c C sV t jdtj| d��0}t�|d��� �� }|dd � W d � S 1 sH0 Y d S )Nr� r i � )r r r r r rb rW )r$ Z cgroup_fd� cgroup_namer) r) r* �get_cgroup_nameE s z CGroupDiskIO.get_cgroup_namec C s� t jdtj| d��b}t�|d�}t�tj|�}|rdt |�
d��}t |�
d��}||fW d � S W d � dS 1 sz0 Y d S )N�ior i r �write)r r ) r r r r r �re�searchr� �IO_REr" �group)r$ Zio_fdr% �match� proc_read�
proc_writer) r) r* � get_proc_ioM s zCGroupDiskIO.get_proc_io)r8 c
C s t �dtjtjB ���}|d }|d }t�|�D ]�}|d dk s0|d dkrNq0zNt j|tj|d��(}| �|�}| �|�\}} W d � n1 s�0 Y W n t t
t fy� Y q0Y n0 |� |d�| ||< |� |d�| ||< q0W d � n1 s�0 Y d S )Nz/procr� r� r �0rx r )
r r r r rj rc �O_PATHr� r� r �PermissionError�ProcessLookupError�get)
r: r8 Zproc_fdZreadsZwritesr� Zpid_fdr� r� r� r) r) r* r; [ s
0
zCGroupDiskIO.sampleN)r/ r0 r1 r� �compile�Sr� r, r6 rn r" r2 r� r r� r< r; r) r) r) r* r� > s �
r� c @ s< e Zd Zeddddd�eddddd�gZedd�d d
�ZdS ) �NetworkSampler�network.interface.txrP r? Tr@ �network.interface.rxNr7 c C s� t d��l}|D ]V}|�� }|d d dkr,q|d d d� }t|d �|d |< t|d �|d |< qW d � n1 sz0 Y d S ) Nz
/proc/net/devr rz rU rV r� rw r� )r rE r" )r: r8 Znetwork_samplesrH r Zifacer) r) r* r; v s
zNetworkSampler.samplerM r) r) r) r* r� p s �r� c @ s< e Zd Zeddddd�eddddd�gZedd�d d
�ZdS ) � MountSampler� mount.totalrP rQ Tr@ �
mount.usedNr7 c
C s� t d���}|D ]�}|d dkr q|�� ^}}}}}d|v rJd|�d�v rJqd|v r\d|v r\qzt�|�} W n ty� Y qY n0 | j}
|
| j } | |d |< | |
| j |d
|< qW d � n1 s�0 Y d S ) Nz /proc/mountsr rh z/loop�ro�,z
revokefs-fuseZflatpakr� r� )r rE r �statvfsr �f_frsize�f_blocks�f_bfree) r: r8 ZmountsrH Zfs_specZfs_fileZ _fs_vfstypeZ
fs_mntoptsZ_rest�resZfrsize�totalr) r) r* r; � s"
zMountSampler.samplerM r) r) r) r* r� � s �r� c @ s< e Zd Zeddddd�eddddd�gZedd�d d
�ZdS ) � BlockSampler�block.device.readrP r? Tr@ �block.device.writtenNr7 c
C s� t d��d}|D ]N}|�� �� ^
}}}}}}}}}}}t|�d |d |< t|�d |d |< qW d � n1 sr0 Y d S )Nrt r| r� r� )r rW rE r" )r: r8 r~ rH r� r� Z sectors_readZsectors_writtenr) r) r* r; � s
"zBlockSampler.samplerM r) r) r) r* r� � s �r� )Nr
)'r Zloggingr r� �typingr r r r r r r Z cockpit._vendor.systemd_ctypesr �sysconf�
sysconf_namesZUSER_HZrF ri r2 r< Z getLoggerr/ r r" rP r+ r, r5 r= rN r] ro r� r� r� r� r� ZSAMPLERSr) r) r) r* �<module> s@ $
"M2:2 �