File: //lib/python3.9/site-packages/__pycache__/speedtest.cpython-39.pyc
a
�;8dM� � @ sH d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl Z
zd dlZejZW n ey� dZeZY n0 dZG dd� de�Zdae� Zejdd� dkZejdd� dkZejdd� d kZz d dlZW n: e�y0 z d dl ZW n e�y* dZY n0 Y n0 z<d dl m m Z zd d
l m!Z" W n e�yj Y n0 W n0 e�y� d d l#m$Z% d d l m&Z& dZ Y n0 z4d d
l'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1 W nD e�y d d
l2m(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1 Y n0 zd dl3m4Z4m5Z5 W n$ e�yR d dl6m4Z4m5Z5 Y n0 zd dl3m7Z7 W n> e�y� zd dl6m7Z7 W n e�y� dZ7Y n0 Y n0 zd dl3m8Z8 W n e�y� dZ8Y n0 zd dl9m9Z9 W n e�y� d dl:m9Z9 Y n0 zd dl;m;Z; W n e�y0 d dl<m;Z; Y n0 zd dl;m=Z= W nF e�y� zd dl<m=Z= W n e�y� d dl>m=Z= Y n0 Y n0 zd dl?m@Z@ W n e�y� d dl@m@Z@ Y n0 z(d dlAmBZC d dlAmDZE eFZGeHZIeJZKW n8 e�y d dlLmMZC d dlLmNZE dZGdZIdZKY n0 zd d lOmPZP dZQW nN e�y� zd d lPmPZP dZQW n$ e�yz d d lRmPZPmQZQ Y n0 Y n0 z d dlSZSW n� e�y d dlTZTd d lRmUZUmVZV G d d � d eU�ZWeXeTd!�ZYzeWejZ�Z[eWej\�Z]W n e^�y ejZZ[ej\Z]Y n0 d"d#� Z_d$d%� Z`Y n0 [Sd&d#� Z_d'd%� Z`e�rBe jajbZcn e�rNe"jdZce�r^e jejfZgne jejhZgzJd dliZiz eijjfZkW n el�y� em� ZkY n0 e*e+ejneijoe5fek ZpW n& e�y� dZie*e+ejne5fZpY n0 G d(d)� d)eq�ZrG d*d+� d+er�ZsG d,d-� d-er�ZtG d.d/� d/er�ZuG d0d1� d1er�ZvG d2d3� d3et�ZwG d4d5� d5et�ZxG d6d7� d7er�ZyG d8d9� d9er�ZzG d:d;� d;er�Z{G d<d=� d=er�Z|G d>d?� d?er�Z}G d@dA� dAer�Z~G dBdC� dCer�ZG dDdE� dEer�Z�G dFdG� dGer�Z�edfdHdI�Z�G dJdK� dKe4�Z�e7�r G dLdM� dMe7�Z�d�dNdO�Z�G dPdQ� dQe,�Z�G dRdS� dSe,�Z�d�dUdV�Z�G dWdX� dXe�Z�dYdZ� Z�d[d\� Z�d]d^� Z�d�d`da�Z�d�dbdc�Z�ddde� Z�dfdg� Z�dhdi� Z�djdk� Z�G dldm� dme je�Z�G dndo� doe�Z�G dpdq� dqe je�Z�G drds� dse�Z�G dtdu� due�Z�dvdw� Z�dxdy� Z�d�d{d|�Z�d}d~� Z�dd�� Z�d�d�d��Z�d�d�� Z�d�d�� Z�e�d�k�rDe�� dS )�� Nz2.1.3c @ s e Zd ZdZedd� �ZdS )�FakeShutdownEventz�Class to fake a threading.Event.isSet so that users of this module
are not required to register their own threading.Event()
c C s dS )z#Dummy method to always return falseF� r r r �-/usr/lib/python3.9/site-packages/speedtest.py�isSet. s zFakeShutdownEvent.isSetN)�__name__�
__module__� __qualname__�__doc__� staticmethodr r r r r r ) s r F� )r � )r � )� r )�_Element)�minidom)�
ExpatError)
�urlopen�Request� HTTPError�URLError�AbstractHTTPHandler� ProxyHandler�HTTPDefaultErrorHandler�HTTPRedirectHandler�HTTPErrorProcessor�OpenerDirector)�HTTPConnection�
BadStatusLine)�HTTPSConnection)�
FakeSocket)�Queue)�urlparse)�parse_qs)�md5)�ArgumentParser)�SUPPRESS)� OptionParser)�
SUPPRESS_HELP�int�string�float)�StringIO)r+ �BytesIO)�
TextIOWrapper�FileIOc s, e Zd ZdZ� fdd�Z� fdd�Z� ZS )�_Py3Utf8OutputzVUTF-8 encoded wrapper around stdout for py3, to override
ASCII stdout
c s( t |�� d�}tt| �j|ddd� d S )N�w�utf8�strict)�encoding�errors)r. �fileno�superr/ �__init__)�self�f�kwargs�buf�� __class__r r r7 � s
�z_Py3Utf8Output.__init__c s t t| ��|� | �� d S �N)r6 r/ �write�flush)r8 �sr< r r r? � s z_Py3Utf8Output.write)r r r r r7 r? �
__classcell__r r r< r r/ � s r/ �printc C s | S )z No-op encode to utf-8 for py3r ��vr r r �to_utf8� s rF c O s< |� d�tjkrt|d< n|� dt�|d< t| i |�� dS )z>Wrapper function for py3 to print, with a utf-8 encoded stdout�fileN)�get�sys�stderr�_py3_utf8_stderr�_py3_utf8_stdout�
_py3_print��argsr: r r r �print_� s
rP c C s* z| � dd�W S ty$ | Y S 0 dS )z)Encode value to utf-8 if possible for py2r1 r2 N)�encode�AttributeErrorrD r r r rF � s c
s. |� dtj�� � du rdS � fdd�}d}|� dd�}|dur`t|t�rNd}nt|t�s`td��|� d d�}|dur�t|t�r�d}nt|t�s�td
��|r�td ��|s�| D ]}t|t�r�d} q�q�|r�td �}td
�}nd }d
}|du r�|}|du r�|}t| �D ] \} }| �r||� ||� �q ||� dS )z�The new-style print function for Python 2.4 and 2.5.
Taken from https://pypi.python.org/pypi/six/
Modified to set encoding to UTF-8 always, and to flush after write
rG Nc sl t | t�st| �} d}t � t�rVt | t�rV|d urVt� dd �}|d u rJd}| �||�} � �| � � �� d S )Nr1 r4 r2 ) �
isinstanceZ
basestring�strrG �unicode�getattrrQ r? r@ )�datar3 r4 ��fpr r r? � s
��
zprint_.<locals>.writeF�sepTz sep must be None or a string�endz end must be None or a stringz$invalid keyword arguments to print()�
� )�poprI �stdoutrS rU rT � TypeError� enumerate)
rO r: r? Z want_unicoderZ r[ �arg�newline�space�ir rX r rP � sL
c @ s e Zd ZdZdS )�SpeedtestExceptionz Base exception for this moduleN�r r r r r r r r rf ( s rf c @ s e Zd ZdZdS )�SpeedtestCLIErrorz9Generic exception for raising errors during CLI operationNrg r r r r rh , s rh c @ s e Zd ZdZdS )�SpeedtestHTTPErrorz#Base HTTP exception for this moduleNrg r r r r ri 0 s ri c @ s e Zd ZdZdS )�SpeedtestConfigErrorz Configuration XML is invalidNrg r r r r rj 4 s rj c @ s e Zd ZdZdS )�SpeedtestServersErrorzServers XML is invalidNrg r r r r rk 8 s rk c @ s e Zd ZdZdS )�ConfigRetrievalErrorz Could not retrieve config.phpNrg r r r r rl < s rl c @ s e Zd ZdZdS )�ServersRetrievalErrorz(Could not retrieve speedtest-servers.phpNrg r r r r rm @ s rm c @ s e Zd ZdZdS )�InvalidServerIDTypez/Server ID used for filtering was not an integerNrg r r r r rn D s rn c @ s e Zd ZdZdS )�NoMatchedServersz!No servers matched when filteringNrg r r r r ro H s ro c @ s e Zd ZdZdS )�SpeedtestMiniConnectFailurez7Could not connect to the provided speedtest mini serverNrg r r r r rp L s rp c @ s e Zd ZdZdS )�InvalidSpeedtestMiniServerzjServer provided as a speedtest mini server does not actually appear
to be a speedtest mini server
Nrg r r r r rq P s rq c @ s e Zd ZdZdS )�ShareResultsConnectFailurez6Could not connect to speedtest.net API to POST resultsNrg r r r r rr V s rr c @ s e Zd ZdZdS )�ShareResultsSubmitFailurezRUnable to successfully POST results to speedtest.net API after
connection
Nrg r r r r rs Z s rs c @ s e Zd ZdZdS )�SpeedtestUploadTimeoutz~testlength configuration reached during upload
Used to ensure the upload halts when no additional data should be sent
Nrg r r r r rt ` s rt c @ s e Zd ZdZdS )�SpeedtestBestServerFailurez Unable to determine best serverNrg r r r r ru f s ru c @ s e Zd ZdZdS )�SpeedtestMissingBestServerz?get_best_server not called or not able to determine best serverNrg r r r r rv j s rv c
C s� | \}}d}t �||dt j�D ]�}|\}}} }
} d} zFt � ||| �} |turZ| �t|�� |rh| �|� | �| � | W S t jy� t � }| dur�| �
� Y q 0 q |dur�|�n
t �d��dS )a� Connect to *address* and return the socket object.
Convenience function. Connect to *address* (a 2-tuple ``(host,
port)``) and return the socket object. Passing the optional
*timeout* parameter will set the timeout on the socket instance
before attempting to connect. If no *timeout* is supplied, the
global default timeout setting returned by :func:`getdefaulttimeout`
is used. If *source_address* is set it must be a tuple of (host, port)
for the socket to bind as a source address before making the connection.
An host of '' or port 0 tells the OS to use the default.
Largely vendored from Python 2.7, modified to work with Python 2.4
Nr z!getaddrinfo returns an empty list) �socketZ getaddrinfoZ SOCK_STREAM�_GLOBAL_DEFAULT_TIMEOUTZ
settimeoutr* Zbind�connect�error�
get_exception�close)
Zaddress�timeout�source_address�host�port�err�resZafZsocktype�protoZ canonnameZsa�sockr r r �create_connectionn s(
r� c @ s e Zd ZdZdd� Zdd� ZdS )�SpeedtestHTTPConnectionzUCustom HTTPConnection to support source_address across
Python 2.4 - Python 3
c O sF |� dd �}|� dd�}d | _tj| g|�R i |�� || _|| _d S �Nr~ r} �
)r^ � _tunnel_hostr r7 r~ r} �r8 rO r: r~ r} r r r r7 � s z SpeedtestHTTPConnection.__init__c C sd z t �| j| jf| j| j�| _W n0 ttfyP t| j| jf| j| j�| _Y n0 | j r`| �
� dS )z3Connect to the host and port specified in __init__.N) rw r� r r� r} r~ r� rR r` r� �_tunnel�r8 r r r ry � s
�
� z SpeedtestHTTPConnection.connectN)r r r r r7 ry r r r r r� � s r� c @ s$ e Zd ZdZdZdd� Zdd� ZdS )�SpeedtestHTTPSConnectionz^Custom HTTPSConnection to support source_address across
Python 2.4 - Python 3
i� c O sF |� dd �}|� dd�}d | _tj| g|�R i |�� || _|| _d S r� )r^ r� r r7 r} r~ r� r r r r7 � s z!SpeedtestHTTPSConnection.__init__c C s< z t �| j| jf| j| j�| _W n0 ttfyP t| j| jf| j| j�| _Y n0 | j r`| �
� t r�zFi }t t d�r�| j r�| j |d< n
| j|d< | j
j| jfi |��| _W nB ty� t �| j�| _z| j| j_W n ty� Y n0 Y n0 nHt�r0zt| jt � | j��| _W n t�y, td��Y n0 ntd��dS )z(Connect to a host on a given (SSL) port.Z
SSLContext�server_hostnamez?This version of Python does not support HTTPS/SSL functionalityN)rw r� r r� r} r~ r� rR r` r� r� �ssl�hasattr�_contextZ wrap_socketr� r rf )r8 r: r r r ry � sL
�
�
� �z SpeedtestHTTPSConnection.connectN)r r r r Z default_portr7 ry r r r r r� � s r� c s � ���fdd�}|S )z�Cross Python 2.4 - Python 3 callable to build an ``HTTPConnection`` or
``HTTPSConnection`` with the args we need
Called from ``http(s)_open`` methods of ``SpeedtestHTTPHandler`` or
``SpeedtestHTTPSHandler``
c s, |� ��d�� �r �|d< � | fi |��S )N�r~ r} �context)�update)r r: ��
connectionr� r~ r} r r �inner s �z _build_connection.<locals>.innerr )r� r~ r} r� r� r r� r �_build_connection� s r� c @ s( e Zd ZdZd dd�Zdd� ZejZdS )
�SpeedtestHTTPHandlerz�Custom ``HTTPHandler`` that can build a ``HTTPConnection`` with the
args we need for ``source_address`` and ``timeout``
r Nr� c C s t �| |� || _|| _d S r> )r r7 r~ r} )r8 �
debuglevelr~ r} r r r r7 s z SpeedtestHTTPHandler.__init__c C s | � tt| j| j�|�S r> )�do_openr� r� r~ r} �r8 �reqr r r � http_open s ��z SpeedtestHTTPHandler.http_open)r Nr� ) r r r r r7 r� r � do_request_Z http_requestr r r r r� s
r� c @ s( e Zd ZdZd dd�Zdd� ZejZdS )
�SpeedtestHTTPSHandlerz�Custom ``HTTPSHandler`` that can build a ``HTTPSConnection`` with the
args we need for ``source_address`` and ``timeout``
r Nr� c C s" t �| |� || _|| _|| _d S r> )r r7 r� r~ r} )r8 r� r� r~ r} r r r r7 ( s z SpeedtestHTTPSHandler.__init__c C s | � tt| j| j| jd�|�S )N)r� )r� r� r� r~ r} r� r� r r r �
https_open/ s ��z SpeedtestHTTPSHandler.https_open)r NNr� ) r r r r r7 r� r r� Z
https_requestr r r r r� $ s �
r� r� c C s� t d| dd� | r0| df}t d|f dd� nd}t� t||d�t||d�t� t� t� g}t� }dt� fg|_ |D ]}|�
|� qt|S ) z�Function similar to ``urllib2.build_opener`` that will build
an ``OpenerDirector`` with the explicit handlers we want,
``source_address`` for binding, ``timeout`` and our custom
`User-Agent`
zTimeout set to %dT��debugr z Binding to source address: %rNr� z
User-agent) �printerr r� r� r r r r �build_user_agentZ
addheadersZ add_handler)r~ r} �source_address_tuple�handlers�openerZhandlerr r r � build_opener= s.
���� r� c @ s e Zd ZdZdd� Zdd� ZdS )�GzipDecodedResponsez�A file-like object to decode a response encoded with the gzip
method, as described in RFC 1952.
Largely copied from ``xmlrpclib``/``xmlrpc.client`` and modified
to work for py2.4-py3
c C sf t s td��tpt}|� | _|�d�}t|�dkr4qB| j�|� q | j�d� t j j
| d| jd� d S )NzEHTTP response body is gzip encoded, but gzip support is not available� r �rb)�mode�fileobj) �gzipri r, r+ �io�read�lenr? �seek�GzipFiler7 )r8 �response�IO�chunkr r r r7 h s
z GzipDecodedResponse.__init__c C s, zt j�| � W | j�� n | j�� 0 d S r> )r� r� r| r� r� r r r r| x s zGzipDecodedResponse.closeN)r r r r r7 r| r r r r r� a s r� c C s t �� d S )zgHelper function to work with py2.4-py3 for getting the current
exception in a try/except block
� )rI �exc_infor r r r r{ s r{ c C s� | \}}|\}}d}t �|| �}t �|| �}t �|d �t �|d � t �t �|��t �t �|�� t �|d � t �|d � } dt �t �| �t �d| �� }
||
} | S )z4Determine distance between 2 sets of [lat,lon] in kmi� r r� )�mathZradiansZsinZcosZatan2Zsqrt) �originZ destinationZlat1Zlon1Zlat2Zlon2ZradiusZdlatZdlon�a�c�dr r r �distance� s" � � �� r� c C sL ddt � � t �� d f dt �� ddt f} d�| �}td| d d
� |S ) z0Build a Mozilla/5.0 compatible User-Agent stringz Mozilla/5.0z(%s; U; %s; en-us)r z Python/%sz(KHTML, like Gecko)zspeedtest-cli/%sr] zUser-Agent: %sTr� )�platformZ architectureZpython_version� __version__�joinr� )Zua_tuple�
user_agentr r r r� � s
�
�
r� �0c C s� |si }| d dkr.dt |� }d|| f }n| }d| v r@d}nd}d||ttj�� d �|f }|�d d
i� td d t |� |f d
d� t|||d�S )znBuild a urllib2 request object
This function automatically adds a User-Agent header to all requests
r �:)�http�httpsz%s%s�?�&z %s%sx=%s.%s�� z
Cache-Controlzno-cache�%s %s)�GETZPOSTTr� )rW �headers)�boolr( �timeit�timer� r� r ) �urlrW r� �bump�secure�schemeZ schemed_url�delimZ final_urlr r r �
build_request� s( ���r� c C sl |r |j }nt}z6|| �}| �� |�� kr>td|�� dd� |dfW S tyf t� }d|f Y S 0 dS )zyHelper function to catch common exceptions encountered when
establishing a connection with a HTTP/HTTPS request
zRedirected to %sTr� FN)�openr � get_full_url�geturlr� � HTTP_ERRORSr{ )�requestr� Z_open�uh�er r r �
catch_request� s
r� c C s> z | j j}W n ty$ | j}Y n0 |d�dkr:t| �S | S )zzHelper function to return either a Gzip reader if
``Content-Encoding`` is ``gzip`` otherwise the response itself
zcontent-encodingr� )r� � getheaderrR r� )r� r� r r r �get_response_stream� s r� c C s | � |�d }tt|j�� ��S )z�Retrieve an attribute from an XML document and return it in a
consistent format
Only used with xml.dom.minidom, which is likely only to be used
with python versions older than 2.5
r )�getElementsByTagName�dict�list�
attributes�items)ZdomZtag_name�elemr r r �get_attributes_by_tag_name� s r� c s d� fdd� }|S )zNBuilt in callback function used by Thread classes for printing
status
Fc sF � � � r d S tj�d� | d |kr8|du r8tj�d� tj�� d S )N�.r� Tr\ )r rI r_ r? r@ )�current�total�startr[ ��shutdown_eventr r r� s zprint_dots.<locals>.inner)FFr �r� r� r r� r �
print_dots� s r� c O s d S r> r rN r r r �
do_nothing
s r� c @ s" e Zd ZdZddd�Zdd� ZdS )�HTTPDownloaderz!Thread class for retrieving a URLNc C sX t j�| � || _dg| _|| _|| _|| _|r:|j| _ nt
| _ |rL|| _ nt � | _ d S �Nr )
� threading�Threadr7 r� �result� starttimer} re r� �_openerr �_shutdown_eventr )r8 re r� r� r} r� r� r r r r7 s
zHTTPDownloader.__init__c C s� zrt �� | j | jkrp| �| j�}| j�� sht �� | j | jkrh| j� t
|� d��� | jd dkr"qhq"|� � W n" t
y� Y n ty� Y n0 d S )N� ( ���r )r� �
default_timerr� r} r� r� r� r r� �appendr� r� r| �IOErrorr� )r8 r9 r r r �run& s
�� zHTTPDownloader.run)NN�r r r r r7 r� r r r r r� s
�
r� c @ s@ e Zd ZdZddd�Zdd� Zedd� �Zdd
d �Zd d
� Z dS )�HTTPUploaderDataz]File like object to improve cutting off the upload once the timeout
has been reached
Nc C s8 || _ || _|| _|r || _nt� | _d | _dg| _d S r� )�lengthr� r} r� r �_datar� )r8 r r� r} r� r r r r7 < s zHTTPUploaderData.__init__c C sn d}t tt | j�d ��}tp t}z,|d|| dt | j�d � �� �| _W n tyh td��Y n0 d S )NZ$0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZg B@z content1=%sr � zMInsufficient memory to pre-allocate upload data. Please use --no-pre-allocate) r( �roundr r, r+ rQ r � MemoryErrorrh )r8 �charsZ
multiplierr� r r r � pre_allocateJ s ��
�z HTTPUploaderData.pre_allocatec C s | j s| �� | j S r> )r r r� r r r rW Z s zHTTPUploaderData.datar� c C sH t �� | j | jkr>| j�� s>| j�|�}| j� t
|�� |S t � �d S r> ) r� r� r� r} r� r rW r� r� r� r� rt )r8 �nr� r r r r� ` s � zHTTPUploaderData.readc C s | j S r> )r r� r r r �__len__i s zHTTPUploaderData.__len__)N)r� )
r r r r r7 r �propertyrW r� r r r r r r 7 s
r c @ s" e Zd ZdZddd�Zdd� ZdS )� HTTPUploaderz Thread class for putting a URLNc C sf t j�| � || _| | jj_| _|| _d| _|| _ || _
|rH|j | _ nt
| _ |rZ|| _nt� | _d S r� )r� r� r7 r� rW r� r� �sizer� r} re r� r� r r� r )r8 re r� r� r r} r� r� r r r r7 p s
zHTTPUploader.__init__c C s� | j }z�t�� | j | jkr�| j�� s�z| �|�}W n8 tyl t | j �
� |j � | j
�d�}| �|�}Y n0 |� d� |�� t| j j j�| _nd| _W n< ttfy� t| j j j�| _Y n ty� d| _Y n0 d S )N)rW � r )r� r� r� r� r} r� r r� r` r� r� rW r� r r| �sumr� r� r� rt r� )r8 r� r9 r r r r� � s( �
�
zHTTPUploader.run)NNr� r r r r r m s
�
r c @ sT e Zd ZdZddd�Zdd� Zd d
� Zd d � Zeddd��Z ddd�Z
ddd�Z dS )�SpeedtestResultsat Class for holding the results of a speedtest, including:
Download speed
Upload speed
Ping/Latency to test server
Data about server that the test was run against
Additionally this class can return a result data as a dictionary or CSV,
as well as submit a POST of the result data to the speedtest.net API
to get a share results image link.
r NFc C sv || _ || _|| _|d u r"i | _n|| _|p.i | _d | _dtj�� �� | _ d| _
d| _ |rd|| _ nt
� | _ || _d S )Nz%sZr )�download�upload�ping�server�client�_share�datetimeZutcnowZ isoformat� timestamp�bytes_received�
bytes_sentr� r� �_secure)r8 r r r r r r� r� r r r r7 � s
zSpeedtestResults.__init__c C s t | �� �S r> )�reprr� r� r r r �__repr__� s zSpeedtestResults.__repr__c
C sb | j r | j S tt| jd d��}tt| jd��}tt| jd d��}d| jd d| ddd| d d
| d d td
|||df �� �� � dddd| j
d| j d| jd g}ddi}t dd�
|��� || jd�}t|| jd�\}}|r�t|��|�� } |j}
|�� t|
�dk�r td ��t| �� �} | �d �} | �rFt| �d k�rNtd ��d | d | _ | j S ) zRPOST data to the speedtest.net API to obtain a share results
link
� @�@r zrecommendedserverid=%s�idzping=%szscreenresolution=zpromo=z download=%sz
screendpi=z upload=%sztestmethod=httpzhash=%sz %s-%s-%s-%sZ297aae72ztouchscreen=nonezstartmode=pingselectz
accuracy=1zbytesreceived=%sz bytessent=%sz serverid=%sZRefererz*http://c.speedtest.net/flash/speedtest.swfz ://www.speedtest.net/api/api.phpr� )rW r� r� �r� �� z)Could not submit results to speedtest.net�resultidr� z&http://www.speedtest.net/result/%s.png)r r( r r r r r r# rQ Z hexdigestr r r� r� r r� r� rr r� �coder| rs r" �decoderH r� )
r8 r r r Zapi_datar� r� r9 r� r� r" Zqsargsr! r r r �share� sT
� � �
zSpeedtestResults.sharec
C s* | j | j| j| j| j| j| j| j| jd� S )z Return dictionary of result data) r r r r r r r r$ r ) r r r r r r r r r r� r r r r� s �zSpeedtestResults.dict�,c C s: g d�}t � }tj|| dd�}|�dd� |D �� |�� S )zReturn CSV Headers)
z Server IDZSponsorz Server NameZ TimestampZDistanceZPingZDownloadZUploadZSharez
IP Address� �� delimiter�lineterminatorc S s g | ] }t |��qS r �rF ��.0rE r r r �
<listcomp> � z/SpeedtestResults.csv_header.<locals>.<listcomp>)r+ �csv�writer�writerow�getvalue)r( �row�outr0 r r r �
csv_header s
zSpeedtestResults.csv_headerc C s� | � � }t� }tj||dd�}|d d |d d |d d |d |d d |d |d
|d | jpfd| jd g
}|�d
d� |D �� |�� S )zReturn data in CSV formatr& r' r r �sponsor�namer r� r r r Zipc S s g | ] }t |��qS r r* r+ r r r r- % r. z(SpeedtestResults.csv.<locals>.<listcomp>)r� r+ r/ r0 r r r1 r2 )r8 r( rW r4 r0 r3 r r r r/ s �zSpeedtestResults.csvc C s. i }|r|� ddd�� tj| �� fi |��S )zReturn data in JSON format� T)ZindentZ sort_keys)r� �json�dumpsr� )r8 Zprettyr: r r r r9 ( s �zSpeedtestResults.json)r r r NNNF)r% )r% )F) r r r r r7 r r$ r� r
r5 r/ r9 r r r r r � s �
=
r c @ sr e Zd ZdZddd�Zedd� �Zd d
� Zdd d �Zd
d� Z ddd�Z
d dd�Z e dfdd�Z
e ddfdd�ZdS ) � Speedtestz>Class for performing standard speedtest.net testing operationsNr� Fc C s� i | _ || _|| _t||�| _|| _|r0|| _nt� | _| �� |d urT| j � |� i | _
g | _ i | _ t
| j d | j|d�| _d S )Nr )r r� r� )�config�_source_address�_timeoutr� r� r r� r �
get_configr� �servers�closest�_bestr �results)r8 r< r~ r} r� r� r r r r7 7 s& �zSpeedtest.__init__c C s | j s| �� | j S r> )rB �get_best_serverr� r r r �bestT s zSpeedtest.bestc C s� i }t rd|d< td|| jd�}t|| jd�\}}|r>t|��g }t|�}z|�|�d�� W n t t
fy~ tt � ��Y n0 t |d �dkrJq�qJ|�
� |�
� t|j�d kr�d
S d �� �|�}td | d
d� zlzt�|�}W n( tj�y t � }td| ��Y n0 |�d�j} |�d�j}
|�d�j} |�d�j} W nr t�y� zt�|�}W n& t �y� t � }td| ��Y n0 t |d�} t |d�}
t |d�} t |d�} Y n0 dd� | d � d�D �}
t| d �}t| d �}g d�}||d d
� g d �d �}t |d �}tt � || ��}|t|
d �d �}t| d �t| d �d! d �}t| d" �t|
d" �d �}| j!�"| |
|||||| d#�� z t#| d$ �t#| d% �f| _$W n0 t%�y� td&| �&d$�| �&d%�f ��Y n0 td'| j! d
d� | j!S )(zgDownload the speedtest.net configuration and return only the data
we are interested in
r� �Accept-Encodingz)://www.speedtest.net/speedtest-config.php�r� r� r r� r� r r Nr& zConfig XML:
%sTr� z)Malformed speedtest.net configuration: %sz
server-configr r r c S s g | ]}|rt |��qS r )r( )r, re r r r r- � s z(Speedtest.get_config.<locals>.<listcomp>Z ignoreidsr% �ratioZ
maxchunkcount)i � i i i i i i p r� )
i^ i� i� r� i� i� i� i� i�
i� )r r Z
threadsperurl�threadsZ threadcountr Z
testlength)r �ignore_servers�sizes�countsrI r �
upload_max�lat�lonz Unknown location: lat=%r lon=%rz
Config:
%r)'r� r� r r� r� rl r� r� r� �OSError�EOFErrorr{ r� r| r( r" rQ r� r� �ET�
fromstring�
ParseErrorrj �find�attribrR �DOM� parseStringr r� �splitr� Zceilr< r� r* �lat_lon�
ValueErrorrH )r8 r� r� r� r� Zconfigxml_list�streamZ configxml�rootZ
server_configr r r rJ rH rM Zup_sizesrK Z
size_countZ upload_countrL rI r r r r r? Z s� ��
�
� �
�
�
��
��
zSpeedtest.get_configc C s� |du r g }|du rg }| j �� ||fD ]F}t|�D ]8\}}zt|�||< W q6 tyl td| ��Y q60 q6q*g d�}i }tr�d|d< g }|D �]�} �z�td| | jd d f || j d �}
t
|
| j d
�\} } | r�|� d | � t
� �t| �}
g }z|� |
�d �� W n" ttf�y* t
t� ��Y n0 t|d
�dkr�qBq�|
�� | �� t| j�dk�rht
� �d�� �|�}td| dd� z�zFzt�|�}W n( tj�y� t� } t d| ��Y n0 t |d�}W nT t �y" zt � |�}W n& t!�y t� } t d| ��Y n0 |�"d�}Y n0 W n$ t#t$j%j&j!f�yJ t
� �Y n0 |D �]
}z
|j'}W n& t �y� t(t)|j*�+� ��}Y n0 |�r�t|�,d��|v�r��qPt|�,d��| jd v �sPt|�,d��|v �rؐqPz(t-| j.t/|�,d��t/|�,d��f�}W n t0�y Y �qPY n0 ||d< z| j | � |� W n t1�yX |g| j |< Y n0 �qPW �q�W q� t
�y� Y q�Y q�0 q�|�s�|�r�| j �s�t2� �| j S )z�Retrieve a the list of speedtest.net servers, optionally filtered
to servers matching those specified in the ``servers`` argument
Nz)%s is an invalid server type, must be int)z1://www.speedtest.net/speedtest-servers-static.phpz3http://c.speedtest.net/speedtest-servers-static.phpz*://www.speedtest.net/speedtest-servers.phpz,http://c.speedtest.net/speedtest-servers.phpr� rF z
%s?threads=%srI r rG r �%sr� r� r r r& zServers XML:
%sTr� z'Malformed speedtest.net server list: %sr r rJ rN rO r� )3r@ �clearra r( r[ rn r� r� r< r r� r� r� rm r� r� rP rQ r{ r� r| r" rQ r� r� rR rS rT rk �
etree_iterrR rW rX r r� � SyntaxError�xmlZparsersZexpatrV r� r� r� r� rH r� rZ r* � Exception�KeyErrorro )r8 r@ �excludeZ server_listre rA �urlsr� r4 r� r� r� r� r\ Zserversxml_listZ
serversxmlr] �elementsr rV r� r r r � get_servers� s�
�
���
�
� �� zSpeedtest.get_serversc
C sB t |�}tj�|d �\}}|r.tj�|�}n|}t|�}t|| jd�\}}|r^td| ��n|� � } |�
� t � d| �
� �}
|
s�dD ]p}z| j�d||f �} W n ty� Y q�0 | � � �� �
� } | jdkr�t| �� �dkr�t �d | �r�|g}
q�q�|�r|
�std
| ��d |d d d|�d
�|
d f d d d�g| _| jS )zaInstead of querying for a list of servers, set a link to a
speedtest mini server
r r zFailed to connect to %sz upload_?[Ee]xtension: "([^"]+)")ZphpZaspZaspxZjspz%s/speedtest/upload.%sr r� z
size=[0-9]z!Invalid Speedtest Mini Server: %szSpeedtest Minir �/)r6 r7 r� r� �latencyr )r! �os�path�splitext�dirnamer� r� r� rp r� r| �re�findallr# r� rc �stripr" r� �
splitlines�matchrq �rstripr@ )
r8 r �urlpartsr7 �extr� r� r� r� �text� extensionr9 rW r r r �set_mini_serverL sX ��
�
�
� ��
zSpeedtest.set_mini_serverr c C sn | j s| �� t| j �� �D ]8}| j | D ]"}| j�|� t| j�|kr* qPq*q qVq td| j dd� | jS )z`Limit servers to the closest speedtest.net servers based on
geographic distance
zClosest Servers:
%rTr� )r@ rh �sorted�keysrA r� r� r� )r8 �limitr� rA r r r �get_closest_servers� s z Speedtest.get_closest_serversc
C s: |s| j s| �� }| j }| jr*| jdf}nd}t� }i }|D �]�}g }tj�|d �}ttj � � d �}d||f } t
dd�D �]&}
d| |
f } t dd | f d
d � t | �} zz| d d kr�t
| d
|d�}
nt| d
|d�}
d|i}d| d | d f }t�� }|
jd ||d� |
�� }t�� | }W n8 t�y^ t� }t d| d
d � |�d� Y q~Y n0 |�d�}t|j�dk�r�|d�� k�r�|�|� n
|�d� |
�� q~tt|�d d d�}|||< q<zt|� � �d }W n t �y� t d��Y n0 || }||d < || j _ || j _!| j"�#|� t d | d
d � |S ) znPerform a speedtest.net "ping" to determine which speedtest.net
server has the lowest latency
r Nr� r� z%s/latency.txt?x=%sr z%s.%sr� r� Tr� r� r� )r~ z
User-Agentz%s?%sr r8 )r� z ERROR: %ri r r z test=testr
r z-Unable to connect to servers to test latency.rj zBest Server:
%r)$rA r} r= r� rk rl rn r( r� r� �ranger� r! r� r� r� r� Z getresponser� r{ r� r� �statusrQ r| r r rz r{ �
IndexErrorru rC r r rB r� )r8 r@ r� r� rC r Zcumr� ZstampZ latency_urlre Zthis_latency_urlru �hr� rl r� �rr� r� rw ZavgZfastestrE r r r rD � sv
� ��
zSpeedtest.get_best_serverc s� g }�j d d D ]@}td�j d d �D ]&}|�dtj��jd �||f � q*qt|�}g }t|�D ] \}} |�t | |�j
d�� qh|p��j d d �ddi�� ����fd d
�}
g �� ��fd d �} t ��} t j
|
| ||fd
�}
t j
| | |fd
�}t�� �|
�� |�� t}||
��r.|
jdd� �q||��rH|jdd� �q.t�� }t���j_�jj|� d �j_�jjdk�r�d�j d d< �jjS )z�Test download speed against speedtest.net
A ``threads`` value of ``None`` will fall back to those dictated
by the speedtest.net configuration
rK r r rL z%s/random%sx%s.jpgr� )r� r� rI c s� t |�D ]v\}}t||��jd d �j�jd�}�d �krLtj�d� q2|�� | � |d� �d d7 < � ||dd� qd S ) Nr r �r� r� rI ���MbP?Tr� �r� )
ra r� r< r� r� r� r� �sleepr� �put��q�requests�
request_countre r� �thread��callback� in_flight� max_threadsr8 r� r r �producer� s � z$Speedtest.download.<locals>.producerc sf t }t��|k rb| �d�}||�r0|jdd� q�d d8 < ��t|j�� � |j|dd� qd S �NTr� �r} rI r� �r[ )�thread_is_aliver� rH r� r� r r� re �r� r� � _is_aliver� �r� �finishedr� r r �consumer s
z$Speedtest.download.<locals>.consumer��targetrO r� r� � @i�� � r )r< r~ r� rk rl rn rE r� ra r� r r r� r� r� r� r� r� r� r rC r r )r8 r� rI rf r �_r� r� re r� r� r� r� � prod_thread� cons_threadr� �stopr �r� r� r� r� r8 r� r r � sN �
�
��
�zSpeedtest.downloadTc s� g }�j d d D ](}td�j d d �D ]}|�|� q*q�j d }g }t|�D ]Z\} }t|d�j d d �jd�}
|r�|
�� d|i} |�t�jd |
�j | d
�|f� qR|p��j d d �d di�� ����fd d
�} g �� ��fdd�}
t
|�p�j d d �}t j | |||fd�}t j |
||fd�}t
�� �|�� |�� t}||��rd|jdd� �qJ||��r~|jdd� �qdt
�� }t���j_�jj|� d �j_�jjS )z�Test upload speed against speedtest.net
A ``threads`` value of ``None`` will fall back to those dictated
by the speedtest.net configuration
rK r r rL rM r r� zContent-lengthr� )r� r� rI c
s� t |d |� �D ]�\}}t||d �|d �jd d �j�jd�}�d �kr^tj�d� qD|�� | � |d� �d d7 < � ||dd � qd S )
Nr r� r r r� rI r� Tr� )
ra r r< r� r� r� r� r� r� r� r� r� r r r� V s � z"Speedtest.upload.<locals>.producerc sb t }t��|k r^| �d�}||�r0|jdd� q�d d8 < ��|j� � |j|dd� qd S r� )r� r� rH r� r� r� re r� r� r r r� j s
z"Speedtest.upload.<locals>.consumerr� g�������?r� r� )r< r~ r� ra r r� r r� rE r r r� r� r� r� r� r� r� r rC r r )r8 r� r rI rK r r� r� r� re rW r� r� r� r� r� r� r� r� r r� r r - s`
����
��
�zSpeedtest.upload)NNr� FN)NN)r )N)r r r r r7 r
rE r? rh ry r} rD r� r r r r r r r; 4 s �
t
~6
LKr; c s � fdd�}|S )zWCatch Ctrl-C key sequence and set a SHUTDOWN_EVENT for our threaded
operations
c s" � � � tddd� t�d� d S )N�
Cancelling...T�rz r )�setr� rI �exit)Zsignum�framer� r r r� � s zctrl_c.<locals>.innerr r� r r� r �ctrl_c� s r� c C s0 t dt � t dtj�dd� � t�d� dS )zPrint the versionzspeedtest-cli %sz Python %sr\ r& r N)r� r� rI �version�replacer� r r r r r� � s r� r% c C s t tj| d�� t�d� dS )zPrint the CSV Headers�r( r N)r� r r5 rI r� r� r r r r5 � s r5 c C s� d} t | d�}z |j|_W n ty, Y n0 |jddddddd � |jd
d dddd d � |jd
dddd� |jddddddd� |jdddd� |jddddd � |jd ddd d � |jd d td!d"� |jd#ddd$d � |jd%ddd&d � |jd'dd(d� |jd)td*d+d,� |jd-td*d.d,� |jd/d0d1� |jd2d3d1� |jd4d5td6d"� |jd7dd8d� |jd9d:dddd;d<� |jd=dd>d� |jd?dttd@� |�� }t |t
��r�|dA }n|}|S )BzAFunction to handle building and parsing of command line argumentsz�Command line interface for testing internet bandwidth using speedtest.net.
--------------------------------------------------------------------------
https://github.com/sivel/speedtest-cli)� descriptionz
--no-downloadr TZ store_constFz Do not perform download test)�dest�default�action�const�helpz --no-uploadr zDo not perform upload testz--singleZ
store_truezYOnly use a single connection instead of multiple. This simulates a typical file transfer.)r� r� r� z--bytes�units)Zbyter� )�bitr� zxDisplay values in bytes instead of bits. Does not affect the image generated by --share, nor output from --json or --csv)r� r� r� r� r� z--sharez]Generate and provide a URL to the speedtest.net share results image, not displayed with --csv)r� r� z--simplez4Suppress verbose output, only show basic information)r� r� r� z--csvzvSuppress verbose output, only show basic information in CSV format. Speeds listed in bit/s and not affected by --bytesz--csv-delimiterr% z<Single character delimiter to use in CSV output. Default ",")r� �typer� z --csv-headerzPrint CSV headersz--jsonzwSuppress verbose output, only show basic information in JSON format. Speeds listed in bit/s and not affected by --bytesz--listz:Display a list of speedtest.net servers sorted by distancez--serverr� zCSpecify a server ID to test against. Can be supplied multiple times)r� r� r� z --excludez?Exclude a server from selection. Can be supplied multiple timesz--miniz URL of the Speedtest Mini server)r� z--sourcez Source IP address to bind toz --timeoutr� z#HTTP timeout in seconds. Default 10z--securezPUse HTTPS instead of HTTP when communicating with speedtest.net operated serversz--no-pre-allocater z�Do not pre allocate upload data. Pre allocation is enabled by default to improve upload performance. To support systems with insufficient memory, use this option to avoid a MemoryError)r� r� r� r� r� z --versionz Show the version number and exitz--debug)r� r� r� r ) � ArgParserZ
add_option� add_argumentrR �PARSER_TYPE_STR�PARSER_TYPE_INT�PARSER_TYPE_FLOAT� ARG_SUPPRESS�
parse_argsrS �tuple)r� �parser�optionsrO r r r r� � s� �
�
�
�
��
�
�
�
�
��
�
�
�����
r� c C sT dt fdtfd�}|�� D ]4\}}t| |d�r|d du rtd|d |f ��qdS ) z�Check if an argument was provided that depends on a module that may
not be part of the Python standard library.
If such an argument is supplied, and the module does not exist, exit
with an error stating which module is missing.
z json/simplejson python modulez SSL support)r9 r� Fr� Nz(%s is not installed. --%s is unavailabler )r9 r r� rV �
SystemExit)rO Z
optional_argsrb �infor r r �validate_optional_args� s �
�r� c K sX |r t s dS |r.tj�� r$d| }q2d| }n| }|r@tj|d< |sTt|fi |�� dS )z4Helper function print a string with various featuresNz[1;30mDEBUG: %s[0mz DEBUG: %srG )�DEBUGrI r_ �isattyrJ rP )r) �quietr� rz r: r4 r r r r� s
r� c
C s� t �� } t�tjt| �� t� }|jr,t� |js@|js@t d��t
|j �dkrVt d��|j rft |j � t
|� t|dd�}|dkr�d}|r�da|js�|js�|jr�d}nd}|js�|jr�d}nd}|s�|r�t}nt| �}td|� zt|j|j|jd �}W n0 tft �y( td
dd � t t � ��Y n0 |j �r�z |� � W n0 t ft �yn td dd � t t � ��Y n0 t |j!�"� �D ]V\}}|D ]F} d
| }
z t|
� W n* t#�y� t � } | j$t$j%k�rȂ Y n0 �q��q~t&�'d� td|j(d |� |j)�s�td|� z|j |j*|j+d� W n� t,�yN t dd�-dd� |j*D �� ��Y n` t ft �y| td dd � t t � ��Y n2 t.�y� t dd�-dd� |j*D �� ��Y n0 |j*�r�t
|j*�dk�r�td|� n
td|� |�/� n|j)�r |�/|�0|j)�� |j1} td| j* |� |j�rvtd |d t2|� d � |j|d |j3 d � td!| jd" d" |j4d |j4d f |� n
td#|� |j�r�td$|d t2|� d � |j||j5d |j3 d%� td&| jd" d" |j4d |j4d f |� n
td'|� td(| �6� dd)� |j�s|j7�r| �7� |j�rptd*| j8| jd" d" |j4d |j4d | jd" d" |j4d |j4d f � n0|j�r�t| j|j d+�� n|j�r�t| �� � |j7�r�|�s�td,| �7� � d-S ).z Run the full speedtest.net testz0Cannot supply both --no-download and --no-uploadr� z*--csv-delimiter must be a single characterr� FZ SUPPRESSHELPTz)Retrieving speedtest.net configuration...)r~ r} r� z'Cannot retrieve speedtest configurationr� z%Cannot retrieve speedtest server listz:%(id)5s) %(sponsor)s (%(name)s, %(country)s) [%(d)0.2f km]r z Testing from %(isp)s (%(ip)s)...r z'Retrieving speedtest.net server list...)r@ re zNo matched servers: %sz, c s s | ]}d | V qdS �r^ Nr �r, rA r r r � <genexpr>y r. zshell.<locals>.<genexpr>z,%s is an invalid server type, must be an intc s s | ]}d | V qdS r� r r� r r r r� � r. z1Retrieving information for the selected server...z&Selecting best server based on ping...z>Hosted by %(sponsor)s (%(name)s) [%(d)0.2f km]: %(latency)s mszTesting download speed)r& r\ r� )Nr� )r� rI zDownload: %0.2f M%s/sr zSkipping download testzTesting upload speed)r� r rI zUpload: %0.2f M%s/szSkipping upload testz Results:
%rr� z5Ping: %s ms
Download: %0.2f M%s/s
Upload: %0.2f M%s/sr� zShare results: %sN)9r� �Event�signal�SIGINTr� r� r� r r rh r� Z
csv_delimiterr5 r� rV r� �simpler/ r9 r� r� r� r; �sourcer} r� rl r� r{ r� rh rm rz r@ r� r� �errnoZEPIPErI r� r< Zminir re ro r� rn rD ry rC r� Zsingler� r r� r$ r )
r� rO r� r� Zmachine_formatr� Z speedtestr� r@ r �liner� rC r r r �shell! s
�
�
�
�� ��
��
�����
�����
�� r� c C sx z
t � W nh ty( tddd� Y nL ttfyr t� } t| dd�dvrnd| }|sbd| }td | ��Y n0 d S )
Nr� Tr� r" r� )r r r^ z%rz ERROR: %s)r� �KeyboardInterruptr� rf r� r{ rV )r� �msgr r r �main� s
r� �__main__)N)Nr� )NNr� F)N)r% )FFF)�rk ro r/ rI r� r� r� rw r� r r� r� Zxml.parsers.expatrb r� r� Z GZIP_BASE� ImportError�objectr� r r� rx � version_infoZPY25PLUSZPY26PLUSZPY32PLUSr9 Z
simplejsonZxml.etree.ElementTreeZetreeZ ElementTreerR r Z
ET_ElementZxml.domr rW r Zurllib2r r r r r r r r r r Zurllib.requestZhttplibr r Z http.clientr r r Zqueuer! � urllib.parser" ZcgiZhashlibr# �argparser$ r� r% r� r( r� rT r� r* r� Zoptparser& r' Z cStringIOr+ r, r� Z __builtin__�builtinsr- r. r/ rV rM r_ rL rJ rK rP rF rP ZElement�iterr` Z getiteratorr� �is_aliver� ZisAliver� ZCertificateErrorZ
CERT_ERRORrR r� rz ZSSLErrorr� rc rf rh ri rj rk rl rm rn ro rp rq rr rs rt ru rv r� r� r� r� r� r� r� r� r{ r� r� r� r� r� r� r� r� r� r r r r; r� r� r5 r� r� r� r� r� r r r r r �<module> s�
46
=
���
)"B
$
$
&60 Z
V
#