python - 如何在 Peewee 中创建一个将其值转换为 MD5 的字段

Question

我在 Peewee 中定义了如下模型:

class User(Model):
    username = CharField(null=False)
    password = FixedCharField(null=False, max_length=32)

问题是我希望将字段 password 散列为 MD5。例如:

user = User.create(username="whatever", password="whatever")
user.password # returns '008c5926ca861023c1d2a36653fd88e2'

我使用 MySQL，它已经有一个 MD5() 函数。所以……

• 有没有办法使用 MySQL 的内置 MD5() 函数？或者……
• 是否有一种内置方法可以像 Django 一样在某种程度上操纵 Model::create 方法？

---

环境

• MySQL 5.7.21
• python 3.5.2
• Peewee 3.1.0

Answer 1

来自peewee documentation hacks页:

def get_hexdigest(salt, raw_password):
    data = salt + raw_password
    return sha1(data.encode('utf8')).hexdigest()

@db.func()
def make_password(raw_password):
    salt = get_hexdigest(str(random()), str(random()))[:5]
    hsh = get_hexdigest(salt, raw_password)
    return '%s$%s' % (salt, hsh)

@db.func()
def check_password(raw_password, enc_password):
    salt, hsh = enc_password.split('$', 1)
    return hsh == get_hexdigest(salt, raw_password)


query = User.insert(
    username='charlie',
    password=fn.make_password('testing')).execute()