Django InlineModelAdmin 上覆盖 save_model

在本文中，我们将介绍如何在 Django InlineModelAdmin 中覆盖 save_model 方法。Django 的 InlineModelAdmin 是一种方便的方式，可以在一个父模型的编辑页面中同时编辑子模型。然而，在某些情况下，我们可能需要在保存子模型之前执行一些额外的逻辑。这时，我们可以覆盖 InlineModelAdmin 的 save_model 方法来实现我们的需求。

阅读更多：Django 教程

Django InlineModelAdmin 的基本使用方法

首先，让我们了解一下 Django InlineModelAdmin 的基本使用方法。在 Django 中，我们可以使用 InlineModelAdmin 将子模型与父模型相关联，实现在父模型的编辑页面中同时编辑子模型的功能。以下是一个简单的示例：

from django.contrib import admin

class ChildModelInline(admin.StackedInline):
    model = ChildModel

class ParentModelAdmin(admin.ModelAdmin):
    inlines = [
        ChildModelInline,
    ]

admin.site.register(ParentModel, ParentModelAdmin)

在这个示例中，我们创建了一个名为 ChildModelInline 的 InlineModelAdmin 类，并将其与子模型 ChildModel 关联。然后，我们在父模型 ParentModelAdmin 中定义了一个 inlines 列表，将 ChildModelInline 加入其中。最后，我们通过调用 admin.site.register() 方法将 ParentModel 和 ParentModelAdmin 关联起来。

通过这样的配置，当我们在 Django admin 后台编辑父模型 ParentModel 时，会显示一个内嵌的子模型 ChildModel 的编辑表单。

覆盖 save_model 方法

有时，我们需要在保存子模型之前或之后执行一些额外的逻辑。例如，我们可能想要在保存子模型之前对数据进行预处理，或者在保存子模型之后发送通知邮件。为了实现这样的需求，我们可以覆盖 InlineModelAdmin 的 save_model 方法。

save_model 方法有三个参数：request、obj 和 form。request 是当前的 HTTP 请求对象，obj 是即将保存的子模型实例，form 是用于编辑子模型的表单实例。我们可以使用这些参数来实现我们的逻辑。

from django.contrib import admin

class ChildModelInline(admin.StackedInline):
    model = ChildModel

    def save_model(self, request, obj, form, change):
        # 额外逻辑
        # ...

        # 保存子模型
        super().save_model(request, obj, form, change)

class ParentModelAdmin(admin.ModelAdmin):
    inlines = [
        ChildModelInline,
    ]

admin.site.register(ParentModel, ParentModelAdmin)

在这个示例中，我们在 ChildModelInline 中定义了一个 save_model 方法，并在其中添加了额外的逻辑。在处理完额外的逻辑之后，我们调用了 super().save_model(request, obj, form, change)，即调用父类的 save_model 方法，来保存子模型。

示例：保存子模型前进行数据预处理

假设我们有一个博客应用程序，其中有一个父模型 Post 和一个子模型 Comment。每当用户在父模型的编辑页面中添加新的评论时，我们希望自动将评论内容中的敏感词替换为星号，并保存到数据库中。

我们可以使用 save_model 方法来实现这一需求：

from django.contrib import admin

class CommentInline(admin.TabularInline):
    model = Comment

    def save_model(self, request, obj, form, change):
        # 数据预处理：将敏感词替换为星号
        sensitive_words = ['敏感词1', '敏感词2', '敏感词3']
        for word in sensitive_words:
            obj.content = obj.content.replace(word, '*' * len(word))

        # 保存子模型
        super().save_model(request, obj, form, change)

class PostAdmin(admin.ModelAdmin):
    inlines = [
        CommentInline,
    ]

admin.site.register(Post, PostAdmin)

在这个示例中，我们在 CommentInline 中覆盖了 save_model 方法，并在其中进行了数据预处理操作。首先，我们定义了一个敏感词列表 sensitive_words，然后遍历该列表，将评论内容中的每个敏感词替换为相同长度的星号字符串。最后，我们调用了父类的 save_model 方法来保存子模型。

通过这样的实现，当用户在编辑父模型 Post 时，新增一个评论，并输入了包含敏感词的评论内容时，保存到数据库中的评论内容中的敏感词将会被替换为星号。

总结

在本文中，我们介绍了如何在 Django InlineModelAdmin 中覆盖 save_model 方法。我们首先了解了 Django InlineModelAdmin 的基本使用方法，然后讲解了如何覆盖 save_model 方法来实现在保存子模型之前或之后执行额外逻辑的需求。最后，我们通过一个示例说明了如何在保存子模型前进行数据预处理。

通过覆盖 save_model 方法，我们可以灵活地在 Django admin 后台中进行特定操作，并实现更多的定制化需求。这使得我们能够更好地管理和操作我们的数据。