diff options
| -rw-r--r-- | scripts/rankmirrors | 116 |
1 files changed, 116 insertions, 0 deletions
diff --git a/scripts/rankmirrors b/scripts/rankmirrors new file mode 100644 index 00000000..21057104 --- /dev/null +++ b/scripts/rankmirrors @@ -0,0 +1,116 @@ +#! /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 <dpmcgee@gmail.com> +# +# 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 sys, datetime, time, socket, urllib2 +from optparse import OptionParser + +def createOptParser(): + usage = "usage: %prog [options] MIRRORFILE" + description = "Ranks pacman mirrors by their connection and +opening speed. Pacman mirror files are located in /etc/pacman.d/." + parser = OptionParser(usage=usage,description=description) + parser.add_option("-f", "--formatted", action="store_true", + dest = "formatted", default=False, help="output in mirror +file format") + parser.add_option("-n", dest="num", default=0, + help="number of servers to output, 0 for all") + parser.add_option("-v", "--verbose", action="store_true", dest="verbose", +- Show quoted text - + default=False, help="be verbose in output") + 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 + +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) + + fl = open(args[0], 'r') + serverToTime = {} + if options.formatted: + print "# Server list generated by rankmirrors on", datetime.date.today() + else: + print 'Querying servers, this may take some time...' + for ln in fl.readlines(): + splitted = ln.split('=') + if splitted[0].strip() != 'Server': + if options.formatted: + print ln, + continue + + serverUrl = splitted[1].strip() + if serverUrl[-1] == '\n': serverUrl = serverUrl[0:-1] + if options.verbose and options.formatted: + print '#',serverUrl,'...', + elif options.verbose: + print serverUrl,'...', + elif not options.formatted: + print ' * ', + sys.stdout.flush() + serverToTime[serverUrl] = timeCmd(getFuncToTime(serverUrl)) + if options.verbose: + print serverToTime[serverUrl] + + items = serverToTime.items() + items.sort(cmpPairBySecond) + numToShow = int(options.num) + if numToShow == 0: numToShow = len(items) + if len(items) > 0: + if options.formatted: + for i in items[0:numToShow]: + print 'Server =', i[0] + else: + print + print ' Servers sorted by time:' + for i in items[0:numToShow]: + print i[1], ':', i[0] + +# vim: set ts=4 sw=4 sta et sts ai: |