From 9779abfb1b1d0513e5efbc2737395bba0e8b6235 Mon Sep 17 00:00:00 2001 From: Duda Dornelles Date: Tue, 12 Aug 2014 20:29:33 -0300 Subject: creating new tag if tag doesnt exist in py-fake-service --- py-fake-service/app/adapter/tagsset.py | 17 +++++++++++++---- 1 file changed, 13 insertions(+), 4 deletions(-) diff --git a/py-fake-service/app/adapter/tagsset.py b/py-fake-service/app/adapter/tagsset.py index abe7ef7e..803c921c 100644 --- a/py-fake-service/app/adapter/tagsset.py +++ b/py-fake-service/app/adapter/tagsset.py @@ -8,9 +8,8 @@ class TagsSet: 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 = self._create_new_tag(tag) tag.increment_count() - self.ident += 1 def all_tags(self): return self.tags.values() @@ -20,9 +19,19 @@ class TagsSet: tag = self.tags.get(tag) tag.increment_read() - def increment_tag_total_count(self, tag): - self.tags.get(tag).increment_count() + def increment_tag_total_count(self, tagname): + tag = self.tags.get(tagname) + if tag: + tag.increment_count() + else: + self._create_new_tag(tagname) def decrement_tag_total_count(self, tag): self.tags.get(tag).decrement_count() + def _create_new_tag(self, tag): + tag = self.tags.setdefault(tag, Tag(tag, self.ident)) + self.ident += 1 + return tag + + -- cgit v1.2.3