使用遗留数据库在 Django 项目中使用复合主键

Working with composite primary key in django project with legacy database

我有一个遗留数据库,其中一些 table 包含复合主键。我通过 运行ning manage.py inspectdb 命令获得的模型如下所示。

class MyTable(models.Model):
    field1_id = models.IntegerField(db_column='field1id', primary_key=True)
    is_favorite = models.BooleanField(db_column='isfavorite', default=False, null=False)
    is_admin = models.BooleanField(db_column='isadmin', default=False, null=False)
    role = models.IntegerField(default=USER_GUEST, null=False)
    field2 = models.BooleanField(null=False, default=True)
    field3 = models.BooleanField(null=False, default=True)
    is_active = models.BooleanField(db_column='isactive', null=False, default=True)

    user = models.ForeignKey(
        CustomUser, models.DO_NOTHING, db_column='userid', primary_key=True)

    class Meta:
        managed = False
        db_table = 'mytable'
        unique_together = (('user', 'field1_id'),)

我可以正常获取数据。但是,当我想在某些模型实例上执行 运行 save() 命令时,问题就出现了。 django 执行的查询不正确。例如:

>>> from web_services.apps.my_app.models import MyTable
>>> g = MyTable.objects.get(field1_id=12)
>>> g.is_active = True
>>> g.save()
>>> connection.queries[-1]
{'time': '0.000', 'sql': 'UPDATE `mytable` SET `isfavorite` = 0, `isadmin` = 1, `role` = 3, `field2` = 1, `field3` = 1, `isactive` = 1 WHERE `mytable`.`field1id` = 12'}

但我需要:

{'time': '0.000', 'sql': 'UPDATE `mytable` SET `isfavorite` = 0, `isadmin` = 1, `role` = 3, `field2` = 1, `field3` = 1, `isactive` = 1 WHERE `mytable`.`field1id` = 12' AND `mytable`.`userid` = 1'}

由于 django 不支持复合主键,克服该问题的最佳解决方案是什么?请注意,它是遗留数据库 table 并且没有 AutoField.

编辑:添加遗留 table 结构

+------------+------------+------+-----+---------+-------+
| Field      | Type       | Null | Key | Default | Extra |
+------------+------------+------+-----+---------+-------+
| userid     | int(11)    | NO   | PRI | NULL    |       |
| field1id   | int(11)    | NO   | PRI | NULL    |       |
| isfavorite | int(11)    | NO   |     | 0       |       |
| isadmin    | int(11)    | NO   |     | 0       |       |
| role       | int(11)    | YES  |     | NULL    |       |
| field2     | tinyint(1) | NO   |     | 1       |       |
| field3     | tinyint(1) | NO   |     | 1       |       |
| isactive   | tinyint(4) | NO   |     | 1       |       |
+------------+------------+------+-----+---------+-------+

可惜django做的not yet have a composite primary key(复合主键也叫多列主键)

有一个第三方库,不出所料,名称为 django-compositekey,它可以创建具有多列主键的模型。但是该项目没有维护并且与最新版本的 django 不兼容。

您最好的选择是在您的 table 中添加一个新列,这是一个 AutoField 并使其成为新的主键。当前构成主键的字段可以标记为unique_together。虽然这对于一些查询来说可能不是理想的。对大多数人来说已经足够了。

如果您使用当前 table 结构更新您的问题(如您的 sql 控制台所示),我将更新我的答案以显示模型的外观。

更新 有许多不同的方法可以删除旧的复合主键并将其替换为新的自动递增主键。这是最简单的,它只涉及 SQL

CREATE TABLE newtable LIKE mytable;
ALTER TABLE newtable DROP PRIMARY KEY;
ALTER TABLE newtable ADD `id` INT NOT NULL AUTO_INCREMENT PRIMARY KEY;
RENAME TABLE mytable to mytable_old;
RENAME TABLE newtable to mytable;
INSERT INTO mytable(userid,field1id, rest of the fields) 
    SELECT * FROM mytable_old;

然后编辑您的模型并从这些字段中删除 primary_key=True 标志。

脚注:
例如 sqlite 等一些数据库不支持 CREATE TABLE 中的 LIKE 子句。在这些情况下,您必须查找原始 table 的 create table 语句,并在编辑 table 名称后复制粘贴。从您的 table 结构来看,您似乎正在使用 mysql 并且它支持 LIKE 子句。

您必须从模型中删除 managed = False 命令才能编辑您的 table