rezometz/re2o
8
0
Fork 0
mirror of https://gitlab2.federez.net/re2o/re2o synced 2024-11-12 12:56:26 +00:00
Code Issues Releases Wiki Activity

Fix error message when failing to decode an aes_key.

Browse source
This commit is contained in:
Hugo LEVY-FALK 2019-01-11 14:02:32 +01:00 committed by chirac
parent 88416a28b4
commit 2ea0d77eb2
1 changed files with 10 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
re2o/aes_field.py
View file

@ -82,16 +82,22 @@ class AESEncryptedField(models.CharField):
return None return None
try: try:
return decrypt(settings.AES_KEY, binascii.a2b_base64(value)).decode('utf-8') return decrypt(settings.AES_KEY, binascii.a2b_base64(value)).decode('utf-8')
except Exception as e: except UnicodeDecodeError as e:
raise ValueError(value) raise ValueError(
"Could not decode your field %s, your settings.AES_KEY "
"is probably wrong." % self.name
)
def from_db_value(self, value, *args, **kwargs): def from_db_value(self, value, *args, **kwargs):
if value is None: if value is None:
return value return value
try: try:
return decrypt(settings.AES_KEY, binascii.a2b_base64(value)).decode('utf-8') return decrypt(settings.AES_KEY, binascii.a2b_base64(value)).decode('utf-8')
except Exception as e: except UnicodeDecodeError as e:
raise ValueError(value) raise ValueError(
"Could not decode your field %s, your settings.AES_KEY "
"is probably wrong." % self.name
)
def get_prep_value(self, value): def get_prep_value(self, value):
if value is None: if value is None: