U
��.eR6 � @ sx d dl mZmZ d dlZd dlZd dlZd dlZd dlZd dlm Z m
Z
mZ d dl m Z d dl
mZmZ d dlmZmZmZ d dlmZ d dlmZ d d lmZ d d
lmZ d d l m Z e r�d d l m Z m Z m!Z! zd d
l m"Z" W n e#k
r� dZ"Y nX e�$e%�Z&dd� Z'e'ee�Z(G dd� de)�Z*G dd� de�Z+G dd� de�Z,G dd� de)�Z-G dd� de)�Z.G dd� de.e*e-�Z/G d d � d e/e(�Z0G d d � d e/e+�Z1G d d!� d!e/e�Z2G d"d#� d#e/e�Z3G d$d%� d%e/e,�Z4G d&d'� d'e.e*e-e�Z5e1e1fe0e5fe2e5fe3e5fe4e5fd(�Z6d7d)d*�Z7ej8d+d,� �Z9G d-d.� d.e)�Z:G d/d0� d0e)�Z;G d1d2� d2e;�ZdS )8� )�absolute_import�divisionN)�SIGINT�default_int_handler�signal)�six)� HIDE_CURSOR� SHOW_CURSOR)�Bar�FillingCirclesBar�IncrementalBar)�Spinner)�WINDOWS)�get_indentation)� format_size)�MYPY_CHECK_RUNNING)�Any�Iterator�IO)�coloramac C s� t | jdd �}|s|S t | dt�� �t | dt�� �g}|tt | dg ��7 }zt�� �|��|� W n tk
rz | Y S X | S d S )N�encodingZ
empty_fillZfill�phases)�getattr�filer Z text_type�list�join�encode�UnicodeEncodeError)Z preferredZfallbackr Z
characters� r �:/usr/lib/python3.8/site-packages/pip/_internal/utils/ui.py�_select_progress_class% s �
r c s4 e Zd ZdZ� fdd�Z� fdd�Zdd� Z� ZS )�InterruptibleMixina�
Helper to ensure that self.finish() gets called on keyboard interrupt.
This allows downloads to be interrupted without leaving temporary state
(like hidden cursors) behind.
This class is similar to the progress library's existing SigIntMixin
helper, but as of version 1.2, that helper has the following problems:
1. It calls sys.exit().
2. It discards the existing SIGINT handler completely.
3. It leaves its own handler in place even after an uninterrupted finish,
which will have unexpected delayed effects if the user triggers an
unrelated keyboard interrupt some time after a progress-displaying
download has already completed, for example.
c s4 t t| �j||� tt| j�| _| jdkr0t| _dS )z=
Save the original SIGINT handler for later.
N)�superr! �__init__r r �
handle_sigint�original_handlerr ��self�args�kwargs�� __class__r r r# U s
zInterruptibleMixin.__init__c s t t| ��� tt| j� dS )z�
Restore the original SIGINT handler after finishing.
This should happen regardless of whether the progress display finishes
normally, or gets interrupted.
N)r" r! �finishr r r% �r' r* r r r, e s zInterruptibleMixin.finishc C s | � � | �||� dS )z�
Call self.finish() before delegating to the original SIGINT handler.
This handler should only be in place while the progress display is
active.
N)r, r% )r' Zsignum�framer r r r$ o s z InterruptibleMixin.handle_sigint)�__name__�
__module__� __qualname__�__doc__r# r, r$ �
__classcell__r r r* r r! C s
r! c @ s e Zd Zdd� ZdS )� SilentBarc C s d S �Nr r- r r r �update| s zSilentBar.updateN)r/ r0 r1 r6 r r r r r4 z s r4 c @ s e Zd ZdZdZdZdZdS )� BlueEmojiBar�
%(percent)d%%� )u 🔹u 🔷u 🔵N)r/ r0 r1 �suffixZ
bar_prefixZ
bar_suffixr r r r r r7 � s r7 c sJ e Zd Z� fdd�Zedd� �Zedd� �Zedd� �Zd d
d �Z� Z S )
�DownloadProgressMixinc s, t t| �j||� dt� d | j | _d S )Nr9 � )r" r; r# r �messager& r* r r r# � s z DownloadProgressMixin.__init__c C s
t | j�S r5 )r �indexr- r r r �
downloaded� s z DownloadProgressMixin.downloadedc C s | j dkrdS td| j �d S )Ng z...� z/s)Zavgr r- r r r �download_speed� s
z$DownloadProgressMixin.download_speedc C s | j rd| j S dS )Nzeta %s� )ZetaZeta_tdr- r r r �
pretty_eta� s
z DownloadProgressMixin.pretty_etar@ c c s& |D ]}|V | � |� q| �� d S r5 )�nextr, )r' �it�n�xr r r �iter� s zDownloadProgressMixin.iter)r@ )
r/ r0 r1 r# �propertyr? rA rC rH r3 r r r* r r; � s
r; c s e Zd Z� fdd�Z� ZS )� WindowsMixinc s\ t r� jrd� _tt� �j||� t rXtrXt�� j�� _� fdd�� j_� fdd�� j_ d S )NFc s � j j�� S r5 )r �wrapped�isattyr r- r r �� � z'WindowsMixin.__init__..c s � j j�� S r5 )r rK �flushr r- r r rM � rN )
r Z hide_cursorr" rJ r# r Z AnsiToWin32r rL rO r& r* r- r r# � s
zWindowsMixin.__init__)r/ r0 r1 r# r3 r r r* r rJ � s rJ c @ s e Zd ZejZdZdZdS )�BaseDownloadProgressBarr8 z0%(downloaded)s %(download_speed)s %(pretty_eta)sN)r/ r0 r1 �sys�stdoutr r= r: r r r r rP � s rP c @ s e Zd ZdS )�DefaultDownloadProgressBarN�r/ r0 r1 r r r r rS � s rS c @ s e Zd ZdS )�DownloadSilentBarNrT r r r r rU � s rU c @ s e Zd ZdS )� DownloadBarNrT r r r r rV � s rV c @ s e Zd ZdS )�DownloadFillingCirclesBarNrT r r r r rW � s rW c @ s e Zd ZdS )� DownloadBlueEmojiProgressBarNrT r r r r rX � s rX c @ s& e Zd ZejZdZdd� Zdd� ZdS )�DownloadProgressSpinnerz!%(downloaded)s %(download_speed)sc C s" t | d�st�| j�| _t| j�S )N�_phaser)�hasattr� itertools�cycler rZ rD r- r r r �
next_phase� s
z"DownloadProgressSpinner.next_phasec C sN | j | }| �� }| j| }d�||r*dnd||r6dnd|g�}| �|� d S )NrB r9 )r= r^ r: r Zwriteln)r' r= Zphaser: �liner r r r6 � s
�z DownloadProgressSpinner.updateN) r/ r0 r1 rQ rR r r: r^ r6 r r r r rY � s rY )ZoffZon�asciiZprettyZemojic C s8 |d ks|dkr t | d � jS t | d |d�jS d S )Nr r@ )�max)� BAR_TYPESrH )Z progress_barra r r r �DownloadProgressProvider s rc c c sP t r d V n@| �� r"t�� tjkr*d V n"| �t� z
d V W 5 | �t� X d S r5 ) r rL �logger�getEffectiveLevel�logging�INFO�writer r )r r r r �
hidden_cursor s
ri c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� RateLimiterc C s || _ d| _d S )Nr )� _min_update_interval_seconds� _last_update)r' �min_update_interval_secondsr r r r# - s zRateLimiter.__init__c C s t � � }|| j }|| jkS r5 )�timerl rk )r' ZnowZdeltar r r �ready2 s
zRateLimiter.readyc C s t � � | _d S r5 )rn rl r- r r r �reset8 s zRateLimiter.resetN)r/ r0 r1 r# ro rp r r r r rj , s rj c @ s e Zd Zdd� Zdd� ZdS )�SpinnerInterfacec C s
t � �d S r5 ��NotImplementedErrorr- r r r �spin> s zSpinnerInterface.spinc C s
t � �d S r5 rr �r' Z final_statusr r r r, B s zSpinnerInterface.finishN)r/ r0 r1 rt r, r r r r rq = s rq c @ s. e Zd Zd dd�Zdd� Zdd � Zd
d � ZdS )
�InteractiveSpinnerN�-\|/� �?c C s\ || _ |d krtj}|| _t|�| _d| _t�|�| _ | j�
dt � | j d � d| _ d S )NFr9 z ... r )
�_messagerQ rR �_filerj �
_rate_limiter� _finishedr\ r] � _spin_cyclerh r �_width)r' r= r Z
spin_charsrm r r r r# H s
zInteractiveSpinner.__init__c C sR d| j }| j�|d| j | � | j�|� t|�| _ | j�� | j�� d S )N�r9 )r~ rz rh �lenrO r{ rp )r' �statusZbackupr r r �_writeW s
zInteractiveSpinner._writec C s, | j r
d S | j�� sd S | �t| j�� d S r5 )r| r{ ro r� rD r} r- r r r rt c s
zInteractiveSpinner.spinc C s4 | j r
d S | �|� | j�d� | j�� d| _ d S )N�
T)r| r� rz rh rO ru r r r r, k s
zInteractiveSpinner.finish)Nrw rx )r/ r0 r1 r# r� rt r, r r r r rv G s �
rv c @ s. e Zd Zd dd�Zdd� Zdd� Zdd � Zd
S ) �NonInteractiveSpinner�< c C s$ || _ d| _t|�| _| �d� d S )NFZstarted)ry r| rj r{ �_update)r' r= rm r r r r# z s
z NonInteractiveSpinner.__init__c C s | j �� t�d| j|� d S )Nz%s: %s)r{ rp rd �infory )r' r� r r r r� � s
z NonInteractiveSpinner._updatec C s& | j r
d S | j�� sd S | �d� d S )Nzstill running...)r| r{ ro r� r- r r r rt � s
zNonInteractiveSpinner.spinc C s$ | j r
d S | �d|f � d| _ d S )Nzfinished with status '%s'T)r| r� ru r r r r, � s z NonInteractiveSpinner.finishN)r� )r/ r0 r1 r# r� rt r, r r r r r� y s
r� c c s� t j�� r"t�� tjkr"t| �}nt| �}z t t j�� |V W 5 Q R X W n> t
k
rj |� d� � Y n* t k
r� |� d� � Y n X |� d� d S )NZcanceled�errorZdone)
rQ rR rL rd re rf rg rv r� ri �KeyboardInterruptr, � Exception)r= Zspinnerr r r � open_spinner� s
r� )N)?Z
__future__r r �
contextlibr\ rf rQ rn r r r Z pip._vendorr Zpip._vendor.progressr r Zpip._vendor.progress.barr
r r Z pip._vendor.progress.spinnerr
Zpip._internal.utils.compatr Zpip._internal.utils.loggingr Zpip._internal.utils.miscr Zpip._internal.utils.typingr �typingr r r r r� Z getLoggerr/ rd r Z_BaseBar�objectr! r4 r7 r; rJ rP rS rU rV rW rX rY rb rc �contextmanagerri rj rq rv r� r� r r r r � s~
7 � � � � � ��
2