У меня есть собственный EncryptedCharField, который я хочу в основном отображать как CharField при взаимодействии с пользовательским интерфейсом, но перед сохранением/извлечением в БД он шифрует/расшифровывает его.
документация по пользовательским полям сообщает:
- добавить
__metaclass__ = models.SubfieldBase - переопределить to_python для преобразования данных из него необработанного хранилища в желаемый формат
- переопределить get_prep_value для преобразования значения перед сохранением db.
Итак, вы думаете, что этого будет достаточно просто - для 2. просто расшифруйте значение и 3. просто зашифруйте его.
Основываясь на фрагменте django, и в документации это поле выглядит следующим образом:
class EncryptedCharField(models.CharField):
"""Just like a char field, but encrypts the value before it enters the database, and decrypts it when it
retrieves it"""
__metaclass__ = models.SubfieldBase
def __init__(self, *args, **kwargs):
super(EncryptedCharField, self).__init__(*args, **kwargs)
cipher_type = kwargs.pop('cipher', 'AES')
self.encryptor = Encryptor(cipher_type)
def get_prep_value(self, value):
return encrypt_if_not_encrypted(value, self.encryptor)
def to_python(self, value):
return decrypt_if_not_decrypted(value, self.encryptor)
def encrypt_if_not_encrypted(value, encryptor):
if isinstance(value, EncryptedString):
return value
else:
encrypted = encryptor.encrypt(value)
return EncryptedString(encrypted)
def decrypt_if_not_decrypted(value, encryptor):
if isinstance(value, DecryptedString):
return value
else:
encrypted = encryptor.decrypt(value)
return DecryptedString(encrypted)
class EncryptedString(str):
pass
class DecryptedString(str):
pass
и Encryptor выглядит так:
class Encryptor(object):
def __init__(self, cipher_type):
imp = __import__('Crypto.Cipher', globals(), locals(), [cipher_type], -1)
self.cipher = getattr(imp, cipher_type).new(settings.SECRET_KEY[:32])
def decrypt(self, value):
#values should always be encrypted no matter what!
#raise an error if tthings may have been tampered with
return self.cipher.decrypt(binascii.a2b_hex(str(value))).split('\0')[0]
def encrypt(self, value):
if value is not None and not isinstance(value, EncryptedString):
padding = self.cipher.block_size - len(value) % self.cipher.block_size
if padding and padding < self.cipher.block_size:
value += "\0" + ''.join([random.choice(string.printable) for index in range(padding-1)])
value = EncryptedString(binascii.b2a_hex(self.cipher.encrypt(value)))
return value
При сохранении модели возникает ошибка строки "Нечетная длина" в результате попытки расшифровать уже расшифрованную строку. При отладке он выглядит как to_python, который вызывается дважды, первый с зашифрованным значением, а второй раз с расшифрованным значением, но не на самом деле как тип Decrypted, а как необработанная строка, вызывающая ошибку. Кроме того, get_prep_value никогда не вызывается.
Что я делаю неправильно?
Это не должно быть так сложно: кто-нибудь еще считает, что этот код поля Django очень плохо написан, особенно когда речь заходит о настраиваемых полях, а не о расширяемости? Простые переопределяемые методы pre_save и post_fetch легко решают эту проблему.