Как передать модель пользователя в поле формы (django)?

По сути, мне нужно использовать хэш пароля пользователя для шифрования некоторых данных через настраиваемое поле модели. Посмотрите фрагмент, который я использовал здесь: Django Encryption.

Я пробовал это:

class MyClass(models.Model):
    owner = models.ForeignKey(User)
    product_id = EncryptedCharField(max_length=255, user_field=owner)

.................................................................................

    def formfield(self, **kwargs):
        defaults = {'max_length': self.max_length, 'user_field': self.user_field}
        defaults.update(kwargs)
        return super(EncryptedCharField, self).formfield(**defaults))

Но когда я пытаюсь использовать user_field, я получаю экземпляр ForeignKey (конечно!):

user_field = kwargs.get('user_field')
cipher = user_field.password[:32]

Любая помощь приветствуется!


django encryption field formfield
person Bryan    schedule 16.11.2009    source источник

Ответы (1)


arrow_upward
1
arrow_downward

может быть что-то вроде этого - переопределить метод save (), в котором вы можете вызвать метод шифрования.

для расшифровки вы можете использовать signal post_init, поэтому каждый раз, когда вы создаете экземпляр модели из базы данных поле product_id расшифровывается автоматически

class MyClass(models.Model):
    user_field = models.ForeignKey(User)
    product_id = EncryptedCharField()
    ...other fields...

    def save(self):
        self.product_id._encrypt(product_id, self.user_field)
        super(MyClass,self).save()

    def decrypt(self):
        if self.product_id != None:
            user = self.user_field
            self.product_id._decrypt(user=user)

def post_init_handler(sender_class, model_instance):
    if isinstance(model_instance, MyClass):
        model_instance.decrypt()

from django.core.signals import post_init
post_init_connect.connect(post_init_handler)


obj = MyClass(user_field=request.user) 
#post_init will be fired but your decrypt method will have
#nothing to decrypt, so it won't garble your input
#you'll either have to remember not to pass value of crypted fields 
#with the constructor, or enforce it with either pre_init method 
#or carefully overriding __init__() method - 
#which is not recommended officially

#decrypt will do real decryption work when you load object form the database

obj.product_id = 'blah'
obj.save() #field will be encrypted

возможно, есть более элегантный "питонический" способ сделать это

person Evgeny    schedule 16.11.2009
comment
Прежде всего, спасибо за ответ! Надеюсь, я смогу придумать что-то похожее на это. Но есть ли у вас какой-нибудь простой пример того, как это можно сделать с помощью сигнала? У меня очень ограниченные знания о сигналах в целом ... - person Bryan; 16.11.2009
comment
БЛАГОДАРНОСТЬ! Вау, я действительно впечатлен. Я обязательно расскажу об этом позже. Спасибо еще раз! - person Bryan; 16.11.2009