mirror of
https://github.com/moparisthebest/pacman
synced 2024-12-25 01:08:51 -05:00
ba1806f5ac
duplicated. * Updated the util Makefile.am to link with the proper libalpm.la. * Fixed bitmasking issues in be_files.c and db.h. * Rankmirrors updates from James Rosten (with some cleaning up of my own). KeyboardInterrupts are now handled gracefully.
158 lines
5.3 KiB
Python
158 lines
5.3 KiB
Python
#! /usr/bin/python
|
|
#
|
|
# rankmirrors : read a list of mirrors from a file and rank them by speed
|
|
#
|
|
# Original Idea copyright (c) 2006 R.G. <chesercat>
|
|
# Modified 2006 by Dan McGee <dan@archlinux.org>
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation; either version 2 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program; if not, write to the Free Software
|
|
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307,
|
|
# USA.
|
|
#
|
|
import os, sys, datetime, time, socket, urllib2
|
|
from optparse import OptionParser
|
|
|
|
def createOptParser():
|
|
usage = "usage: %prog [options] MIRRORFILE | URL"
|
|
description = "Ranks pacman mirrors by their connection and opening " \
|
|
"speed. Pacman mirror files are located in /etc/pacman.d/. It " \
|
|
"can also rank one mirror if the URL is provided."
|
|
parser = OptionParser(usage = usage, description = description)
|
|
parser.add_option("-n", type = "int", dest = "num", default = 0,
|
|
help = "number of servers to output, 0 for all")
|
|
parser.add_option("-t", "--times", action = "store_true",
|
|
dest = "times", default = False,
|
|
help = "only output mirrors and their response times")
|
|
parser.add_option("-u", "--url", action = "store_true", dest = "url",
|
|
default = False, help = "test a specific url")
|
|
parser.add_option("-v", "--verbose", action = "store_true",
|
|
dest = "verbose", default = False,
|
|
help = "be verbose in ouptut")
|
|
return parser
|
|
|
|
def timeCmd(cmd):
|
|
before = time.time()
|
|
try:
|
|
cmd()
|
|
except KeyboardInterrupt, ki:
|
|
raise ki
|
|
except socket.timeout, ioe:
|
|
return 'timeout'
|
|
except Exception, e:
|
|
return 'unreachable'
|
|
return time.time() - before
|
|
|
|
def talkToServer(serverUrl):
|
|
opener = urllib2.build_opener()
|
|
tmp = opener.open(serverUrl).read()
|
|
|
|
def getFuncToTime(serverUrl):
|
|
return lambda : talkToServer(serverUrl)
|
|
|
|
def cmpPairBySecond(p1, p2):
|
|
if p1[1] == p2[1]:
|
|
return 0
|
|
if p1[1] < p2[1]:
|
|
return -1
|
|
return 1
|
|
|
|
def printResults(servers, time, verbose, num):
|
|
items = servers.items()
|
|
items.sort(cmpPairBySecond)
|
|
itemsLen = len(items)
|
|
numToShow = num
|
|
if numToShow > itemsLen or numToShow == 0:
|
|
numToShow = itemsLen
|
|
if itemsLen > 0:
|
|
if time:
|
|
print
|
|
print ' Servers sorted by time (seconds):'
|
|
for i in items[0:numToShow]:
|
|
if i[1] == 'timeout' or i[1] == 'unreachable':
|
|
print i[0], ':', i[1]
|
|
else:
|
|
print i[0], ':', "%.2f" % i[1]
|
|
else:
|
|
for i in items[0:numToShow]:
|
|
print 'Server =', i[0]
|
|
|
|
if __name__ == "__main__":
|
|
parser = createOptParser()
|
|
(options, args) = parser.parse_args()
|
|
|
|
if len(args) != 1:
|
|
parser.print_help(sys.stderr)
|
|
sys.exit(0)
|
|
|
|
# allows connections to time out if they take too long
|
|
socket.setdefaulttimeout(10)
|
|
|
|
if options.url:
|
|
if options.verbose:
|
|
print 'Testing', args[0] + '...'
|
|
try:
|
|
serverToTime = timeCmd(getFuncToTime(args[0]))
|
|
except KeyboardInterrupt, ki:
|
|
sys.exit(1)
|
|
if serverToTime == 'timeout' or serverToTime == 'unreachable':
|
|
print args[0], ':', serverToTime
|
|
else:
|
|
print args[0], ':', "%.2f" % serverToTime
|
|
sys.exit(0)
|
|
|
|
if not os.path.isfile(args[0]):
|
|
print >>sys.stderr, 'rankmirrors: file', args[0], 'does not exist.'
|
|
sys.exit(1)
|
|
|
|
fl = open(args[0], 'r')
|
|
serverToTime = {}
|
|
if options.times:
|
|
print 'Querying servers, this may take some time...'
|
|
else:
|
|
print "# Server list generated by rankmirrors on",
|
|
print datetime.date.today()
|
|
for ln in fl.readlines():
|
|
splitted = ln.split('=')
|
|
if splitted[0].strip() != 'Server':
|
|
if not options.times:
|
|
print ln,
|
|
continue
|
|
|
|
serverUrl = splitted[1].strip()
|
|
if serverUrl[-1] == '\n':
|
|
serverUrl = serverUrl[0:-1]
|
|
if options.verbose and options.times:
|
|
print serverUrl, '...',
|
|
elif options.verbose:
|
|
print '#', serverUrl, '...',
|
|
elif options.times:
|
|
print ' * ',
|
|
sys.stdout.flush()
|
|
try:
|
|
serverToTime[serverUrl] = timeCmd(getFuncToTime(serverUrl))
|
|
if options.verbose:
|
|
try:
|
|
print "%.2f" % serverToTime[serverUrl]
|
|
except:
|
|
print serverToTime[serverUrl]
|
|
except:
|
|
print
|
|
printResults(serverToTime, options.times, options.verbose,
|
|
options.num)
|
|
sys.exit(0)
|
|
|
|
printResults(serverToTime, options.times, options.verbose, options.num)
|
|
|
|
# vim: set ts=4 sw=4 et:
|