From 6df9e2c2646975ffc7c312f675591870eb4b2ad8 Mon Sep 17 00:00:00 2001 From: Jefferson Stachelski Date: Fri, 25 Sep 2015 19:05:01 -0300 Subject: Issue #470 handling the update draft Deleting the new draft created if the old one was not found to fix the duplication drafts --- service/pixelated/adapter/errors/__init__.py | 3 +++ service/pixelated/adapter/services/draft_service.py | 13 +++++++++++-- service/pixelated/resources/mails_resource.py | 8 +++++++- 3 files changed, 21 insertions(+), 3 deletions(-) create mode 100644 service/pixelated/adapter/errors/__init__.py (limited to 'service') diff --git a/service/pixelated/adapter/errors/__init__.py b/service/pixelated/adapter/errors/__init__.py new file mode 100644 index 00000000..31ad4947 --- /dev/null +++ b/service/pixelated/adapter/errors/__init__.py @@ -0,0 +1,3 @@ +class DuplicatedDraftException(Exception): + def __init__(self, message): + super(Exception, self).__init__(message) diff --git a/service/pixelated/adapter/services/draft_service.py b/service/pixelated/adapter/services/draft_service.py index c3a14928..65794f04 100644 --- a/service/pixelated/adapter/services/draft_service.py +++ b/service/pixelated/adapter/services/draft_service.py @@ -14,6 +14,7 @@ # You should have received a copy of the GNU Affero General Public License # along with Pixelated. If not, see . from twisted.internet import defer +from pixelated.adapter.errors import DuplicatedDraftException class DraftService(object): @@ -32,8 +33,16 @@ class DraftService(object): @defer.inlineCallbacks def update_draft(self, ident, input_mail): new_draft = yield self.create_draft(input_mail) - yield self._mail_store.delete_mail(ident) - defer.returnValue(new_draft) + try: + yield self._mail_store.delete_mail(ident) + defer.returnValue(new_draft) + except Exception as error: + errorMessage = error.args[0].getErrorMessage() + + if errorMessage == 'Need to create doc before deleting': + yield self._mail_store.delete_mail(new_draft.ident) + raise DuplicatedDraftException(errorMessage) + # pixelated_mail = yield self.create_draft(input_mail) # yield (yield self._mailboxes.drafts).remove(ident) # defer.returnValue(pixelated_mail) diff --git a/service/pixelated/resources/mails_resource.py b/service/pixelated/resources/mails_resource.py index 75c0e14f..93a19a9b 100644 --- a/service/pixelated/resources/mails_resource.py +++ b/service/pixelated/resources/mails_resource.py @@ -166,11 +166,17 @@ class MailsResource(Resource): if draft_id: deferred_check = self._mail_service.mail_exists(draft_id) + def handleDuplicatedDraftException(error): + respond_json_deferred("", request, status_code=422) + def return422otherwise(mail_exists): if not mail_exists: respond_json_deferred("", request, status_code=422) else: - defer_response(self._draft_service.update_draft(draft_id, _mail)) + new_draft = self._draft_service.update_draft(draft_id, _mail) + new_draft.addErrback(handleDuplicatedDraftException) + defer_response(new_draft) + deferred_check.addCallback(return422otherwise) else: defer_response(self._draft_service.create_draft(_mail)) -- cgit v1.2.3