server/autoserv_parser.py

__author__ = "raphtee@google.com (Travis Miller)"


import os, sys, getopt, optparse


class base_autoserv_parser(object):
	"""Custom command-line options parser for autoserv.

	We can't use the general getopt methods here, as there will be unknown
	extra arguments that we pass down into the control file instead.
	Thus we process the arguments by hand, for which we are duly repentant.
	Making a single function here just makes it harder to read. Suck it up.
	"""
	def __init__(self):
		self.args = sys.argv[1:]
		self.parser = optparse.OptionParser()
		self.setup_options()
		self.parse_args()
		

	def setup_options(self):
		self.parser.add_option("-m", action="store", type="string",
				       dest="machines",
				       help="list of machines")
		self.parser.add_option("-M", action="store", type="string",
				       dest="machines_file",
				       help="list of machines from file")
		self.parser.add_option("-c", action="store_true",
				       dest="client", default=False,
				       help="control file is client side")
		self.parser.add_option("-r", action="store", type="string",
				       dest="results", default='.',
				       help="specify results directory")
		self.parser.add_option("-l", action="store", type="string",
				       dest="label", default='',
				       help="label for the job")
		self.parser.add_option("-u", action="store", type="string",
				       dest="user", 
				       default=os.environ.get('USER'),
				       help="username for the job")
		self.parser.add_option("-P", action="store", type="string",
				       dest="parse_job", 
				       default='',
				       help="parse the results of the job")
		self.parser.add_option("-i", action="store_true",
				       dest="install_before", default=False,
			       help="reinstall machines before running the job")
		self.parser.add_option("-I", action="store_true",
				       dest="install_after", default=False,
			        help="reinstall machines after running the job")
	        self.parser.add_option("-b", action="store_true",
				       dest="reboot", default=False,
			               help="reboot all machines after job")
		self.parser.add_option("-v", action="store_true",
				       dest="verify", default=False,
			               help="verify the machines only")
		self.parser.add_option("-R", action="store_true",
				       dest="repair", default=False,
			               help="repair the machines")
		self.parser.add_option("-n", action="store_true",
				       dest="no_tee", default=False,
			              help="no teeing the status to stdout/err")
		self.parser.add_option("-p", action="store_true",
				       dest="write_pidfile", default=False,
			              help="write pidfile (.autoserv_execute)")
		self.parser.add_option("--ssh-user", action="store",
				       type="string", dest="ssh_user",
				       default="root",
				       help=("specify the user for ssh"
				       "connections"))
		self.parser.add_option("--ssh-port", action="store",
				       type="int", dest="ssh_port",
				       default=22,
				       help=("specify the port to use for "
					     "ssh connections"))
		self.parser.add_option("--ssh-pass", action="store",
				       type="string", dest="ssh_pass",
				       default="",
				       help=("specify the password to use "
					     "for ssh connections"))


	def parse_args(self):
		(self.options, self.args) = self.parser.parse_args()



try:
	from autotest_lib.server.site_autoserv_parser \
	     import site_autoserv_parser
except ImportError:
	class site_autoserv_parser(base_autoserv_parser):
		pass


class autoserv_parser(site_autoserv_parser):
	pass


# create the one and only one instance of autoserv_parser
autoserv_parser = autoserv_parser()