X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=config.py;h=a3032c084c6d76cc29104434d568a7986b3ad123;hb=dc70043f70f08f0d6540c77e53d30c0ed6e58ad5;hp=112173fba238e36ca008cde62f307692e054377c;hpb=55ec8f3eb860029ecf552d8c0a3cc6dbd2f66b68;p=monitor.git

diff --git a/config.py b/config.py
index 112173f..a3032c0 100644
--- a/config.py
+++ b/config.py
@@ -1,13 +1,175 @@
-from xml.sax import saxutils
+#!/usr/bin/python
+import pickle
+import os
+import getopt
+import sys
+import __main__
+from optparse import OptionParser
 
-class config(saxutils.DefaultHandler):
-	def __init__(self, file, start):
-		self.file = file
-		self.start = start
-		self.config = {}
+config_command = False
 
-	def startElement(self,name, attrs):
-		if name != self.start: return
+XMLRPC_SERVER="https://boot.planet-lab.org/PLCAPI/"
+def parse_bool(option, opt_str, value, parser):
+	if opt_str in ["--debug"]:
+		parser.values.debug = int(int(value))
+	elif opt_str in ["--mail"]:
+		parser.values.mail = int(int(value))
+	elif opt_str in ["--bcc"]:
+		parser.values.bcc = int(int(value))
+	elif opt_str in ["--policysavedb"]:
+		parser.values.policysavedb = int(int(value))
+	elif opt_str in ["--squeeze"]:
+		parser.values.squeeze = int(int(value))
+	else:
+		print "blue"
 
+class config:
+	debug=0
+	mail=0
+	bcc=0
+	email="soltesz@cs.utk.edu"
+	run=False
+	checkopt=False
+	squeeze=0
+	policysavedb=0
+	__file = ".config"
 
-#incomplete
+	def __init__(self, parser=None):
+		if os.path.exists(self.__file): # file exists, read that.
+			f = open(self.__file, 'r')
+			o = pickle.load(f)
+			self.__dict__.update(o)
+			f.close()
+
+		if parser == None:
+			self.parser = OptionParser()
+		else:
+			self.parser = parser
+
+		self.parser.set_defaults(debug = self.debug,
+								mail = self.mail,
+								bcc  = self.bcc,
+								email = self.email,
+								run = self.run,
+								checkopt = False,
+								squeeze = self.squeeze,
+								policysavedb = self.policysavedb)
+
+		self.parser.add_option("", "--debug", dest="debug",
+						  help="Enable debugging", 
+						  type="int",
+						  metavar="[0|1]",
+						  action="callback", 
+						  callback=parse_bool)
+		self.parser.add_option("", "--mail", dest="mail",
+						  help="Enable sending email",
+						  type="int",
+						  metavar="[0|1]",
+						  action="callback", 
+						  callback=parse_bool)
+		self.parser.add_option("", "--bcc", dest="bcc",
+						  help="Include BCC to user",
+						  type="int",
+						  metavar="[0|1]",
+						  action="callback", 
+						  callback=parse_bool)
+		self.parser.add_option("", "--squeeze", dest="squeeze",
+						  help="Squeeze sites or not",
+						  type="int",
+						  metavar="[0|1]",
+						  action="callback", 
+						  callback=parse_bool)
+		self.parser.add_option("", "--policysavedb", dest="policysavedb",
+						  help="Save the policy event database after a run",
+						  type="int",
+						  metavar="[0|1]",
+						  action="callback", 
+						  callback=parse_bool)
+		self.parser.add_option("", "--checkopt", dest="checkopt", 
+						  action="store_true",
+						  help="print current options")
+		self.parser.add_option("", "--run", dest="run", 
+						  action="store_true",
+						  help="Perform monitor or print configs")
+		self.parser.add_option("", "--email", dest="email",
+						  help="Specify an email address to use for mail when "+\
+								 "debug is enabled or for bcc when it is not")
+		
+		# config_command is needed to keep subsequent loads of config() from
+		# trying to parse the arguments that have already been parsed by
+		# the new main().
+		if parser == None and config_command:
+			print "calling parse_args"
+			self.parse_args()
+
+	def parse_args(self):
+		#print "self: %s" % self
+		#import traceback
+		#print traceback.print_stack()
+		#print "Ccalling parse_args"
+		(options, args) = self.parser.parse_args()
+		#for o in options.__dict__:
+		#	print "optin: %s == %s" % (o, options.__dict__[o])
+		self.__dict__.update(options.__dict__)
+		self.__dict__['args'] = args
+		self.save(options)
+		if options.checkopt:
+			self.usage()
+		#	print "\nAdd --run to actually perform the command"
+			sys.exit(1)
+
+	def getListFromFile(self, file):
+		f = open(file, 'r')
+		list = []
+		for line in f:
+			line = line.strip()
+			list += [line]
+		return list
+
+	def print_values(self):
+		exclude = ['parser']
+		for key in self.__dict__.keys():
+			if key not in exclude:
+				print "%20s == %s" % (key, self.__dict__[key])
+		
+	def save(self, options=None):
+		f = open(self.__file, 'w')
+		if options == None:
+			o = {'debug': self.debug, 
+				 'mail': self.mail, 
+				 'bcc': self.bcc, 
+				 'email':self.email,
+				 'squeeze':self.squeeze,
+				 'policysavedb':self.policysavedb}
+		else:
+			o = options.__dict__
+
+		pickle.dump(o, f)
+		f.close()
+
+	def usage(self):
+		self.print_values()
+		self.parser.print_help()
+
+
+def main():
+	""" Start threads, do some housekeeping, then daemonize. """
+	# Defaults
+	global config_command
+	config_command = True
+	config = __main__.config()
+
+	try:
+		print "acalling parse_args"
+		config.parse_args()
+		
+	except Exception, err:
+		print "Error: %s " %  err
+		config.usage()
+		sys.exit(1)
+
+	config.usage()
+
+
+if __name__ == '__main__':
+	main()