Mercurial > hg > config
diff python/pypi.py @ 369:11cf7018ebff
new crappy package for helping me with http://k0s.org/portfolio/python/python-tools.txt
author | Jeff Hammel <jhammel@mozilla.com> |
---|---|
date | Sun, 21 Jul 2013 05:27:28 -0700 |
parents | |
children | 016d27ca39e4 |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/python/pypi.py Sun Jul 21 05:27:28 2013 -0700 @@ -0,0 +1,92 @@ +#!/usr/bin/env python + +""" +pypi strawman that should really be in PaInt + +ref http://wiki.python.org/moin/PyPIJSON +""" + +import json +import optparse +import os +import sys +import textwrap +import urllib2 + +def package_data(package, index='http://pypi.python.org/pypi'): + """ + returns dict of {url:, description:, home_page: } + """ + + # if package name given, -> url + url = package + if '://' not in url: + url = '%s/%s' % (index, package) + url = url.rstrip('/') + + if not url.endswith('/json'): + url = '%s/json' % (url) + + # read the URL + try: + page = urllib2.urlopen(url).read() + except: + raise # TODO: better error handling + + data = json.loads(page) + info = data['info'] + + # massage the data + return {'summary': info.get('summary', ''), + 'package_url': info['package_url'], + 'name': info['name'], + 'home_page': info.get('home_page') + # etc + } + + +def main(args=sys.argv[1:]): + + # command line parser + usage = '%prog [options] package <another_package> <...>' + parser = optparse.OptionParser(usage=usage, description=__doc__) + parser.add_option('-t', '--tag', dest='tags', + action='append', help="tags for packages") + options, args = parser.parse_args(args) + if not args: + parser.print_usage() + parser.exit() + + # one crappy ass template + template = """ * %(package_url)s +%(summary)s + %(tags)s +""" + + # get package data + packages = {} + for package in args: + data = package_data(package) + packages[data['name']] = data + + # render the damn thing + rendered = [] + for name in sorted(packages.keys(), key=lambda x: x.lower()): + values = packages[name].copy() + if options.tags: + values['tags'] = '{%s}' % ', '.join(options.tags) + else: + values['tags'] = '' + if values['home_page'] and values['home_page'] != values['package_url']: + values['summary'] = '%s %s' % (values['summary'], + values['home_page']) + values['summary'] = textwrap.fill(values['summary'], 69, + initial_indent=' ', + subsequent_indent=' ') + string = template % values + rendered.append(string.strip()) + + print '\n\n'.join([i.strip() for i in rendered]) + +if __name__ == '__main__': + main()