640?wx_fmt=gif


django 框架配有一个功能强大的管理站点,里面已经预定义了许多管理站点需要进行的操作,但是有时可能默认的操作不能满足需求,如需要批量更新选中的对象,这需要编写自定义 action 实现。本文基于 django 官方英文文档梳理了一下自定义管理操作需要做的工作,方便快速了解自定义管理操作的实现过程。


640?wx_fmt=jpeg


作者 | 单雨,责编 | 黄浩然

出品 | CSDN(ID:CSDNnews)



640?wx_fmt=png


Admin actions 简介



admin actions 是一些在 admin 站点下拉列表中可以选择执行的简单函数,比如删除一个对象操作。

640?wx_fmt=png



640?wx_fmt=png


编写 Admin actions



下面举例说明如何编写Admin action。例如批量更新一个名为 Article 的模型:

 
 

from django.db import models

STATUS_CHOICES = [
    ('d''Draft'),
    ('p''Published'),
    ('w''Withdrawn'),
]

classArticle(models.Model):
    title = models.CharField(max_length=100)
    body = models.TextField()
    status = models.CharField(max_length=1,choices=STATUS_CHOICES)

    def__str__(self):
        returnself.title


下面来编写 action 函数,这个函数接受三个函数:

•  当前的 ModelAdmin

•  一个代表当前 HTTP 请求的 HttpRequest 对象

•  一个包含用户选定的对象的 QuerySet (数据库检索对象)。

 
 

def make_published(modeladmin, request,queryset):
   queryset.update(status='p')


为了达到最佳性能,推荐使用 QuerySet 的 update 方法,

 
 

for obj in queryset:
   do_something_with(obj)


默认情况下,该操作将以“ Make published ”的形式出现在操作列表中,就是函数名,下划线将由空格替换。为了让这个方法更容易被人了解,我们可以给它添加一个 short_description 属性。

 
 

def make_published(modeladmin, request,queryset):
   queryset.update(status='p')
make_published.short_description ="Markselected stories as published"



640?wx_fmt=png


添加 action 到 ModelAdmin



完整程序如下:

 
 

from django.contrib import admin
from myapp.models import Article

defmake_published(modeladmin, request, queryset):
   queryset.update(status='p')
make_published.short_description ="Markselected stories as published"

classArticleAdmin(admin.ModelAdmin):
    list_display = ['title''status']
    ordering = ['title']
    actions = [make_published]

admin.site.register(Article, ArticleAdmin)


效果如下:

640?wx_fmt=png



640?wx_fmt=png

 

处理 action 中的错误



如果在自定义的 actions 中可能发生错误,需要捕获这些错误并调用 django.contrib.admin.ModelAdmin.message_user() 来向用户显示错误消息。



640?wx_fmt=png


将 action 定义为 ModelAdmin 的内部方法



上述定义的 action 存在一个不足之处就是它和 Article 对象紧密耦合,这会导致大量的重复代码出现,即使这些 action 的功能都相同。将 action 定义为 ModelAdmin 的方法可以避免出现这个问题。


修改上述 ArticleAdmin:

 
 

class ArticleAdmin(admin.ModelAdmin):
    ...

    actions = ['make_published']

    def make_published(self, request, queryset):
       queryset.update(status='p')
   make_published.short_description ="Mark selected stories aspublished"


注意 make_published 变为 ArticleAdmin 的内部方法,第一个参数变为了 self。将 action 变为 ModelAdmin 的方法,使得它可以调用任何admin 提供的方法。例如向用户发送消息:

 
 

class ArticleAdmin(admin.ModelAdmin):
    ...

    def make_published(self, request, queryset):
       rows_updated = queryset.update(status='p')
        if rows_updated ==1:
           message_bit ="1 story was"
        else:
           message_bit ="%s stories were"%rows_updated
        self.message_user(request, "%s successfully marked as published."%message_bit)


这使得 action 执行成功之后自动将 action 的行为和 admin 站点的行为相匹配。

效果如下:

640?wx_fmt=png



640?wx_fmt=png

 

提供中间页的 action



在一些情况下,action 需要提供一个中间页来提示用户是否确认执行操作,比如删除对象时提醒用户是否确认执行操作:


640?wx_fmt=png

要提供中间页,只需要返回一个 HttpResponse 或者子类即可。例如一个导出选中对象 json 文件的操作:

 
 

from django.core import serializers
from django.http import HttpResponse

defexport_as_json(modeladmin, request, queryset):
    response = HttpResponse(content_type="application/json")
   serializers.serialize("json", queryset, stream=response)
    return response


