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

Question

我有一个遗留数据库，其中一些表包含复合主键。我通过运行 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 不支持复合主键，克服该问题的最佳解决方案是什么？请注意，它是遗留数据库表并且没有 AutoField。

编辑:添加遗留表结构

+------------+------------+------+-----+---------+-------+
| 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       |       |
+------------+------------+------+-----+---------+-------+

Answer 1

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

有个第三方库，名字不出所料django-compositekey这使得创建具有多列主键的模型成为可能。然而，该项目没有得到维护，并且与最新版本的 django 不兼容。

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

如果您使用当前的表结构更新您的问题(如您的 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 子句。在这些情况下，您将不得不查找原始表的创建表语句，并在编辑表名后复制粘贴。从您的表结构来看，您似乎正在使用 mysql 并且它支持 LIKE 子句。