File manager - Edit - /opt/saltstack/salt/lib/python3.10/site-packages/relenv/build/common/__pycache__/ui.cpython-310.pyc
Back
o ;jR6 � @ s� d Z ddlmZ ddlZddlZddlZddlZddlmZm Z m Z er;ddl Z ddl Z ddl mZmZ ddlmZ ndZddlmZ dd lmZ e�e�Zd Zd Zd Z d Z dZ dZ ej �!d�rcdZ nej"dkrzej �!d�Z#ej �!d�Z$e#sze$szdZ e r�g d�Z%dZ&dZ'dZ(dZ)n g d�Z%dZ&dZ'dZ(d Z)d Z*dFd d!�Z+G d"d#� d#�Z,e,� Z-G d$d%� d%e �Z. dGdHd/d0�Z/ 1dIdJd8d9�Z0dKd:d;�Z1dLd=d>�Z2dMdBdC�Z3G dDdE� dEej4�Z5dS )Nz$ UI and build statistics utilities. � )� annotationsN)� TYPE_CHECKING� TypedDict�cast)�MutableMapping�Sequence)�Event)�DATA_DIR� )�CICDz[0;32mz[1;33mz[0;31mz[0mz[FTZ RELENV_ASCIIF�win32Z WT_SESSIONZ TERM_PROGRAM) u ⠋u ⠙u ⠹u ⠸u ⠼u ⠴u ⠦u ⠧u ⠇u ⠏u ◯u ✓u ✗)�|�/�-�\�o�+�Xzbuild_stats.json�return� pathlib.Pathc C s t t S )a Get the path to the build statistics file. Returns: Path to build_stats.json in the relenv data directory. Note: This is a function rather than a module-level constant to avoid import-time dependencies on DATA_DIR, following CPython conventions. )r �BUILD_STATS_FILENAME� r r �J/opt/saltstack/salt/lib/python3.10/site-packages/relenv/build/common/ui.py�_get_build_stats_fileK s r c @ s: e Zd ZdZddd�Zdd d �Zdd d �Zdddd�Zd S )� SpinnerStatez�Thread-safe spinner state management. Tracks the animation frame index for each named spinner to ensure smooth, consistent animation across multiple UI updates. r �Nonec C s i | _ t�� | _dS )z2Initialize empty spinner state with thread safety.N)�_state� threading�Lock�_lock)�selfr r r �__init___ s zSpinnerState.__init__�name�str�intc C s8 | j � | j�|d�W d � S 1 sw Y dS )z�Get the current frame index for a named spinner. Args: name: The spinner identifier Returns: The current frame index (0 if spinner hasn't been used yet) r N�r r �get�r r" r r r r& d s $�zSpinnerState.getc C sD | j � | j�|d�d | j|< W d � dS 1 sw Y dS )zoIncrement the frame index for a named spinner. Args: name: The spinner identifier r r Nr% r' r r r � incrementp s "�zSpinnerState.incrementN� str | Nonec C sr | j �, |du r| j�� n|| jv r | j|= W d � dS W d � dS W d � dS 1 s2w Y dS )zrReset spinner state. Args: name: The spinner to reset, or None to reset all spinners N)r r �clearr' r r r �resety s ��"�zSpinnerState.reset)r r )r" r# r r$ )r" r# r r �N)r" r) r r )�__name__� __module__� __qualname__�__doc__r! r& r( r+ r r r r r X s r c @ s* e Zd ZU dZded< ded< ded< dS )� BuildStatsz-Structure for tracking build step statistics.r$ � avg_lines�samples� last_linesN)r- r. r/ r0 �__annotations__r r r r r1 � s r1 �events�6MutableMapping[str, multiprocessing.synchronize.Event]� processes�,MutableMapping[str, multiprocessing.Process]�fails� Sequence[str]�flipstat�#dict[str, tuple[int, float]] | Noner c C s� t r tj�� dS g }| D ]F}| | �� s dt� t� �}n1||v r;t�|�t t � }t | }t� |� dt � |� �}n||v rGdt � t� �}ndt � t� �}|�|� q |�dt � tj�d� tj�d�|�� tj�� dS )a� Prints the UI during the relenv building process. :param events: A dictionary of events that are updated during the build process :type events: dict :param processes: A dictionary of build processes :type processes: dict :param fails: A list of processes that have failed :type fails: list :param flipstat: Deprecated parameter, no longer used :type flipstat: dict, optional N� z � � )r �sys�stdout�flush�is_set�YELLOW�SYMBOL_PENDING�_spinner_stater& �len�SPINNER_FRAMESr( �GREEN�RED� SYMBOL_FAILED�SYMBOL_SUCCESS�append�END�write�join) r6 r8 r: r< Zuiliner" �status� frame_idx�spinnerr r r �print_ui� s( rU �build� line_counts�MutableMapping[str, int]� build_stats�dict[str, BuildStats]�phaser# c C s t r tj�� dS ttd�si t_tj}t| �d }||v r+|| }tj�t | � n|dkr9d|v r9tj�d� |||< |dkrCdnd} tj�d � tj�| � d�� tj�d � | D �]#} | | � � srt � t � t � �} d } d } n�| |v �r?t�| �tt� }t| }t�| � t� |� t � �} |dkr�d nd}|�| d�}|dkr�|}|�| � d�d�}|dkr�tdt|| d ��}|� d|d�d�} t|d �}tr�d| dd| n d| dd| }d|� d �} nt|} d } no| |v �r:|| d dk�r:|| d }tdt|| d ��}|� d|d�d�} t|d �}t�r)d| dd| n d| dd| }d|� d �} n#|} d } n | |v �rQt� t� t � �} d } d } n t� t� t � �} d } d } | d �dd� }| d!�}tj�d � tj�| � d|� d|� | � d�� q\tj�� dS )"aj Prints an expanded UI with progress bars during the relenv building process. :param events: A dictionary of events that are updated during the build process :type events: dict :param processes: A dictionary of build processes :type processes: dict :param fails: A list of processes that have failed :type fails: list :param line_counts: Current line counts for each step :type line_counts: MutableMapping[str, int] :param build_stats: Historical build statistics :type build_stats: dict :param phase: The current phase ("download" or "build") :type phase: str N� _phase_state� rV �download� Z DownloadsZBuildsz [Ku� ────────────────────────────────────────────────────────────────────── ZPendingr@ Z DownloadingZBuildingr Z_total�d r> Z3d�%� u █u ░� �#r z [�]r2 ZFailedZDonez<20z<12)r rA rB rC �hasattr�print_ui_expandedr\ rH rP �MOVEUPrD rE rF rO rG r& rI r( rJ �minr$ � USE_UNICODErK rL rM )r6 r8 r: rW rY r[ Z phase_stateZ num_linesZ prev_linesZ phase_namer" Z status_symbolZ status_textZ progress_barrS rT Z phase_actionZ current_lines� downloaded�totalZprogressZfilledZbarr2 Z name_displayZstatus_displayr r r rg � sz , . $rg c C s� t � } | �� s i S z&ddl}t| ��}|�|�}td|�W d � W S 1 s(w Y W dS t|jfyB t� d� i Y S w )z� Load historical build statistics from disk. :return: Dictionary mapping step names to their statistics :rtype: dict r NrZ z*Failed to load build stats, starting fresh) r �exists�json�open�loadr �OSError�JSONDecodeError�log�warning)� stats_filern �f�datar r r �load_build_stats? s (� �rx �statsc C s� z1ddl }t� }|jjddd� t|d��}|j| |dd� W d � W dS 1 s*w Y W dS ty@ t�d� Y dS w ) z� Save build statistics to disk. :param stats: Dictionary mapping step names to their statistics :type stats: dict r NT)�parents�exist_ok�wr] )�indentzFailed to save build stats) rn r �parent�mkdirro �dumprq rs rt )ry rn ru rv r r r �save_build_statsT s &� �r� � step_name� line_countr$ c C sl t � }| |vrt|d|d�|| < n || d }td| d| �}t||| d d |d�|| < t|� dS )a Update statistics for a build step with a new sample. Uses exponential moving average with weight 0.7 for new samples. :param step_name: Name of the build step :type step_name: str :param line_count: Number of log lines for this build :type line_count: int r )r2 r3 r4 r2 gffffff�?g333333�?r3 N)rx r1 r$ r� )r� r� ry Zold_avgZnew_avgr r r �update_build_statsf s � r� c s, e Zd ZdZd� fdd � Zdd d �Z� ZS )�LineCountHandlerz� Custom logging handler that counts log lines for progress tracking. This handler increments a counter in a shared multiprocessing dict for each log message emitted, enabling real-time progress estimation. r� r# � shared_dictrX r r c s t � �� || _|| _dS )a Initialize the line count handler. :param step_name: Name of the build step being tracked :type step_name: str :param shared_dict: Multiprocessing-safe dict for sharing counts :type shared_dict: MutableMapping[str, int] N)�superr! r� r� )r r� r� �� __class__r r r! � s zLineCountHandler.__init__�record�logging.LogRecordc C s< z| j �| jd�}|d | j | j< W dS ty Y dS w )z� Count each log record as a line. :param record: The log record to process :type record: logging.LogRecord r r N)r� r& r� � Exception)r r� �currentr r r �emit� s �zLineCountHandler.emit)r� r# r� rX r r )r� r� r r )r- r. r/ r0 r! r� � __classcell__r r r� r r� � s r� )r r r, ) r6 r7 r8 r9 r: r; r< r= r r )rV )r6 r7 r8 r9 r: r; rW rX rY rZ r[ r# r r )r rZ )ry rZ r r )r� r# r� r$ r r )6r0 � __future__r �logging�osrA r �typingr r r �multiprocessing�pathlib�collections.abcr r Zmultiprocessing.synchronizer Z SyncEventZ relenv.commonr r^ r � getLoggerr- rs rJ rE rK rO rh rj �environr& �platformZ wt_sessionZ term_programrI rF ZSYMBOL_RUNNINGrM rL r r r rG r1 rU rg rx r� r� �Handlerr� r r r r �<module> sl / �3 �
| ver. 1.4 |
Github
|
.
| PHP 8.2.30 | Generation time: 0 |
proxy
|
phpinfo
|
Settings