【SQLAlchemy】基于RSA双向加密存储

用sqlalchemy的时候要实现密码的双向加密不方便调用mysql的ENCODE函数

这里给出一个基于rsa实现双向加密，在sqlalchemy定义model的时候用起来很方便，其他ORM应该也能采用相同的思路（话说我也不知道mysql是基于什么双向加密的）

加密解密函数

import rsa
def encrypt(msg, priv_str):
    if isinstance(msg, str):
        msg = msg.encode()
    if isinstance(priv_str, str):
        priv_str = priv_str.encode()
    privkey = rsa.PrivateKey.load_pkcs1(priv_str)
    return rsa.encrypt(msg, privkey)

def decrypt(msg, priv_str):
    if isinstance(msg, str):
        msg = msg.encode()
    if isinstance(priv_str, str):
        priv_str = priv_str.encode()
    privkey = rsa.PrivateKey.load_pkcs1(priv_str)
    return rsa.decrypt(msg, privkey).decode('utf-8')

有几个注意的地方：

• 参数是私钥字符串，因为直接从rem文件读取比较方便
• rsa加解密函数返回的都是bytes
• 解密decode了加密没有，是因为rsa加密后的bytes是没法decode到utf-8的，所以存储类型是Binary。这点在http发送签名的时候会出现坑，那就是另外的故事了

model定义

class User(Base):
    _pw = Column(Binary)

    @property
    def pw(self):
        with open('private.rem', 'r') as f:
            return decrypt(self._pw, f.read())
    @pw.setter
    def pw(self, pw):
        with open('private.rem', 'r') as f:
            self._pw = encrypt(pw, f.read())
    def pw_verify(self, pw):
        return pw == self.pw
# Usage
user = User(pw='password')
user.pw = 'password'
print(user.pw) # 'password'
print(user._pw) # b'eK:\x0f)\xa7}Gx\xe6\xfd\x14\xf3\xd8\xf1\x08f\xac\xdbL\xf23\x07#\...'

这里没啥说的，就是用了property装饰器。

有个问题是，如果在初始化对象的时候给pw赋值，能行是能行，IDE不干，非要给你标出来，强迫症就很烦，我都在下一行赋