X-Git-Url: https://git.ao2.it/winfreed.git/blobdiff_plain/b0483c09d50e673ec761c0518d285d3f9df250ee..81ffa4db01ae7dc3d57adf72776c89ba1ed1f6ab:/winfreed.py diff --git a/winfreed.py b/winfreed.py index c465b63..8b674c3 100755 --- a/winfreed.py +++ b/winfreed.py @@ -2,7 +2,7 @@ # # winfreed - download a selection of Free Software for MS Windows. # -# Copyright (C) 2011 Antonio Ospite +# Copyright (C) 2011-2014 Antonio Ospite # # 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 @@ -20,58 +20,102 @@ import os import sys import glob -import json +import ConfigParser import urllib2 from progressbar import Bar, ETA, FileTransferSpeed, Percentage, ProgressBar -# TODO make OUTPUT_DIR and LANGCODE configurable from command line OUTPUT_DIR = 'downloads' LANGCODE = 'en-US' -# TODO PKG_DIR in $(datadir) or something like that for python projects +# One of i386 or x64 +ARCHITECTURE = 'x64' + PKG_DIR = 'pkgs' CHUNK_SIZE = 8192 + +def get_pkg(pkg_file): + config = ConfigParser.SafeConfigParser({'language': LANGCODE}) + with open(pkg_file, mode='r') as f: + config.readfp(f) + pkg = dict(config.items('Package')) + basename = os.path.basename(pkg_file) + package_name = os.path.splitext(basename)[0] + pkg['package_name'] = package_name + f.close() + return pkg + + return None + + def process_all(path, cb): - listing = glob.glob(os.path.join(path, '*.json')) - for json_file in listing: - with open(json_file, mode='r') as f: - entry = json.load(f) - basename = os.path.basename(json_file) - package_name = os.path.splitext(basename)[0] - entry['package'] = package_name - cb(entry) - f.close() - -def show(entry): - print 'Package: ', entry['package'] - print 'Program: ', entry['name'] - print 'Homepage: ', entry['homepage'] + listing = sorted(glob.glob(os.path.join(path, '*.ini'))) + for pkg_file in listing: + pkg = get_pkg(pkg_file) + if not pkg: + sys.stderr.write("Error: cannot get a pkg for: %s\n" % pkg_file) + continue + cb(pkg) + + +def show(pkg): + print 'Package: ', pkg['package_name'] + print 'Program: ', pkg['name'] + if pkg.has_key('version'): + print 'Version: ', pkg['version'] + print 'Homepage: ', pkg['homepage'] print -def download(entry): - # the "%s" in URLs are meant to be replaced with LANGCODE + +def download_file(src_url, dest_dir): try: - url = entry['URL'] % LANGCODE - except: - url = entry['URL'] - pass - - response = urllib2.urlopen(url) - filename = urllib2.unquote(os.path.basename(response.geturl())) - destfile = os.path.join(OUTPUT_DIR, filename) + response = urllib2.urlopen(src_url) + except Exception, e: + print 'Downloading from', src_url, 'failed.', e + return + + # get the final URL after possible redirect have been followed + url = response.geturl() + + filename = "" + + # From http://paste.pound-python.org/show/9545/ + # TODO: use a proper module to parse HTTP headers + if 'Content-Disposition' in response.info() and len(response.info()['Content-Disposition'].split('filename=')) > 1: + # If the response has Content-Disposition, we take file name from it + filename = response.info()['Content-Disposition'].split('filename=')[1].decode('utf-8') + if filename[0] == '"' or filename[0] == "'": + filename = urllib2.unquote(filename.split('"')[1]) + else: + filename = urllib2.unquote(url.split('/')[-1].decode('utf_8')) + + if filename == "": + sys.stderr.write("Debug (%s): filename: %s url: %s\n" % (pkg['package_name'], filename, url)) + return + + # TODO: Add some integrity verification of downloaded files (md5, sha256?) + + destfile = os.path.join(dest_dir, filename) if os.path.exists(destfile): + # TODO: check if the file is a full download from previous run, + # if not download again discarding the existing file? sys.stderr.write("Warning: %s exists!\n" % destfile) return - outfile = open(destfile, mode='w') + + if 'Content-Length' not in response.info(): + sys.stderr.write("Cannot get the size of the file at %s\n" % src_url) + return total_size = response.info().getheader('Content-Length').strip() total_size = int(total_size) - widgets = [entry['name'], ' ', Percentage(), ' ', Bar(marker='=', left='[', right=']'), + #widgets = [pkg['name'], ' ', Percentage(), ' ', Bar(marker='=', left='[', right=']'), + widgets = [filename, ' ', Percentage(), ' ', Bar(marker='=', left='[', right=']'), ' ', ETA(), ' ', FileTransferSpeed()] pbar = ProgressBar(widgets=widgets, maxval=total_size).start() + outfile = open(destfile, mode='w') + bytes_so_far = 0 while 1: chunk = response.read(CHUNK_SIZE) @@ -83,15 +127,28 @@ def download(entry): pbar.update(bytes_so_far) pbar.finish() + outfile.close() + + +def download(pkg): + try: + pkg_url = pkg['url_' + ARCHITECTURE] + except KeyError: + pkg_url = pkg['url'] + download_file(pkg_url, OUTPUT_DIR) + + def show_all(): process_all(PKG_DIR, show) + def download_all(): if os.path.exists(OUTPUT_DIR) == False: os.mkdir(OUTPUT_DIR, 0755) process_all(PKG_DIR, download) + def usage(): usage = "winfreed - download a selection of Free Software for MS Windows.\n\n" usage += "usage: %s \n\n" % sys.argv[0] @@ -107,11 +164,15 @@ if __name__ == "__main__": sys.exit(1) if sys.argv[1] == 'download': - download_all() + if len(sys.argv) > 2: + pkg = get_pkg(sys.argv[2]) + download(pkg) + else: + download_all() elif sys.argv[1] == 'show': show_all() else: usage() sys.exit(1) - + sys.exit(0)