From 8687f923ec5f159970276d56b19183b2219827bd Mon Sep 17 00:00:00 2001 From: Patrick Maia Date: Thu, 14 Aug 2014 14:15:13 -0300 Subject: refactored update tags logic --- service/app/adapter/mail_service.py | 3 ++- service/app/pixelated_user_agent.py | 5 ++--- service/test/adapter/mail_service_test.py | 2 +- 3 files changed, 5 insertions(+), 5 deletions(-) (limited to 'service') diff --git a/service/app/adapter/mail_service.py b/service/app/adapter/mail_service.py index 73baef56..1025bce8 100644 --- a/service/app/adapter/mail_service.py +++ b/service/app/adapter/mail_service.py @@ -54,7 +54,8 @@ class MailService: def mails(self, query): return self.mailbox.messages or [] - def update_tags(self, mail, new_tags): + def update_tags(self, mail_id, new_tags): + mail = self.mail(mail_id) new_tags = mail.update_tags(new_tags) self._update_tag_list(new_tags) return mail diff --git a/service/app/pixelated_user_agent.py b/service/app/pixelated_user_agent.py index 3315999b..739cb57f 100644 --- a/service/app/pixelated_user_agent.py +++ b/service/app/pixelated_user_agent.py @@ -100,9 +100,8 @@ def mail(mail_id): @app.route('/mail//tags', methods=['POST']) def mail_tags(mail_id): new_tags = request.get_json()['newtags'] - mail = mail_service.mail(mail_id) - mail_service.update_tags(mail, new_tags) - tag_names = [tag.name for tag in mail.tags] + tags = mail_service.update_tags(mail_id, new_tags) + tag_names = [tag.name for tag in tags] return respond_json(tag_names) diff --git a/service/test/adapter/mail_service_test.py b/service/test/adapter/mail_service_test.py index 7f655147..7b70ac9b 100644 --- a/service/test/adapter/mail_service_test.py +++ b/service/test/adapter/mail_service_test.py @@ -33,7 +33,7 @@ class TestMailService(unittest.TestCase): mailservice = MailService() new_tags = ['test'] - mails = mailservice.update_tags(mailservice.mail(6), new_tags) + mails = mailservice.update_tags(6, new_tags) for tag in mailservice.all_tags(): print tag.name -- cgit v1.2.3