Current File : //usr/lib/python3.9/site-packages/cockpit/__pycache__/samples.cpython-39.pyc
a
��hD � @ sz d dl Z d dlZd dlZd dlZd dlmZmZmZmZm Z m
Z
m Z m Z d dl
mZ e�ejd �Zded krrend ZdZeeef Ze�e�Zd"eee e ee e d �d
d �ZG d d
� d
e
�Z G 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�Callable� 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)�selfr9 r* r* r+ �sampleM s zSampler.sample)r0 r1 r2 r r- r4 �Samplesr< r* r* r* r+ r6 J s
r6 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.iowaitNr8 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; r9 �stat�linerB �user�nice�systemZ_idleZiowaitZcore�prefixr* r* r+ r< ^ s"
zCPUSampler.sample�r0 r1 r2 r- r7 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�instantFrA � memory.used�
memory.cached�memory.swap-usedNr8 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
)rF r# �strip)�.0rI �k�vr* r* r+ �
<dictcomp>} r z(MemorySampler.sample.<locals>.<dictcomp>r ZMemFreerP ZMemTotalZ MemAvailablerS ZBuffersZCachedrT Z SwapTotalZSwapFreerU )r )r; r9 Zmeminfo�itemsr* r* r+ r< { s
,zMemorySampler.samplerN r* r* r* r+ rO s s �rO 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.temperatureZcelsiusrR TrA )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 �decoderX r �listdir�endswith�replace)r r% r. � predicateZinput_filenameZlabel_filenamera 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 rd r^ rh )Ztop_fdZ
hwmon_nameZ subdir_fd�sensorr* r* r+ � scan_sensors� s z"CPUTemperatureSampler.scan_sensorsr8 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^ rm r r# r rX )r; r9 Z sensor_pathrl Z temperaturer* r* r+ r< � s
0z CPUTemperatureSampler.sample)r0 r1 r2 r_ r r r3 r4 r- r7 � staticmethodr# r rh rm 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.readrQ r@ FrA �disk.all.written�
disk.dev.readT�disk.dev.writtenNr8 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� rs rt rq rr )r rX rF rE �isdigitr# )
r; r9 � diskstatsZall_read_bytesZall_written_bytesrI �fieldsZ dev_major�dev_nameZnum_sectors_readZnum_sectors_written�
read_bytesZ
written_bytesr* r* r+ r< � s2
zDiskSampler.samplerN r* r* r* r+ rp � s �rp 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.usagerQ rR TrA �cgroup.memory.limit�cgroup.memory.sw-usage�cgroup.memory.sw-limit�cgroup.cpu.usager? r@ �cgroup.cpu.shares�countN�
cgroups_v2r8 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�fwalkrf r, ) r; r9 Zcgroups_v2_pathr� �_r �cgroupZ
usage_usecZ memory_pathZcpu_pathZ
usage_nsecr* r* r+ r< sD
zCGroupSampler.sample) r0 r1 r2 r- r7 r� r r5 r4 r= r< r* r* r* r+ r� s
� r� c @ s| e Zd Zejdejd�Zeddddd�eddddd�gZe e
e d �d
d ��Z e e
e
e
e
f d �d d
��Zedd�dd�ZdS )� CGroupDiskIOs: \bread_bytes: (?P<read>\d+).*\nwrite_bytes: (?P<write>\d+))�flags�disk.cgroup.readrQ r@ TrA �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 rc rX )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_ioNr8 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 �0ry r )
r
r r r rk rd �O_PATHr� r� r �PermissionError�ProcessLookupError�get)
r; r9 Zproc_fdZreadsZwritesr� Zpid_fdr� r� r� r* r* r+ r< [ s
0
zCGroupDiskIO.sample)r0 r1 r2 r� �compile�Sr� r- r7 ro r# r3 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.txrQ r@ TrA �network.interface.rxNr8 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 r{ rV rW r� rx r� )r rF r# )r; r9 Znetwork_samplesrI r� Zifacer* r* r+ r< v s
zNetworkSampler.samplerN r* r* r* r+ r� p s �r� c @ s< e Zd Zeddddd�eddddd�gZedd�d d
�ZdS ) � MountSampler� mount.totalrQ rR TrA �
mount.usedNr8 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 ri z/loop�ro�,z
revokefs-fuseZflatpakr� r� )r rF r �statvfsr �f_frsize�f_blocks�f_bfree) r; r9 ZmountsrI Zfs_specZfs_fileZ _fs_vfstypeZ
fs_mntoptsZ_rest�resZfrsize�totalr* r* r+ r< � s"
zMountSampler.samplerN r* r* r* r+ r� � s �r� c @ s< e Zd Zeddddd�eddddd�gZedd�d d
�ZdS ) � BlockSampler�block.device.readrQ r@ TrA �block.device.writtenNr8 c
C s� t d��d}|D ]N}|�� �� ^
}}}}}}}}}}}t|�d |d |< t|�d |d |< qW d � n1 sr0 Y d S )Nru r} r� r� )r rX rF r# )r; r9 r rI r� r� Z sectors_readZsectors_writtenr* r* r+ r< � s
"zBlockSampler.samplerN r* r* r* r+ r� � s �r� )Nr )(r Zloggingr r� �typingr r r r r r r r Z cockpit._vendor.systemd_ctypesr
�sysconf�
sysconf_namesZUSER_HZrG rj r3 r= Z getLoggerr0 r r# rQ r, r- r6 r> rO r^ rp r� r� r� r� r� ZSAMPLERSr* r* r* r+ �<module> s@ (
"M2:2 �
Mini Shell