summaryrefslogtreecommitdiff
path: root/service/app/tags.py
diff options
context:
space:
mode:
Diffstat (limited to 'service/app/tags.py')
-rw-r--r--service/app/tags.py32
1 files changed, 32 insertions, 0 deletions
diff --git a/service/app/tags.py b/service/app/tags.py
new file mode 100644
index 00000000..39709a4c
--- /dev/null
+++ b/service/app/tags.py
@@ -0,0 +1,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__() \ No newline at end of file