def test_get_holidays(self):
holidays = team.get_holidays(self.testfile)
self.assertIsNotNone(holidays)
self.assertEqual(len(holidays), 2)
def parse_arguments():
parser = argparse.ArgumentParser()
parser.add_argument("year", type=int, help="year e.g. 2016")
parser.add_argument("month", type=int, help="month e.g. 03")
parser.add_argument("inputfile", help="location to file with configuration")
return parser.parse_args()
# PROGRAM FLOW
args = parse_arguments()
logging.basicConfig(level=logging.DEBUG)
workdate = date(args.year, args.month, 1)
holidays = team.get_holidays(args.inputfile)
# team = member.init_team(workdate)
myteam = team.get_team(args.inputfile)
schedule = create_schedule(myteam, holidays, workdate)
print("\nSchedule for %s\n" % str(workdate))
print("Assignment per team member\n")
for m in myteam:
print("%s %d" % (m.name, m.count))
print("\n\nSchedule\n")
for day in schedule:
print("%s %s" % (day, schedule[day]))
