0x1949 Team - FAZEMRX - MANAGER
Edit File: MirrorTest.cpython-310.pyc
o U�7e � @ sL d dl Z z d dlmZmZ W n ey d dlmZmZ Y nw d dlZd dlZd dlZd dlZd dl Z d dl Z zd dl m Z W n eyO d dlm Z Y nw d dlZG dd� de j�Zedkr�e 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 dS ) � N)�Empty�Queue)�urlopenc @ sf 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 s0 d }| j �� s�| jj�� r�z5| j �d�}|j}| j�d| � tj ddddd|gtj tj dd �j } |� � }|s9nt�| j|�}q2W n tyL Y d S | j�d | � Y tj�� t jd 7 _| j�tjtj| j| j� |r�| j�t |d �||g� tj� � | j �� s�| jj�� s d S d S d S d S ) NFz Pinging %s...�pingz-qz-c 2z-W 1z-i 0.5T)�stdout�stderr�universal_newlinesz Skipping %sr r ) r �emptyr �running�isSet�get�hostname� report_action� subprocess�Popen�PIPE�STDOUTr �readliner �findallr r r �completed_lock�acquire� completed�report_progress�todor r r �append�float�release)r �result�mirror�host�commando�liner r r �run# sF ��� � &�zMirrorTest.PingWorker.runN)r r )�__name__� __module__� __qualname__�__doc__r r5 r r r r � PingWorker s r: Nc C sl t j�| � d| _d| _|| _d | _|| _g | _dt _ t � � t _ t |�t _|| _|s1t �� | _d S || _d S )N� )r r g r )r r r �action�progress�event�best� test_file�threadsr r* �Lockr( �lenr, �mirrors�Eventr )r rD r@ r> r r r r r C s zMirrorTest.__init__c C s || _ | jr | j�� d S d S �N)r< r> �set)r �textr r r r! T s �zMirrorTest.report_action�r �d r c C sP ||d ||d |d |d |d | | f| _ | jr&| j�� dS dS )zTSubclasses should override this method to receive progress status updatesr r N)r= r> rG )r �current�maxr r r r r r+ Y s ��zMirrorTest.report_progressc C s� | j dddd�}t| j�}|dkr3|�dd| jt�d|d � g� |�dd| jt�d|d � g� | jdd � |D �d tjtjfd �}|D ] \}}t d |j |f � qF|sXd S |d d j }t d | � |S )N� )r � �?)r � )rL r r � r r c S s g | ]}|d �qS )rP r )�.0�rr r r � <listcomp>j s z,MirrorTest.run_full_test.<locals>.<listcomp>�rN r )r r zmirror: %s - time: %szand the winner is: %s) � run_ping_testrC rD r- �random�randint�run_download_testr r, �printr )r � results_ping�sizer �t�h�winnerr r r � run_full_testb s$ ""�� zMirrorTest.run_full_testr c C s� |dkr| j }t� }|D ]}|�|� q g }td�D ]}t�|||| ||�} | j�| � | �� q| jD ]} | � � q4|� � |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 ) rD r �put�ranger r: rA r- �start�join�sort) r rD rL r r r �mr �ir\ r r r rU w s zMirrorTest.run_ping_testc s� � fdd�}|dkr � j }g }|D ]'}� j�� s n ||�}|dkr)|�||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 sZ d| � � d � jf }� �d| � t�� }zt|dd��d� t�� | W S Y dS )Nz%s/%sr zDownloading %s...rP )�timeouti � )� get_repo_urlsr@ r! �timer �read)r1 �urlrc �r r r �test_download_speed� s �z9MirrorTest.run_download_test.<locals>.test_download_speedNr r rT )rD r r r- r+ �indexrC re ) r rD rL r r rn r rf � download_timer rm r rX � s z MirrorTest.run_download_testc C s | � � | _| j�� dS )z/Complete test exercise, set self.best when doneN)r_ r? r �clearrm r r r r5 � s zMirrorTest.runrF )rI r )NNr r )r6 r7 r8 r9 r r r: r r! r+ r_ rU rX r5 r r r r r s - � r �__main__zdpkg --print-architecturez!dists/%s/%s/binary-%s/Packages.gz))r �queuer r � ImportErrorrj r r" �os�aptsources.distro� aptsources�aptsources.sourceslist�urllib.requestr �urllib2rV r r r6 �distro� get_distro� get_sources� sourceslist� SourcesList�popen�piperk �strip�archrE r rG �source_template�name� componentsr@ �list� mirror_set�values�appr_ r r r r �<module> sL � � �� � �