使用 Django 自定义 Field 加密数据

作者：赖铭
导语：出于安全考虑，一些敏感信息是需要加密存入数据库的。这样即使被黑客脱库，也能最大限度的避免损失。

数据加密并不是难事，各种现成的加密库可以轻松实现高强度的加密。对于加密标准的选择，根据实际场景需要，在《IDC开发运维安全标准》 4.1算法标准中选择合适算法即可。

在实际开发中，我是在业务现成逻辑代码已经大体完成时接手，被要求将部分字段加密处理的。因此，我希望加密逻辑对业务逻辑完全不可见，做到不需要修改任何业务逻辑代码，即可接入数据加密。

要做到对业务逻辑不可见，就需要在字段入库和出库时，自动进行加密和解密。Django的自定义Field提供的一些接口，可以在数据出库后、数据入库前，做一些特定的操作。如：DatetimeField，就是使用了这些接口，做到了可以直接使用python的datetime进行操作，免去了与数据库类型转换的工作。

下面这个例子，选择的是AES对称加密算法，对CharField进行加密处理。

代码语言：javascript复制

class AESCharField(models.CharField):
"""
    在数据库中AES256加密的 CharField
    - 兼容未加密数据，加密后字符串会带上
    """

    def __init__(self, *args, **kwargs):
"""
        初始化
        :param prefix: 加密串前缀
        """
        if 'prefix' in kwargs:
self.prefix = kwargs['prefix']
del kwargs['prefix']
else:
self.prefix = "aes_str:::"

        self.cipher = AESCipher(settings.SECRET_KEY)
super(AESCharField, self).__init__(*args, **kwargs)

def deconstruct(self):
        name, path, args, kwargs = super(AESCharField, self).deconstruct()
if self.prefix != "aes_str:::":
            kwargs['prefix'] = self.prefix
return name, path, args, kwargs

def from_db_value(self, value, expression, connection, context):
"""
        出库后解密数据
        """
        if value is None:
return value
if value.startswith(self.prefix):
            value = value[len(self.prefix):]
            value = self.cipher.decrypt(value)

return value

def to_python(self, value):
"""
        反序列化和Form clean()时调用，解密数据
        """
        if value is None:
return value
elif value.startswith(self.prefix):
            value = value[len(self.prefix):]
            value = self.cipher.decrypt(value)

return value

def get_prep_value(self, value):
"""
        入库前加密数据
        """
        if isinstance(value, str) or isinstance(value, unicode):
            value = self.cipher.encrypt(value)
            value = self.prefix value
elif value is not None:
raise TypeError(str(value) " is not a valid value for AESCharField")

return value

init中使用秘钥生成加密器，用于后面的加密，这里的秘钥是写在django settings中的。
prefix是标志加密后字符串的前缀，用于区分加密前和加密后数据。为了兼容已有旧数据的情况。
from_db_value是数据库返回值后，调用的函数。在此函数中，进行数据解密，需要处理None的情况。
get_prep_value是数据入库前调用的函数。在此函数中，加密数据，加上前缀。需要判断None的情况。
to_python是需要在反序列化和Form表单clean()时使用的，可能遇上加密后数据，也可能遇上未加密数据和None，需要注意判断。
这里兼容了存在旧数据的情况，因此需要使用startwith判断前缀是否匹配。

WARNING：

AES加密算法，如果设置随机的向量iv，哪怕加密相同的明文，每次加密产生的密文也是不相同的，因此无法对加密后字段进行数据库查询操作，业务代码需要注意。如果需要实现数据库查询，需要固定向量iv，并且实现get_prep_lookup方法。
这里的写法兼容了可能存在的旧数据，如果不对这些旧数据进行操作，这些旧数据将保持明文状态。如果需要加密旧数据，只需要进行一次取值存入操作，即可将旧数据加密。
注意max_length的设置。加密后的密文长度，会与明文有所出入，需要根据所选加密算法和前缀进行计算。该例子只提供了基本的加解密功能，展示了如何编写Django的自定义Model Field，用途不仅仅是加密。如果需要更加复杂的功能，参照官方文档：https://docs.djangoproject.com/en/1.11/howto/custom-model-fields/

下面附上可以直接拿来使用的代码

使用说明

安装依赖six,pycrypto
将CharField直接换成AESCharField。
密文比明文长，注意修改max_length。AESCipher的predict_length(length)可以计算加密后的长度，最后要加上密文前缀的长度，默认前缀aes_str:::长度为10。即：AESCipher.predict_length(old_max_length) len(prefix)
密文前缀默认为aes_str:::，可以通过添加参数prefix修改设置。class TestModel(models.Model): field1 = AESCharField("field1", max_length=100, prefix="orz:::")
使用的是Django的SECRET_KEY作为密钥，做了一次sha256。
由于兼容明文旧记录，旧记录至少需要一次保存操作，才能将明文替换为密文。如果需要立刻加密，可以通过脚本逐条保存一遍，注意避免与用户并发操作导致数据一致性问题。
使用随机向量iv的AES加密算法，加密后的数据无法进行查询，保证业务中没有以字段作为查询条件的语句，如果需要实现数据库查询，需要固定向量iv，并且实现get_prep_lookup方法。。
老版本Django使用south做同步，需要配置AESCharField的路径。具体可以参考south文档https://south.readthedocs.io/en/latest/customfields.html#custom-fieldsfrom south.modelsinspector import add_introspection_rules add_introspection_rules([], ["^mywebsite1_3.aes.AESCharField"]) # 示例

django

0 人点赞