Update manager mails and ceo mail

1 files changed, 15 insertions, 0 deletions

diff --git a/maconomy/utils.py b/maconomy/utils.py
new file mode 100644
index 0000000..e22239b
--- /dev/null
+++ b/maconomy/utils.py
@@ -0,0 +1,15 @@
+from collections import defaultdict
+
+def per_manager(timesheets):
+    per_manager = defaultdict(list)
+    # filter timesheets per manager
+    for timesheet in [t for t in timesheets if need_manager_mail(t)]:
+        manager_id = timesheet.approver
+        per_manager[manager_id].append(timesheet)
+    return per_manager
+
+def need_manager_mail(timesheet):
+    return not timesheet.is_submitted() or not timesheet.is_approved()
+
+def employees(timesheets):
+    return dict([(t.employee.id, t.employee) for t in timesheets])