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."
