From 0d5f395c0004129079924f45a65ae9a218980359 Mon Sep 17 00:00:00 2001 From: Manuel Gomez <34234948+lologf@users.noreply.github.com> Date: Fri, 12 Apr 2019 10:30:02 +0200 Subject: [PATCH] Fix decrypt_uri() and _protocol_info() methods --- django_mailbox/models.py | 16 +++++++++------- 1 file changed, 9 insertions(+), 7 deletions(-) diff --git a/django_mailbox/models.py b/django_mailbox/models.py index 9d60a78..650d523 100644 --- a/django_mailbox/models.py +++ b/django_mailbox/models.py @@ -133,18 +133,20 @@ class Mailbox(models.Model): def decrypt_uri(self): secret_key = self.pad(django_settings.SECRET_KEY) - self.uri = self.pad(self.uri) + uri = self.uri + uri = self.pad(uri) + uri = base64.b64decode(uri) cipher = AES.new(secret_key) - self.uri = base64.b64decode(self.uri) - self.uri = cipher.decrypt(self.uri) + uri = cipher.decrypt(uri) + + return self.unpad(uri) - return self.unpad(self.uri) - @property def _protocol_info(self): - self.uri = self.decrypt_uri() - return urlparse(self.uri) + uri = self.uri + uri = self.decrypt_uri() + return urlparse(uri) @property def _query_string(self):