← Back
Editing: 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 �� �
Save File
Cancel