但是这样可能会使 action 的逻辑非常复杂,推荐做法是返回一个 HTTP 重定向,重定向到一个新的页面上,这样可以在页面的视图中执行更复杂的逻辑,避免 action 中充满复杂的逻辑。

 
 

from django.contrib import admin
fromdjango.contrib.contenttypes.models import ContentType
from django.http import HttpResponseRedirect

defexport_selected_objects(modeladmin, request, queryset):
    selected =request.POST.getlist(admin.ACTION_CHECKBOX_NAME)
    ct = ContentType.objects.get_for_model(queryset.model)
    return HttpResponseRedirect("/export/?ct=%s&ids=%s"% (ct.pk, ",".join(selected)))




640?wx_fmt=png


让 actions 在整个站点中都有效



如果想要让上述中的导出操作在整个管理站点中都能使用,可以这样做:

 
 

from django.contrib import admin

admin.site.add_action(export_selected_objects)


为了方便管理这个操作,比如使用代码移除这个操作,可以给它一个名称

 
 

from django.contrib import admin

admin.site.add_action(export_selected_objects,"export_selected")




640?wx_fmt=png


禁用 action



禁用站点范围作用的的 action:

 
 

admin.site.disable_action('delete_selected')


这将在整个管理站点范围内禁用该 action。

如果需要复用已经禁用的 action。只需要在相应的 ModelAdmin 的actions 属性中添加这个 action即可。例如复用上述禁用的 action:

 
 

# This one will
classAnotherModelAdmin(admin.ModelAdmin):
    actions = ['delete_selected']


在某个 ModelAdmin 中禁用所有的 action,只需要将 action 属性设置为None 即可。

 
 

class MyModelAdmin(admin.ModelAdmin):
    actions =None




640?wx_fmt=png


有条件的启用或者禁用 action



通过重写 ModelAdmin的get_actions() 方法可以实现有条件的启用或者禁用 action。该方法返回一个允许使用的 actions 字典,键是操作名称,值是(函数、名称、short_description)元组。例如,如果您只希望名称以“J”开头的用户能够批量删除对象:

 
 

class MyModelAdmin(admin.ModelAdmin):
    ...

    def get_actions(self, request):
        actions =super().get_actions(request)
        if request.user.username[0].upper() !='J':
            if'delete_selected'in actions:
                del actions['delete_selected']
        return actions




640?wx_fmt=png


设置 action 的权限



action 可以通过设置 allowed_permissions 属性来限制具有特定权限的用户的可用性:

 
 

efmake_published(modeladmin, request, queryset):
   queryset.update(status='p')
make_published.allowed_permissions = ('change',)


如果 allowed_permissions 具有多个权限,则只要用户通过至少一个权限检查,该操作就可用。allowed_permissions 可用值为:

•          add

•          change

•          delete

•          view

如果需要指定其他值,需要在ModelAdmin中定义一个has<value>\permission( self, request )方法,例如:

 
 

from django.contrib import admin
from django.contrib.auth import get_permission_codename

classArticleAdmin(admin.ModelAdmin):
    actions = ['make_published']

    def make_published(self, request, queryset):
       queryset.update(status='p')
   make_published.allowed_permissions = ('publish',)

    def has_publish_permission(self, request):
        """Does the user have the publishpermission?"""
        opts =self.opts
        codename = get_permission_codename('publish', opts)
        return request.user.has_perm('%s.%s'% (opts.app_label, codename))


最后附上官方文档地址:https://docs.djangoproject.com/zh-hans/2.2/ref/contrib/admin/actions/


作者简介:单雨,90 后工科男,伪文艺青年。目前就读于北京理工大学宇航系,喜欢研究 AI,网络爬虫,微信小程序以及机器人,痴迷于 Coding,睡前必撸码。

【END】

640?wx_fmt=jpeg

 热 文 推 荐 


☞华为发布麒麟 990 芯片;苹果召回部分电源插头转换器;KDevelop 5.4.2 发布 | 极客头条

☞高级软件工程师教会小白的那些事!

我如何在 16 岁成为全栈开发者?

2亿日活,日均千万级视频上传,快手推荐系统如何应对技术挑战?

Docker容器化部署Python应用

☞给面试官讲明白:一致性Hash的原理和实践

预警,CSW的50万枚尘封BTC即将重返市场?

☞她说:行!没事别嫁程序员!

640?wx_fmt=gif点击阅读原文,输入关键词,即可搜索您想要的 CSDN 文章。

640?wx_fmt=png

你点的每个“在看”,我都认真当成了喜欢


Logo

20年前,《新程序员》创刊时,我们的心愿是全面关注程序员成长,中国将拥有新一代世界级的程序员。20年后的今天,我们有了新的使命:助力中国IT技术人成长,成就一亿技术人!

更多推荐