summaryrefslogtreecommitdiff
path: root/service/app/tags.py
blob: 39709a4c7d0c50616d4bd956a05e02fedc79421f (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
class Tag:

    def __init__(self, name):
        self.name = name

    def __eq__(self, other):
        return self.name == other.name

    def __hash__(self):
        return self.name.__hash__()


class Tags:

    SPECIAL_TAGS = ['inbox', 'sent', 'drafts', 'trash']

    def __init__(self):
        self.tags = set([Tag(name) for name in self.SPECIAL_TAGS])

    def add(self, name):
        self.tags.add(Tag(name))

    def find(self, name):
        for tag in self.tags:
            if tag.name == name:
                return tag

    def __len__(self):
        return len(self.tags)

    def __iter__(self):
        return self.tags.__iter__()