File: //usr/lib64/python3.9/site-packages/borg/helpers/__pycache__/progress.cpython-39.opt-1.pyc
a
HZ�h& � @ s� d dl Z d dlZd dlZd dlZd dlmZ ddlmZ e� Zddlm Z dd� Z
G d d
� d
�Z G d d � d e �Z G d
d� de �Z
G dd� d�ZdS )� N)�get_terminal_size� )�
create_logger� )�ellipsis_truncatec C s$ t dd�d }|dkr | �|�S | S )N����r ��fallbackr r )r �ljust)�message�terminal_space� r �;/usr/lib64/python3.9/site-packages/borg/helpers/progress.py�justify_to_terminal_size
s
r c @ sP e Zd ZdZdZdZdZedd� �Zddd�Z d d
� Z
dd �d d
�Z dd� Z dS )�ProgressIndicatorBasezborg.output.progressNFr c C s | j d7 _ | j S )zYUnique number, can be used by receiving applications to distinguish different operations.r )�operation_id_counter)�clsr r r � operation_id s z"ProgressIndicatorBase.operation_idc C s� d | _ t�| j�| _| �� | _|| _| jjs�tj t
j d�| _ | j � tj
� t�d�}z |j}|jrfdnd}|j| _W n ty� d}Y n0 | j �|� || j _| j�| j � | jjtjkr�| j� tj� d| j_| j�� tj
k| _d S )N)�streamZborg�
�
F)�handler�loggingZ getLogger�LOGGER�loggerr �id�msgid�handlersZ
StreamHandler�sys�stderrZsetLevel�INFO� formatter�json�AttributeErrorZ setFormatter�
terminatorZ
addHandler�levelZNOTSETZWARNZ propagateZgetEffectiveLevel�emit)�selfr r r" r% r r r �__init__" s*
z ProgressIndicatorBase.__init__c C s& | j d ur"| j�| j � | j �� d S �N)r r Z
removeHandler�close�r( r r r �__del__I s
z ProgressIndicatorBase.__del__��finishedc K sF | j s
d S |�t| j| j| j|t�� d�� tt� |�t
j dd� d S )N)Z operationr �typer/ �timeT)�file�flush) r' �update�dictr r � JSON_TYPEr1 �printr# �dumpsr r )r( r/ �kwargsr r r � output_jsonN s �z!ProgressIndicatorBase.output_jsonc C s" | j r| jdd� n
| �d� d S )NTr. � )r# r: �outputr, r r r �finish[ s z ProgressIndicatorBase.finish)N)
�__name__�
__module__� __qualname__r r6 r# r � classmethodr r) r- r: r= r r r r r s
'
r c @ s e Zd ZdZdd� ZdS )�ProgressIndicatorMessageZprogress_messagec C s( | j r| j|d� n| j�t|�� d S )N)r )r# r: r �infor )r( �msgr r r r< e s z ProgressIndicatorMessage.outputN)r> r? r@ r6 r< r r r r rB b s rB c s@ e Zd ZdZd� fdd� Zdd d
�Zdd d �Zddd�Z� ZS )�ProgressIndicatorPercentZprogress_percentr � �%3.0f%%Nc s0 d| _ || _|| _|| _|| _t� j|d� dS )a
Percentage-based progress indicator
:param total: total amount of items
:param step: step size in percent
:param start: at which percent value to start
:param msg: output message, must contain one %f placeholder for the percentage
r )r N)�counter�total�
trigger_at�steprD �superr) )r( rI rK �startrD r �� __class__r r r) o s z!ProgressIndicatorPercent.__init__r c C sN |d ur|| _ | j d | j }| j |7 _ || jkrJ| j| j7 _|S d S )N�d )rH rI rJ rK )r( �current�increase�pctr r r �progress� s
z!ProgressIndicatorPercent.progressc C s� | � ||�}|dur�|dur�| js�ddlm} tdd�d }|dkr�||| jt|g|dd� dg � � }t|d |�|d< | j| jt|g| � d |d
�S | �| j| �S dS ) z�
Show and output the progress message
:param current: set the current percentage [None]
:param increase: increase the current percentage [None]
:param info: array of strings to be formatted with msg [None]
Nr )�swidthr r r r r; F)�justifyrC ) rT r# �platformrU r rD �tupler r< )r( rQ rR rC rS rU r
�spacer r r �show� s * z ProgressIndicatorPercent.showTc C s: | j r | j|| j| j|d� n|r*t|�}| j�|� d S )N)r rQ rI rC )r# r: rH rI r r rC )r( r rV rC r r r r< � s
z ProgressIndicatorPercent.output)r rF r rG N)Nr )Nr N)TN) r> r? r@ r6 r) rT rZ r< �
__classcell__r r rN r rE l s
rE c @ s6 e Zd Zd
dd�Zdd� Zdd� Zd d
� Zd d � ZdS )�ProgressIndicatorEndless�
Nc C s* d| _ d| _|| _|du r tj}|| _dS )z�
Progress indicator (long row of dots)
:param step: every Nth call, call the func
:param file: output file, default: sys.stderr
r N)rH � triggeredrK r r r2 )r( rK r2 r r r r) � s z!ProgressIndicatorEndless.__init__c C s4 | j d7 _ | j | j dk}|r0| jd7 _|S )Nr r )rH rK r^ �r( Ztriggerr r r rT � s
z!ProgressIndicatorEndless.progressc C s | � � }|r| �| j�S d S r* )rT r< r^ r_ r r r rZ � s z ProgressIndicatorEndless.showc C s t dd| jdd� d S )N�.r; T)�endr2 r3 �r7 r2 )r( r^ r r r r< � s z ProgressIndicatorEndless.outputc C s t | jd� d S )N)r2 rb r, r r r r= � s z ProgressIndicatorEndless.finish)r] N)r> r? r@ r) rT rZ r< r= r r r r r\ � s
r\ )r r# r r1 �shutilr r r Z parseformatr r r rB rE r\ r r r r �<module> s M
>