diff options
Diffstat (limited to 'py-fake-service/app/adapter/tagsset.py')
-rw-r--r-- | py-fake-service/app/adapter/tagsset.py | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/py-fake-service/app/adapter/tagsset.py b/py-fake-service/app/adapter/tagsset.py new file mode 100644 index 00000000..23941735 --- /dev/null +++ b/py-fake-service/app/adapter/tagsset.py @@ -0,0 +1,16 @@ +from tag import Tag + +class TagsSet: + def __init__(self): + self.tags = {} + self.ident = 0 + + def add(self, mbox_mail): + tags = mbox_mail.get('X-TW-Pixelated-Tags').split(', ') + for tag in tags: + tag = self.tags.setdefault(tag, Tag(tag, self.ident)) + tag.increment_count() + self.ident += 1 + + def all_tags(self): + return self.tags.values() |