summaryrefslogtreecommitdiff
path: root/maconomy/templates.py
blob: 07cbeb11d2cd7e41eec678db9fa68ec927a7ab04 (plain)
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
from string import Template

class BaseTemplate(object):
    def __init__(self, template_file):
        self.set_template(template_file)

    def build(self, **kwargs):
        return self.template.substitute(**kwargs)

    def set_template(self, template_file):
        with open(template_file, 'r') as f:
            template_base = f.read()
        self.template = Template(template_base)


class UnsubmittedEmailTemplate(BaseTemplate):
    def __init__(self):
        self.set_template("templates/unsubmitted.html")

    def build(self, week, maconomyurl, helpurl):
        return self.template.substitute(week=week, maconomyurl=maconomyurl, helpurl=helpurl)

class MissingEmailTemplate(BaseTemplate):
    def __init__(self):
        self.set_template("templates/missing.html")

    def build(self, maconomyurl, helpurl):
        return self.template.substitute(maconomyurl=maconomyurl, helpurl=helpurl)

class ManagerEmailTemplate(BaseTemplate):
    def __init__(self):
        self.set_template("templates/manager.html")

    def build(self, employee, week, maconomyurl, **kwargs):
        submitted = '' if 'submitted' in kwargs and kwargs['submitted'] else 'not'
        approved = '' if 'approved' in kwargs and kwargs['approved'] else 'not'
        return self.template.substitute(employee=employee.__unicode__(),week=week,maconomyurl=maconomyurl, submitted=submitted, approved=approved)