Daniel Holbach ad837ae9cf harvest, setup.py: install tool that queries Harvest for information
about open opportunities for a given source package.
2011-01-26 09:33:18 +01:00

52 lines
1.4 KiB
Python
Executable File

#!/usr/bin/python
# Daniel Holbach
# (c) 2011 Canonical
import urllib2
import json
import sys
BASE_URL = "http://harvest.ubuntu.com/"
URL_STUB = BASE_URL + "opportunities/json/"
def opportunity_summary(data):
l = []
for key in filter(lambda a: a != "total", data.keys()):
l += ["%s (%s)" % (key, data[key])]
return ", ".join(l)
def main():
if len(sys.argv) == 1 or not sys.argv[1].strip() or \
sys.argv[1].strip() in ["--help", "-h"]:
print >> sys.stderr, "Usage: %s <source package name>" % sys.argv[0]
sys.exit(0)
pkg = sys.argv[1]
url = URL_STUB + pkg.strip()
try:
sock = urllib2.urlopen(url)
except IOError, e:
try:
urllib2.urlopen(BASE_URL)
except urllib2.URLError, _e:
print >> sys.stderr, "Harvest is down."
sys.exit(1)
print "There is no information in Harvest about package '%s'." % pkg
sys.exit(1)
response = sock.read()
sock.close()
data = json.loads(response)
print >> sys.stdout, \
"""%s has %s opportunities: %s
Find out more: %sopportunities/package/%s""" % (pkg,
data["total"],
opportunity_summary(data),
BASE_URL,
pkg)
if __name__ == '__main__':
try:
main()
except KeyboardInterrupt:
print >> sys.stderr, "Aborted."
sys.exit(1)