Peewee 自定义自增主键

在本文中，我们将介绍如何在 Peewee 模型中自定义自增主键。

在 Peewee 中，默认情况下，每个模型都有一个自增的主键字段，称为 id。这个主键字段会在每次插入新记录时自动递增。不过，有时候我们可能需要自定义自增主键的命名，或者使用其他类型的主键。

自定义命名

Peewee 允许我们在模型中使用自定义的字段作为主键，只需要在字段定义中将 primary_key 参数设置为 True 即可。例如，假设我们有一个 User 模型，我们想要将 uid 字段作为主键：

from peewee import *

class User(Model):
    uid = AutoField()
    username = CharField()
    password = CharField()

    class Meta:
        database = db
        primary_key = CompositeKey('uid')

上述代码中，我们将 uid 字段设置为主键，并通过 CompositeKey 函数告诉 Peewee 使用 uid 字段作为主键。

使用其他类型的主键

除了自定义命名外，我们还可以使用其他类型的主键。Peewee 支持多种类型的字段作为主键，包括整数、字符串、甚至自定义的字段类型。

例如，我们可以定义一个使用 UUID 字符串作为主键的模型：

from peewee import *

class User(Model):
    uuid = UUIDField(primary_key=True)
    username = CharField()
    password = CharField()

    class Meta:
        database = db

上述代码中，我们使用 UUIDField 类型来定义 uuid 字段作为主键。当插入新记录时，我们需要手动为 uuid 字段生成一个合适的 UUID 值。

自定义自增实现

如果需要更加精细地控制自增主键的逻辑，我们可以手动实现自定义的自增逻辑。Peewee 提供了一个叫做 fn 的函数，可以用于执行底层数据库函数。

例如，假设我们想要实现一个自增的字符串主键，每次插入新记录时，这个主键的值应该是前一个记录主键的值加上一个固定的前缀。我们可以通过在插入前查询当前最大主键值，然后在插入时手动拼接新主键值来实现：

from peewee import *

class CustomAutoincrementField(AutoField):
    def __init__(self, prefix='', **kwargs):
        super().__init__(**kwargs)
        self.prefix = prefix

    def db_value(self, value):
        return value

    def python_value(self, value):
        return self.prefix + value

class User(Model):
    uid = CustomAutoincrementField(prefix='USER_')
    username = CharField()
    password = CharField()

    class Meta:
        database = db

上述代码中，我们定义了一个 CustomAutoincrementField 类，继承自 Peewee 的 AutoField 类。我们重写了 db_value 和 python_value 方法来实现自定义的自增逻辑。在插入新记录时，我们会查询当前最大主键值，然后手动拼接新主键值。在读取数据时，我们会将字段值的前缀去除。

以上是如何在 Peewee 模型中自定义自增主键的示例。通过自定义命名、使用其他类型的主键字段或者自定义的自增逻辑，我们可以根据自己的需求来灵活地定制主键的行为。