#!/usr/bin/env python

import sys, os

def usage():
	print "Usage: %s <repo-set>" % (sys.argv[0],)

def main(argv):
	if argv[1] != None:
		sources_path = "/etc/apt/sources.list.%s" % (argv[1],)
		if os.path.exists(sources_path):
			os.unlink("/etc/apt/sources.list")
			os.symlink(sources_path, "/etc/apt/sources.list")
			print "SUCCESS: Don't for get to run 'sudo apt-get update'"
			return 0
		else:
			print "ERROR: Unknown repository set '%s'" % (argv[1],)
			usage()
	else:
		usage()
	return 1

if __name__ == '__main__':
	sys.exit(main(sys.argv))