Django模型可以使用MySQL函数吗？

Question

有没有办法强制Django模型在每次读取或加载模型数据时将字段传递给MySQL函数？ 为了澄清我在SQL中的含义，我希望Django模型能够产生如下内容：

在模型加载时：SELECT AES_DECRYPT（fieldname，password）FROM tablename

在模型保存上：INSERT INTO tablename VALUES（AES_ENCRYPT（userinput，password））

Answer 1

您可以在模型上创建属性，而不是模型加载，当访问属性时，它可以读取数据库：

def _get_foobar(self):
    if not hasattr(self, '_foobar'):

        cursor = connection.cursor()
        self._foobar = cursor.execute('SELECT AES_DECRYPT(fieldname, password) FROM tablename')[0]
    return self._foobar
foobar = property(_get_foobar)

加载后，您可以参考mything.foobar ，第一次访问将从数据库中检索解密，并保留以供以后访问。

这也有一个好处，如果你的一些代码没有用于解密，它就不会发生。

Answer 2

我会为您想要加密/解密的列定义自定义modelfield 。 加载to_python方法以在加载模型时运行解密，并使用get_db_prep_value在保存时运行加密。

请记住将字段的元类设置为models.SubfieldBase否则将不会调用这些方法。

Answer 3

这是一个有效的解决方案，部分基于（ http://www.djangosnippets.org/snippets/824/ ）：

class Employee(models.Model):
   social_security_number = models.CharField(max_length=32)

   def _get_ssn(self):
       cursor = connection.cursor()
       cursor.execute("SELECT AES_DECRYPT(UNHEX(social_security_number), %s) as ssn FROM tablename WHERE id=%s", [settings.SECRET_KEY, self.id])
       return cursor.fetchone()[0]

   def _set_ssn(self, ssn_value):
       cursor = connection.cursor()
       cursor.execute("SELECT HEX(AES_ENCRYPT(%s, %s)) as ssn", [ssn_value, settings.SECRET_KEY])
       self.social_security_number = cursor.fetchone()[0]

   ssn = property(_get_ssn, _set_ssn)

结果如下：

>>> from foo.bar.models import Employee
>>> p=Employee.objects.create(ssn='123-45-6789')
>>> p.ssn
'123-45-6789'

mysql> select * from foo_employee;
+----+----------------------------------+
| id | social_security_number           |
+----+----------------------------------+
| 31 | 41DF2D946C9186BEF77DD3307B85CC8C |
+----+----------------------------------+
1 row in set (0.00 sec)

Answer 4

这绝对是hackish，但似乎Django现在不会让你以任何其他方式做到这一点。 值得注意的是，除了首次加载时，每次更改python中的值时都会调用to_python。

from django.db import connection, models
import re

class EncryptedField(models.TextField):
    __metaclass__ = models.SubfieldBase

    def to_python(self, value):
        if not re.match('^*some pattern here*$', value):
            cursor = connection.cursor()
            cursor.execute('SELECT AES_DECRYPT(%s, %s)', [value, settings.SECRET_KEY])
            return cursor.fetchone()[0]
        return value

    def get_db_prep_value(self, value):
        cursor = connection.cursor()
        cursor.execute('SELECT AES_ENCRYPT(%s, %s)', [value, settings.SECRET_KEY])
        return cursor.fetchone()[0]


class Encrypt(models.Model):
    encrypted = EncryptedField(max_length = 32)

Answer 5

在深入搜索Django ORM的实现后，

我发现它可以通过以下方式解决：

class EncryptedField(models.BinaryField):
    @staticmethod
    def _pad(value):
        return value + (AES.block_size - len(value) % AES.block_size) * b('\x00')

    def _encrypt(self, data):
        if not data:
            return None
        return self.cipher.encrypt(self._pad(data.encode('utf8')))

    def _decrypt(self, data):
        if not data:
            return None
        return self.cipher.decrypt(force_bytes(data)).rstrip(b'\x00').decode('utf8')

    @property
    def cipher(self):
        return AES.new(KEY, mode=AES.MODE_CBC, IV=self._iv)

    def get_db_prep_value(self, value, connection, prepared=False):
        if value is not None:
            value = self._encrypt(value)
            if value:
                value = binascii.hexlify(value)
        return value

    def get_placeholder(self, value, compiler, connection):
        return 'unhex(%s)'

Answer 6

使用Django信号，您可以在保存模型实例时执行操作，但据我所知，您无法在读取时触发任何内容。

编辑：我的不好，似乎你可以在初始化模型实例时做一些事情。