Fix du AESEncryptedField

f02e6c6e · Hugo LEVY-FALK · 9cae5822 · f02e6c6e
Commit f02e6c6e authored Jun 17, 2018 by Hugo LEVY-FALK
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 0 deletions

preferences/aes_field.py preferences/aes_field.py +6 -0

No files found.
--- a/preferences/aes_field.py
+++ b/preferences/aes_field.py
@@ -79,6 +79,12 @@ class AESEncryptedField(models.CharField):
        return decrypt(settings.AES_KEY,
                       binascii.a2b_base64(value)).decode('utf-8')

+    def from_db_value(self, value, *args, **kwargs):
+        if value is None:
+            return value
+        return decrypt(settings.AES_KEY,
+                       binascii.a2b_base64(value)).decode('utf-8')
+
    def get_prep_value(self, value):
        if value is None:
            return value