def query_hours( timesheet_pathname, nr_hours_to_work, project_pattern, nr_weeks_to_report): selected_records = track_time.parse(file(timesheet_pathname, "r")) to_time_point = track_time.last_day_of_week(datetime.date.today()) from_time_point = to_time_point - datetime.timedelta( days=(nr_weeks_to_report * 7) - 1) assert from_time_point.isocalendar()[2] == 1 # Monday. assert to_time_point.isocalendar()[2] == 7 # Sunday. selected_records = track_time.filter_projects_by_name(selected_records, project_pattern) selected_records = track_time.filter_projects_by_date(selected_records, from_time_point, to_time_point) merged_records = track_time.merge_records_by_date(selected_records) merged_records = sorted(merged_records, key=lambda record: record.date) # Hours per day (work + sick + holiday + vacation). table = prettytable.PrettyTable(["Date", "Hours"]) table.align = "r" for record in merged_records: table.add_row([ "{} {}".format(record.date.strftime("%a"), record.date), "{:.2f}".format(record.nr_hours) ]) write_table(table) merged_records = track_time.merge_records_by_week(selected_records) merged_records = sorted(merged_records, key=lambda record: record.date) # Weekly balance. table = prettytable.PrettyTable(["Week", "Balance"]) table.align = "r" for record in merged_records: table.add_row([ "{} {}".format(record.date.strftime("%a"), record.date), "{:+.2f}".format(record.nr_hours - nr_hours_to_work) ]) write_table(table) # Balance of the whole period. if merged_records: table = prettytable.PrettyTable(["Period", "Balance"]) table.align = "r" record = track_time.merge_records(selected_records) record.date = merged_records[0].date table.add_row([ "{} {}".format(record.date.strftime("%a"), record.date), "{:+.2f}".format(record.nr_hours - (nr_weeks_to_report * nr_hours_to_work)) ]) write_table(table)
def test_merge_records_by_week(self): records = track_time.parse(file("sub_projects-001.txt")) merged_records = track_time.merge_records_by_week(records) self.assertEqual(len(merged_records), 5) self.assertEqual(merged_records[0].date, datetime.date(2013, 12, 30)) self.assertEqual(merged_records[1].date, datetime.date(2014, 1, 6)) self.assertEqual(merged_records[2].date, datetime.date(2014, 1, 27)) self.assertEqual(merged_records[3].date, datetime.date(2014, 2, 3)) self.assertEqual(merged_records[4].date, datetime.date(2014, 2, 10)) self.assertEqual(merged_records[0].nr_hours, 30) self.assertEqual(merged_records[1].nr_hours, 18) self.assertEqual(merged_records[2].nr_hours, 13) self.assertEqual(merged_records[3].nr_hours, 23) self.assertEqual(merged_records[4].nr_hours, 12)