import os.path

from timesheets.timesheet import Timesheet

parser = argparse.ArgumentParser(description="Convert timesheets to CSV")
parser.add_argument("--timesheet", type=str, required=True, help="timesheet file name")
args = parser.parse_args()
timesheet_yml_fname = args.timesheet

yml_fh = open(timesheet_yml_fname)
timesheet_items = yaml.load_all(yml_fh)

timesheet = Timesheet()
total_hours = 0.0
for item in timesheet_items:
    timecard = timesheet.get_timecard(item.pop("date"))
    timecard.add(**item)

timesheet_csv = timesheet.to_csv()
timesheet_basename = os.path.basename(timesheet_yml_fname)
timesheet_stem = os.path.splitext(timesheet_basename)[0]
timesheet_csv_fname = timesheet_stem + ".csv"

csv_fh = open(timesheet_csv_fname, "w")
csv_fh.write(timesheet_csv)
csv_fh.close()

print "Total hours: {0:.2f}".format(timesheet.total_hours_worked())

# vim: expandtab shiftwidth=4 softtabstop=4