CasperSecurity
o
4��f�( � @ s� d Z ddlmZ ddlZddlZddlZddlZddlZddlm Z m
Z
m Z ddl Z ddl
mZ g d�Zdd� ZG d d
� d
e�ZG d d � d eje�ZG d
d� deje�ZG dd� deje�ZdS )z'Progress reporting for text interfaces.� )�print_functionN)�Callable�Optional�Union)�base)�AcquireProgress�
CdromProgress�
OpProgressc C s" t �| �}|| krt �| d�}|S )z>Translate the message, also try apt if translation is missing.�apt)�apt_pkg�gettext)�msg�res� r �3/usr/lib/python3/dist-packages/apt/progress/text.py�_$ s
r c @ s$ e Zd ZdZd dd�Zd
dd�ZdS ) � TextProgressz.Internal Base class for text progress classes.Nc C s |pt j| _d| _d S )Nr )�sys�stdout�_file�_width��self�outfiler r r �__init__0 s
zTextProgress.__init__TFc C sz | j �d� | j �|� | jt|�kr!| j �| jt|� d � n |r,t| jt|��| _|r6| j �d� dS | j �� dS )z8Write the message on the terminal, fill remaining space.�
� �
N)r �writer �len�max�flush)r r
�newline�maximizer r r �_write5 s zTextProgress._write�N)TF)�__name__�
__module__� __qualname__�__doc__r r$ r r r r r - s
r c @ s, e Zd ZdZd dd�Zd dd�Zdd� ZdS )
r z\Operation progress reporting.
This closely resembles OpTextProgress in libapt-pkg.
Nc C s" t �| |� tj�| � d| _d S )N� )r r r r �old_opr r r r r M s
zOpProgress.__init__c C sL t j�| |� | jr| jr| �| j� | �d| j| jf dd� | j| _dS )z1Called periodically to update the user interface.z %s... %i%%
FTN)r r �update� major_changer+ r$ �op�percent)r r/ r r r r, S s
zOpProgress.updatec C s8 t j�| � | jr| �td�d| jf dd� d| _dS )z,Called once an operation has been completed.z %c%s... Doner Tr* N)r r �doner+ r$ r �r r r r r0 \ s
zOpProgress.doner% )r& r' r( r) r r, r0 r r r r r G s
r c @ sZ e Zd ZdZddd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dS )r z'AcquireProgress for the text interface.Nc C s. t �| |� tj�| � d | _d| _d| _d S )N�P � )r r r r �_signalr �_idr r r r r h s
zAcquireProgress.__init__c C s0 t j�| � t�tj| j�| _| �� d| _dS )z�Start an Acquire progress.
In this case, the function sets up a signal handler for SIGWINCH, i.e.
window resize signals. And it also sets id to 1.
r3 N)r r �start�signal�SIGWINCH�_winchr4 r5 r1 r r r r6 p s
zAcquireProgress.startc G sp t | jd�r4t�| j�� �r6ddl}ddl}ddl}|�| j|j d�}|�
d|�\}}}}|d | _ dS dS dS )z)Signal handler for window resize signals.�filenor Ns �hhhhr3 ) �hasattrr �os�isattyr: �fcntl�termios�struct�ioctl�
TIOCGWINSZ�unpackr )r �dummyr? r@ rA �buf�colr r r r9 } s �zAcquireProgress._winchc C sH t j�| |� td�|j }|jjr |dt�|jj� 7 }| � |� dS )z@Called when an item is update (e.g. not modified on the server).zHit � [%sB]N)
r r �ims_hitr � description�owner�filesizer � size_to_strr$ �r �item�liner r r rI � s
zAcquireProgress.ims_hitc C s` t j�| |� |jj|jjkr| �td�|j � dS | �td�|j � | �d|jj � dS )z Called when an item is failed.zIgn zErr z %sN)
r r �failrK �status� STAT_DONEr$ r rJ �
error_text)r rO r r r rQ � s
zAcquireProgress.failc C sx t j�| |� |jjr
dS | j|j_| jd7 _td�d|jj|jf }|jj r5|dt
� |jj � 7 }| � |� dS )z/Called when some of the item's data is fetched.Nr3 zGet:z%s %srH )
r r �fetchrK �completer5 �idr rJ rL r rM r$ rN r r r rU � s
zAcquireProgress.fetchc C s t j�| |� t| jd�rt�| j�� �sdS | j| j d t
| j | j � }d}d| }d}| j
rLtt
| j | j �| j
�}dt�| j
�t�|�f }|jD ]�}d}|jsv|jrud|j }t|�t|� t|� | jkro np||7 }d}qOd}|jjjr�|d |jjj|jjf 7 }n|d
|jj 7 }|jjjr�|d |jjj 7 }|d t�|j � 7 }|j r�|jjj s�|d
t�|j �|j d |j f 7 }|d7 }t|�t|� t|� | jkr� n||7 }qO|s�|t d�7 }| j
r�|| jt|� t|� d | 7 }| � |d� dS )z�Periodically invoked while the Acquire process is underway.
Return False if the user asked to cancel the whole Acquire process.r: Tg Y@Fz%i%%r* z %sB/s %sz [%s]z [%i %sz [%sz %sz %sBz /%sB %i%%�]z
[Working]r )!r r �pulser< r r= r> r: �
current_bytes�
current_items�float� total_bytes� total_items� current_cps�intr rM � time_to_str�workers� current_itemrR r r rK rW � shortdescrJ �active_subprocess� current_size�
total_sizerV r r$ ) r rK r/ �shown�tval�end�eta�worker�valr r r rY � sh ��� �
�
�
" zAcquireProgress.pulsec C s0 t j�| ||� | �td�||f � t� dvS )z7Prompt the user to change the inserted removable media.zUMedia change: please insert the disc labeled
'%s'
in the drive '%s' and press enter
)�c�C)r r � media_changer$ r �input)r �medium�driver r r rp � s
�
z AcquireProgress.media_changec C s^ t j�| � | �td�t�| j�t�| j �t�| j
�f � d�� ddl }|� |j
| j� dS )z/Invoked when the Acquire process stops running.zFetched %sB in %s (%sB/s)
r r N)r r �stopr$ r r rM �
fetched_bytesra � elapsed_timer_ �rstripr7 r8 r4 )r r7 r r r rt � s
��zAcquireProgress.stopr% )
r& r' r( r) r r6 r9 rI rQ rU rY rp rt r r r r r e s
E r c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) r zText CD-ROM progress.c C s@ t j�| � | �td�d� zttd��W S ty Y dS w )z,Ask the user to provide a name for the disc.zDPlease provide a name for this medium, such as 'Debian 2.1r1 Disk 1'F�:N)r r �ask_cdrom_namer$ r �strrq �KeyboardInterruptr1 r r r ry
s
� �z CdromProgress.ask_cdrom_namec C s( t j�| ||� |r| �|d� dS dS )zSet the current progress.FN)r r r, r$ )r �text�currentr r r r, s �zCdromProgress.updatec C sB t j�| � | �td�d� ztt� dk�W S ty Y dS w )z"Ask the user to change the CD-ROM.z4Please insert an installation medium and press enterFr* )r r � change_cdromr$ r �boolrq r{ r1 r r r r~ s
� �zCdromProgress.change_cdromN)r& r' r( r) ry r, r~ r r r r r s
r )r) �
__future__r �ior= r7 r �types�typingr r r r � apt.progressr �__all__r �objectr r r r r r r r �<module> s" #