1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
|
from maconomy import cli, create_db, TimeRegistrationRepository
from maconomy.views import EmployeeEmailView, ManagerEmailView, CEOEmailView
def main(config, options):
db = create_db(config)
timereg_repo = TimeRegistrationRepository(db)
timesheets = timereg_repo.all_active()
if options.manager:
manager(timesheets, config, options.dry)
elif options.ceo:
ceo(timesheets, config, options.dry)
elif options.summary:
summary(timesheets)
else:
normal(timesheets, config, options.dry)
# Close stuff
timereg_repo.close()
db.close()
def normal(timesheets, config, dry_run):
view = EmployeeEmailView(config)
for timesheet in timesheets:
if not timesheet.is_submitted():
mail = view.render(timesheet)
subjct = u"Timesheet reminder for {}".format(timesheet.employee)
if dry_run:
print subjct
print mail
else:
pass
def manager(timesheets, config, dry_run):
relevant = [t for t in timesheets if need_manager_mail(t)]
view = ManagerEmailView(config)
mail = view.render(relevant)
if dry_run:
print mail
def ceo(timesheets, config, dry_run):
relevant = [t for t in timesheets if need_manager_mail(t)]
view = CEOEmailView()
mail = view.render(relevant)
if dry_run:
print mail
def need_manager_mail(timesheet):
return not timesheet.is_submitted() or not timesheet.is_approved()
def summary(timesheets):
for timesheet in timesheets:
if timesheet.is_missing():
print u"[Not created] {}".format(timesheet.employee)
elif not timesheet.is_submitted():
print u"[Unsubmitted] {}".format(timesheet.employee)
elif not timesheet.is_approved():
print u"[Not approved] {}".format(timesheet.employee)
if __name__ == '__main__':
args = cli.parse()
config = cli.load_config(args.config)
main(config, args)
|