Django允许输入自定义外键值|极客笔记

Django允许输入自定义外键值

在Django中，外键（ForeignKey）是用于建立表与表之间关联关系的重要字段之一。通常情况下，外键字段会引用另一个表的主键值作为其值。然而，有时候我们希望能够允许用户输入自定义的外键值，而不是限定在已有的主键值中选择。本文将详细介绍如何在Django中实现这一功能。

1. 创建Django项目和应用

首先，我们需要创建一个Django项目和应用。假设我们的项目名为custom_foreign_key，应用名为custom_fk_example。通过以下命令创建项目和应用：

django-admin startproject custom_foreign_key
cd custom_foreign_key
python manage.py startapp custom_fk_example

2. 定义模型

接下来，我们需要定义模型，并在模型中创建一个外键字段，允许用户输入自定义值。在custom_fk_example应用下的models.py文件中定义如下模型：

from django.db import models

class Author(models.Model):
    name = models.CharField(max_length=100, unique=True)

    def __str__(self):
        return self.name

class Book(models.Model):
    title = models.CharField(max_length=100)
    author = models.ForeignKey(Author, on_delete=models.CASCADE)

    def __str__(self):
        return self.title

在上面的模型中，我们定义了两个模型Author和Book，Book模型中有一个外键字段author指向Author模型。现在外键字段的值必须是Author表中已存在的主键值。

3. 自定义外键值输入

为了实现允许用户输入自定义的外键值，我们可以自定义一个字段，而不使用Django提供的ForeignKey字段。在custom_fk_example应用下创建一个新的文件fields.py，并在其中定义一个自定义的外键字段CustomForeignKey：

from django.db import models

class CustomForeignKey(models.ForeignKey):
    def formfield(self, **kwargs):
        kwargs['queryset'] = self.remote_field.model._default_manager.all()
        return super().formfield(**kwargs)

在自定义的外键字段CustomForeignKey中重写了formfield方法，通过设置queryset为关联模型的所有对象，实现了允许用户输入自定义值的功能。

4. 修改Book模型

接下来，我们需要在Book模型中使用自定义的外键字段CustomForeignKey替换原有的ForeignKey字段。修改Book模型的定义如下：

from django.db import models
from custom_fk_example.fields import CustomForeignKey

class Book(models.Model):
    title = models.CharField(max_length=100)
    author = CustomForeignKey(Author, on_delete=models.CASCADE)

    def __str__(self):
        return self.title

5. 迁移数据库

我们需要执行数据库迁移命令，将模型的更改同步到数据库中：

python manage.py makemigrations
python manage.py migrate

6. 创建书籍

现在我们可以在Django的管理后台中创建书籍。登录管理后台，在custom_fk_example应用下的Book模型中添加一本书籍。在添加书籍时，可以输入现有作者的名称，也可以直接输入新作者的名称作为外键值。

7. 测试功能

为了验证功能是否实现，我们可以在Django的Shell中进行测试。首先启动Django的Shell：

python manage.py shell

然后尝试创建一本书籍，并输入自定义的作者名称：

from custom_fk_example.models import Book, Author

author = Author.objects.create(name='John Doe')
book = Book.objects.create(title='Custom Foreign Key Example', author=author)

通过以上步骤，我们成功地创建了一本书籍，并输入了自定义的作者名称作为外键值。