CSCMonitorModule/data/buildHistoTypes.py

0001 from Cheetah.Template import Template
0002 from getopt import getopt
0003 from sys import argv, exit
0004
0005 def usage():
0006   print("Usage: " + argv[0] + " -d histo_map_file -t template_file")
0007   print("  Arguments:")
0008   print("    -d, --data=    : file of histogram map")
0009   print("    -t, --tmpl=    : template file")
0010
0011
0012 def main():
0013
0014   data = None
0015   tmpl = None
0016
0017   try:
0018     opts, args = getopt(argv[1:], "d:t:", ["data=","tmpl="])
0019   except getopt.GetoptError as err:
0020     print(str(err))
0021     usage()
0022     exit(1)
0023
0024   for o, a in opts:
0025     if o in ("-h", "--help"):
0026       usage()
0027       exit()
0028     elif o in ("-d", "--data"):
0029       data = a
0030     elif o in ("-t", "--tmpl"):
0031       tmpl = a
0032     else:
0033       print("unhandled option: " + o)
0034       usage()
0035       exit(2)
0036
0037   if data == None or tmpl == None:
0038     usage()
0039     exit(3)
0040
0041   # Taking data into hashmap
0042   f=open(data,'r')
0043   map = {}
0044   for l in f:
0045     a = l.strip().split()
0046     value = 0
0047     if len(a) > 1:
0048       value = a[1].strip()
0049     map[a[0].strip()] = value
0050   f.close()
0051
0052   # Process stuff
0053   t = Template(file=tmpl)
0054   t.datamap = map
0055   print(t)
0056
0057 if __name__ == "__main__":
0058   main()
0059