getGolangPackages.py

from modules.Packages import getPackagesFromPkgDb, loadPackages, LocalDB

import optparse

def getPkgs():
	pkgs_pkgdb = getPackagesFromPkgDb()
	pkgs_local = loadPackages()

	# Any new or removed packages
	set_pkgdb = set(pkgs_pkgdb)
	set_local = set(pkgs_local)

	return set_pkgdb, set_local

def printPkgs(pkgs):
	for pkg in pkgs:
		print pkg

if __name__ == "__main__":

	parser = optparse.OptionParser("%prog [-n] [-r] [-l] [-u]")

	parser.add_option(
	    "", "-n", "--new", dest="new", action = "store_true", default = False,
	    help = "Scan for new golang packages"
	)

	parser.add_option(
	    "", "-r", "--removed", dest="removed", action = "store_true", default = False,
	    help = "Scan for removed golang packages"
	)

	parser.add_option(
	    "", "-l", "--list", dest="list", action = "store_true", default = False,
	    help = "List all golang packages saved in a local database"
	)

	parser.add_option(
	    "", "-u", "--update", dest="update", action = "store_true", default = False,
	    help = "Update local databases with new golang packages"
	)

	options, args = parser.parse_args()

	# new packages
	if options.new:
		set_pkgdb, set_local = getPkgs()
		printPkgs(list(set_pkgdb - set_local))
	elif options.removed:
		set_pkgdb, set_local = getPkgs()
		printPkgs(list(set_local - set_pkgdb))
	elif options.list:
		pkgs = loadPackages()
		printPkgs(sorted(pkgs))
	elif options.update:
		set_pkgdb, set_local = getPkgs()	
		ldb = LocalDB()
		new_pkgs = list(set_pkgdb - set_local)
		err, ret = ldb.addPackages(new_pkgs)
		if ret:
			print "%s packages added" % len(new_pkgs)
			if err != []:
				print "\n".join(err)
		else:
			print "\n".join(err)
	else:
		print "Synopsis: [-n] [-r] [-l] [-u]"