def main(): try: try: opts, args = getopt.getopt(sys.argv[1:], "p:g:m:i:vh",\ ["periods=", "groups=", "html=", "ics=", "verbose",\ "list-formations", "list-periods", "list-rooms", "codename-sort", "output-formation-name", "output-period-name", "max-retries", "retry-delay", "help"]) except getopt.GetoptError, err: print >> sys.stderr, str(err) usage() sys.exit(2) sort_func = lambda (a,b), (c,d): cmp(b.lower(), d.lower()) output_formation_name = 0 output_period_name = 0 if ('-h', '') in opts or ('--help', '') in opts: usage() sys.exit() if ("--codename-sort", "") in opts: sort_func = lambda (a,b), (c,d): cmp(a.lower(), c.lower()) if ("--output-formation-name", "") in opts: output_formation_name = 1 if ("--output-periods-name", "") in opts: output_period_name = 1 if ("--list-formations", "") in opts: formations_list = UMonsTimesheetFetcher( max_retries, retry_delay).formations.items() formations_list.sort(sort_func) for fcodename, fname in formations_list: print "%s - %s" % (fname, fcodename) sys.exit(0) if ("--list-periods", "") in opts: periods_list = UMonsTimesheetFetcher( max_retries, retry_delay).periods.items() periods_list.sort(sort_func) for pcodename, pname in periods_list: print "%s - %s" % (pname, pcodename) sys.exit(0) if ("--list-rooms", "") in opts: rooms_list = UMonsTimesheetFetcher( max_retries, retry_delay).rooms.items() rooms_list.sort(sort_func) for rcodename, rname in rooms_list: print "%s - %s" % (rname, rcodename) sys.exit(0) if len(args) == 0: print >> sys.stderr, 'ERROR: Missing formation parameter.' print >> sys.stderr, ' %s -h or --help to get help.' %\ sys.argv[0] sys.exit(2) formations = map(decode_input, args) periods = [] groups = [] output_funcs = [] verb = False opts = map(lambda (x, y): (decode_input(x), decode_input(y)), opts) for o, a in opts: if o in ("-p", "--periods"): periods.extend(a.split(",")) elif o in ("-g", "--groups"): groups.extend(a.split(",")) elif o in ("-m", "--html"): if not os.path.isdir(a): print u"\"%s\" not a directory, creating." % a try: os.makedirs(a) except OSError, err: print >> sys.stderr, err sys.exit(2) output_funcs.append(partial(output_html, dir=a, output_formation_name=output_formation_name, output_period_name=output_period_name)) elif o in ("-i", "--ics"): print >> sys.stderr, "iCalendar format output",\ "is not yet implemented. Option ignored."