c$ddlZ ddlmZmZn#e$r ddlmZmZYnwxYwddlZddlZddlZddlZddl Z ddl Z ddl m Z n#e$r ddlm Z YnwxYwddlZGddejZedkre jZee jejdZeZejZedej j!d ej j"dj!d ed Z#ee$ej j%&e#ejeZ'e'(dSdS) N)EmptyQueue)urlopencneZdZdZGddejZddZdZdd Z d Z dd Z dd Z dZ dS) MirrorTestz@Determines the best mirrors by perfoming ping and download test.c eZdZdZddZdZdS)MirrorTest.PingWorkerzUse the command line command ping to determine the server's response time. Using multiple threads allows to run several test simultaneously.rrrc||_||_||_||_||_||_t jd|_tj |dS)Nz^rtt .* = [\.\d]+/([\.\d]+)/.*) bordersmodparentidjobsresultsrecompile match_result threadingThread__init__)selfrrrrrrs ?/usr/lib/python3/dist-packages/softwareproperties/MirrorTest.pyrzMirrorTest.PingWorker.__init__s\"DLDH DKDGDI"DL " +L M MD    % %d + + + + +c(d}|js|jjr |jd}|j}|jd|ztj ddddd|gtj tj d j } | }|sntj|j|}2n0#t"$rYdS|jd |zYnxYwt$jt$xjd z c_|jt$jt$j|j|j|r0|jt9|d ||gt$j|js%|jjdSdSdSdS) NFz Pinging %s...pingz-qz-c 2z-W 1z-i 0.5T)stdoutstderruniversal_newlinesz Skipping %sr r)remptyrrunningisSetgethostname report_action subprocessPopenPIPESTDOUTrreadlinerfindallrrrcompleted_lockacquire completedreport_progresstodorrrappendfloatrelease)rresultmirrorhostcommandolines rrunzMirrorTest.PingWorker.run#s+Fioo'' 4DK,?,E,E,G,G 4D!Y]]511F!?DK--o.DEEE)/vvxY]0^7A7A7HCG I I IJPE'0022#"!!#D,=t!D!D E"FFDK--md.BCCCCC)11333$$)$$ ++J,@,6O,0L,0H666JL''vay)9)94(HIII)113339ioo'' 4DK,?,E,E,G,G 4 4 4 4 4 4 4 4 4sB(C&& D3DN)r r )__name__ __module__ __qualname____doc__rr;rr PingWorkerr sA # # , , , , 4 4 4 4 4rrANcptj|d|_d|_||_d|_||_g|_dt_ tj t_ t|t_||_|stj|_dS||_dS)N)rrgr)rrractionprogresseventbest test_filethreadsrr0Lockr.lenr2mirrorsEventr#)rrLrHrFr#s rrzMirrorTest.__init__Cs!!$''' #   "  $-N$4$4 !g,,   #$?,,DLLL"DLLLrcX||_|jr|jdSdSN)rDrFset)rtexts rr'zMirrorTest.report_actionTs5 :  JNN       rrdr c||dz||dz|d|d|dz |z |zzf|_|jr|jdSdS)zTSubclasses should override this method to receive progress status updatesrr N)rErFrP)rcurrentmaxrrs rr1zMirrorTest.report_progressYsv!3q6)s1v wqzGAJ'>#&E&OOQ  :  JNN       rc T|ddd}t|j}|dkrr|dd|jt jd|dz g|dd|jt jd|dz g|d|Dd tjtjf }|D]\}}td |j d | |sdS|ddj }td |z|S)N)r?)r)rVrrrr cg|] }|d S)r[r@).0rs r z,MirrorTest.run_full_test..js)E)E)E1!A$)E)E)ErrYr )rrzmirror: z - time: zand the winner is: %s) run_ping_testrKrLr3randomrandintrun_download_testrr2printr&)r results_pingsizerthwinners r run_full_testzMirrorTest.run_full_testbsB))au)MM 4<   !88   At|FN1d1f4M4M'N O P P P   At|FN1d1f4M4M'N O P P P(()E)E )E)E)E19.8o.8o.?)@@ = =FQ EQZZZ; < < < < 4QZ]+F )F2 3 3 3Mrr c |dkr|j}t}|D]}||g}tdD]O}t||||||} |j| | P|jD]} | | |d|S)zPerforms 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.Nr) rLrputrangerrArIr3startjoinsort) rrLrVrrrmrirhs rrazMirrorTest.run_ping_testws d??lGww  A HHQKKKKr  A%%dGQgsKKA L   " " " GGIIII  A FFHHHH qu~rcnfd}|dkrj}g}|D]}jsnd||}|dkr|||g||dzt |d|||d|S)zPerforms 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(|ddj}d|ztj} t |ddtj|z S#YdSxYw)Nr/zDownloading %s...r[)timeouti) get_repo_urlsrHr'timerread)r7urlrprs rtest_download_speedz9MirrorTest.run_download_test..test_download_speeds#1133A666!^^-C   2S8 9 9 9IKKE Q''',,V444y{{U** qqs 9B BNrr r`)rLr#r$r3r1indexrKrr) rrLrVrrr}rrs download_times ` rrdzMirrorTest.run_download_tests      d??lG T TA<%%'' //22Mq   q1222  q!1!1A!5s7||Xs S S S S qu~rcj||_|jdS)z/Complete test exercise, set self.best when doneN)rkrGr#clear)rs rr;zMirrorTest.runs/&&((  rrO)rRr )NNr r )r<r=r>r?rrrArr'r1rkrardr;r@rrrrsJJ+4+4+4+4+4Y%+4+4+4Z####" *.AF#<rr__main__zdpkg --print-architecturezdists/rwz/binary-z /Packages.gz))rqueuerr ImportErrorrzrr(osaptsources.distro aptsourcesaptsources.sourceslisturllib.requestrurllib2rbrrr<distro get_distro get_sources sourceslist SourcesListpopenpiper{striparchrMr#rPsource_templatename componentsrHlist mirror_setvaluesapprkr@rrrsZ#"""""""""###""""""""""#   &&&&&&&    [[[[[![[[z z   ) ) + +F z-99;;<<< 28/ 0 0D 99;;    DioG KKMMMM',,,'215:::I *TT&0;BBDDEE  1 17 <