File: //lib/python3/dist-packages/softwareproperties/__pycache__/MirrorTest.cpython-38.pyc
U
��d � @ sN d dl Z zd dlmZmZ W n$ ek
r@ d dlmZmZ Y nX d dlZd dlZd dlZd dlZd dl Z
d dl Z
zd dl m
Z
W n ek
r� d dlm
Z
Y nX d dlZG dd� de j�Zedk�rJe
j�� Ze�e
j�� � e�d�Ze�� �� Z e � � Z e � � dej j!ej j"d j!e f Z#ee$ej j%�&� �e#e � � e �Z'e'�(� dS ) � N)�Empty�Queue)�urlopenc @ sb e Zd ZdZG dd� dej�Zddd�Zdd� Zdd d �Z d
d� Z
ddd�Z ddd�Z dd� Z
dS )�
MirrorTestz@Determines the best mirrors by perfoming ping and download test.c @ s" e Zd ZdZd dd�Zdd� ZdS )
zMirrorTest.PingWorkerz�Use the command line command ping to determine the server's
response time. Using multiple threads allows to run several
test simultaneously.�r � �r r c C s@ || _ || _|| _|| _|| _|| _t�d�| _t j
� | � d S )Nz ^rtt .* = [\.\d]+/([\.\d]+)/.*) �borders�mod�parent�id�jobs�results�re�compile� match_result� threading�Thread�__init__)�selfr
r r r r r
� r �?/usr/lib/python3/dist-packages/softwareproperties/MirrorTest.pyr s z MirrorTest.PingWorker.__init__c C s d }| j �� �s| jj�� �rzh| j �d�}|j}| j�d| � tj ddddd|gtj
tj dd �j }|�
� }|stq�t�| j|�}qfW n0 tk
r� Y d S | j�d
| � Y nX tj�� t jd 7 _| j�tjtj| j| j� |�r
| j�t |d �||g� tj� � qd S )
NFz
Pinging %s...Zpingz-qz-c 2z-W 1z-i 0.5T)�stdout�stderrZuniversal_newlinesz Skipping %sr r ) r
�emptyr �running�isSet�get�hostname�
report_action�
subprocess�Popen�PIPEZSTDOUTr �readliner �findallr r r �completed_lock�acquire� completed�report_progress�todor r
r �append�float�release)r �result�mirrorZhostZcommando�liner r r �run# s: �
�zMirrorTest.PingWorker.runN)r r )�__name__�
__module__� __qualname__�__doc__r r0 r r r r �
PingWorker s
r5 Nc C sj t j�| � d| _d| _|| _d | _|| _g | _dt _
t � � t _ t
|�t _|| _|s`t �� | _n|| _d S )N� )r r g r )r r r �action�progress�event�best� test_file�threadsr r' ZLockr% �lenr) �mirrors�Eventr )r r> r; r9 r r r r r C s
zMirrorTest.__init__c C s || _ | jr| j�� d S )N)r7 r9 �set)r �textr r r r T s zMirrorTest.report_action�r �d r c C sL ||d ||d |d |d |d | | f| _ | jrH| j�� dS )zTSubclasses should override this method to receive
progress status updatesr r N)r8 r9 r@ )r Zcurrent�maxr r
r r r r( Y s
�zMirrorTest.report_progressc
C s� | j dddd�}t| j�}|dkrf|�dd| jt�d|d � g� |�dd| jt�d|d � g� | jdd � |D �d
tjtjfd �}|D ]\}}t d |j
|f � q�|s�d S |d d j
}t d
| � |S d S )N� )r � �?)r � )rD r r
� r r c S s g | ] }|d �qS )rH r )�.0�rr r r �
<listcomp>j s z,MirrorTest.run_full_test.<locals>.<listcomp>�rF r )r r
zmirror: %s - time: %szand the winner is: %s) �
run_ping_testr= r> r* �randomZrandint�run_download_testr r) �printr )r Z results_ping�sizer �t�h�winnerr r r �
run_full_testb s$
""�� zMirrorTest.run_full_testr c
C s� |dkr| j }t� }|D ]}|�|� qg }td�D ],}t�|||| ||�} | j�| � | �� q4| jD ] } | � � qh|�
� |d|� S )z�Performs ping tests of the given mirrors and returns the
best results (specified by max).
Mod and borders could be used to tweak the reported result if
the download test is only a part of a whole series of tests.N� r ) r> r Zput�ranger r5 r< r* �start�join�sort)
r r> rD r r
r
�mr �irR r r r rM w s
zMirrorTest.run_ping_testc s� � fdd�}|dkr� j }g }|D ]N}� j�� s4 qr||�}|dkrR|�||g� � �|�|�d t|�d|� q"|�� |d|� S )z�Performs download tests of the given mirrors and returns the
best results (specified by max).
Mod and borders could be used to tweak the reported result if
the download test is only a part of a whole series of tests.c s` d| � � d � jf }� �d| � t�� }z t|dd��d� t�� | W S Y dS X d S )Nz%s/%sr zDownloading %s...rH )Ztimeouti � )Z
get_repo_urlsr; r �timer �read)r. ZurlrX �r r r �test_download_speed� s �z9MirrorTest.run_download_test.<locals>.test_download_speedNr r rL )r> r r r* r( �indexr= rZ ) r r> rD r r
r` r r[ Z
download_timer r_ r rO � s
z MirrorTest.run_download_testc C s | � � | _| j�� dS )z/Complete test exercise, set self.best when doneN)rU r: r �clearr_ r r r r0 � s
zMirrorTest.run)N)rB r )NNr r )NNr r )r1 r2 r3 r4 r r r5 r r r( rU rM rO r0 r r r r r s -
�
r �__main__zdpkg --print-architecturez!dists/%s/%s/binary-%s/Packages.gz))r Zqueuer r � ImportErrorr] r r �osZaptsources.distroZ
aptsourcesZaptsources.sourceslistZurllib.requestr Zurllib2rN r r r1 ZdistroZ
get_distroZ get_sourcesZ sourceslistZ SourcesList�popen�piper^ �stripZarchr? r r@ Zsource_template�nameZ
componentsr; �listZ
mirror_set�valuesZapprU r r r r �<module> sH
�� �