Peewee - 主键和复合主键



建议关系数据库中的表应该有一列应用主键约束。相应地,Peewee 模型类也可以指定字段属性,并将 primary-key 参数设置为 True。但是,如果模型类没有任何主键,Peewee 会自动创建一个名为“id”的主键。请注意,上面定义的 User 模型没有任何字段明确定义为主键。因此,我们数据库中映射的 User 表有一个 id 字段。

要定义一个自动递增的整数主键,请在模型中使用 **AutoField** 对象作为属性。

class User (Model):
   user_id=AutoField()
   name=TextField()
   age=IntegerField()
   class Meta:
      database=db
      db_table='User'

这将转换为以下 CREATE TABLE 查询:

CREATE TABLE User (
   user_id INTEGER NOT NULL
   PRIMARY KEY,
   name TEXT NOT NULL,
   age INTEGER NOT NULL
);

您也可以将任何非整数字段指定为主键,方法是将 primary_key 参数设置为 True。假设我们想存储某个字母数字值作为 user_id。

class User (Model):
   user_id=TextField(primary_key=True)
   name=TextField()
   age=IntegerField()
   class Meta:
      database=db
      db_table='User'

但是,当模型包含非整数字段作为主键时,模型实例的 **save()** 方法不会导致数据库驱动程序自动生成新 ID,因此我们需要传递 **force_insert=True** 参数。但是,请注意,**create()** 方法隐式指定了 force_insert 参数。

User.create(user_id='A001',name="Rajesh", age=21)
b=User(user_id='A002',name="Amar", age=20)
b.save(force_insert=True)

save() 方法还会更新表中的现有行,此时不需要 force_insert 主键,因为具有唯一主键的 ID 已经存在。

Peewee 允许定义复合主键的功能。**CompositeKey** 类的对象在 Meta 类中定义为主键。在下面的示例中,由 User 模型的 name 和 city 字段组成的复合键已被指定为复合主键。

class User (Model):
   name=TextField()
   city=TextField()
   age=IntegerField()
   class Meta:
      database=db
      db_table='User'
      primary_key=CompositeKey('name', 'city')

此模型转换为以下 CREATE TABLE 查询。

CREATE TABLE User (
   name TEXT NOT NULL,
   city TEXT NOT NULL,
   age INTEGER NOT NULL,
   PRIMARY KEY (
      name,
      city
   )
);

如果您希望表没有主键,则在模型的 Meta 类中指定 primary_key=False。

广告