根据需求定制 admin

定义 list 页面

自定义 list_filter

首先，完成过滤器的功能，需要自定义过滤器。在 PostAdmin 定义的上方定义如下代码：

class CategoryOwnerFilter(admin.SimpleListFilter):

    """

    自定义过滤器只展示当前用户分类

    """

    title = '分类过滤器'

    parameter_name = 'owner_category'

    def lookups(self, request, model_admin):

        return Category.objects.filter(owner=request.user).values_list('id', 'name')

    def queryset(self, request, queryset):

        category_id = self.value()

        if category_id:

            return queryset.filter(category_id=self.value())

        return queryset

通过继承 Django admin 提供的 SimpleListFilter 类来实现自定义过滤器，之后只需要把自定义过滤器配置到 ModelAdmin 中即可。

SimpleListFilter 类提供了两个属性和两个方法。title 用于展示标题，parameter_name 就是查询时 URL 参数的名字，比如查询分类 id 为 1 的内容时，URL 后面的 Query 部分是 ?owner_category=1 ，此时就可以通过过滤器拿到这个 id，从而进行过滤。

lookups：返回要展示的内容和查询用的 id（就是上面 Query 用的）。

queryset：根据 URL Query 的内容返回列表页数据。比如如果 URL 最后的 Query 是 ?owner_category=1 ，那么这里拿到的 self.value() 就是 1，此时就会根据 1 来过滤 Queryset 。这里的 Queryset 是列表页所有展示数据的合集，即 post 的数据集。

编写完之后，只需要修改 list_filter 为：

list_filter = [CategoryOwnerFilter]

就能让用户在侧边栏的过滤器中只看到自己创建的分类了。

自定义列表页数据

完成了 list_filter 的定制，还需要继续定制，让当前登录的用户在列表页中只能看到自己创建的文章。

PostAdmin 继承自 admin.ModelAdmin，显然需要看 ModelAdmin 提供了哪些方法，可以使其重写。

blog/admin.py：

@admin.register(Post)

class PostAdmin(admin.ModelAdmin):

    list_display = [

        'title', 'category', 'status',

        'created_time', 'owner', 'operator'

    ]

	# 省略部分代码

    def save_model(self, request, obj, form, change):

        obj.owner = request.user

        return super(PostAdmin, self).save_model(request, obj, form, change)

    def get_queryset(self, request):

        qs = super(PostAdmin, self).get_queryset(request)

        return qs.filter(owner=request.user)

从这两个定制可以看出来，关于数据过滤的部分，只需要找到数据源在哪，也就是 QuerySet 最终在哪生成，然后对其进行过滤即可。

编辑页面的配置

首先，是按钮位置，在编辑页面中主要的按钮就是 “保存” ，不过 Django 提供了另外两个便于操作的按钮：“保存并继续” 以及 “保存并新增另一个” 。关于按钮的位置，有一个配置项可以完成：save_on_top 用来控制是否在页面顶部展示上述的三个按钮。

对于字段是否展示以及展示顺序的需求，可以通过 fields 或者 fieldset 来配置。通过 exclude 可以指定哪些字段是不展示的，比如下面的 owner，是在程序中自动赋值当前用户的。代码如下：

# blog/admin.py

@admin.register(Post)

class PostAdmin(admin.ModelAdmin):

    # 省略其他代码

    exclude = ('owner', )

    fields = (

        ('category', 'title'),

        'desc',

        'status',

        'content',

        'tag',

    )

    # 省略其他代码

fields 配置有两个作用，一个是限定要展示的字段，另外一个是配置展示字段的顺序。

fieldsets 用来控制页面布局，先用它来替换上述代码的 fields ：

fieldsets = (

    ('基础配置', {

        'description': '基础配置描述',

        'fields': (

            ('title', 'category'),

            'status',

        ),

    }),

    ('内容', {

        'fields': (

            'desc',

            'content',

        ),

    }),

    ('额外信息', {

        'classes': ('collapse', ),

        'fields': ('tag', ),

    })

)

fieldsets 要求的格式是有两个元素的 tuple 的 list，如：

fieldsets = (

    (名称, {内容}),

    (名称, {内容}),

)

其中包含两个元素的 tuple 内容，第一个元素是当前版块的名称，第二个元素是当前版块的描述、字段和样式配置。也就是说，第一个元素是 string，第二个元素是 dict，而 dict 的 key 可以是 fields、description 和 classes。

fields 的配置效果同上面一样，可以控制展示哪些元素，也可以给元素排序并组合元素的位置。classes 的作用就是给要配置的版块加上一些 CSS 属性，Django admin 默认支持的是 collapse 和 wide 。当然，也可以写其他属性，然后自己来处理样式。

关于编辑页的配置，还有针对多对多字段展示的配置 filter_horizontal 和 filter_vertica1，它们用来控制多对多字段的展示效果。

自定义静态资源引入

可以通过自定义 Media 类来往页面上增加想要添加的 JavaScript 以及 CSS 资源。

class PostAdmin(admin.ModelAdmin):

    class Media:

        css = {

            'all': ("https://cdn.bootcss.com/bootstrap/4.0.0-beta.2/css/bootstrap.min.css", ),

        }

        js = ('https://cdn.bootcss.com/bootstrap/4.0.0-beta.2/js/bootstrap.bundle.js', )

自定义 Form

上面的所有配置都是基于 ModelAdmin 的，如果有更多的定制需求，比如文章描述字段能够以多行多列的方式展示，这就需要用到 ModelForm 了。

先在 blog 的目录下新增一个文件 adminforms.py。因为这是用作后台管理的 Form，所以这里要命名为 adminforms 而不是 forms。这只是为了与前台针对用户输入进行处理的 Form 区分开来。

接着，需要在里面编写代码，定义 Form。关于 Form 的作用，可以说 Form 跟 Model 的逻辑是一致的，Model 是对数据库中字段的抽象，Form 是对用户输入以及 Model 中要展示数据的抽象。

在 adminforms.py 中，通过 Form 来定制 status 这个字段的展示：

from django import forms

class PostAdminForm(forms.ModelForm):

    desc = forms.CharField(widget=forms.Textarea, label='摘要', required=False)

接着将其配置到 admin 定义中：

# blog/admin.py

from .adminforms import PostAdminForm

class PostAdmin(admin.ModelAdmin):

    form = PostAdminForm

    # 省略其他代码

这时，刷新页面，就能看到文章描述字段已经改为 Textarea 组件了。

定制 site

一个 site 对应一个站点，现在有一个新的需求：用户模块的管理应该跟文章分类等数据的管理分开。另外，也需要修改后台的默认展示。

url(r'^admin/', admin.site.urls)

这里的 site 是 django.contrib.admin.AdminSite 的一个实例，因此，可以继承 AdminSite 来定义自己的 site，代码如下：

from django.contrib.admin import AdminSite

class CustomSite(AdminSite):

    site_header = 'Typeidea'

    site_title = 'Typeidea管理后台'

    index_title = '首页'

custom_site = CustomSite(name='cus_admin')

把代码放到 typeidea/typeidea/custom_site.py 中，然后修改所有 App 下 register 部分的代码，以 PostAdmin 为例：

把 @admin.register(Post) 改为 @admin.register(Post, site=custom_site) ，这需要在模块上引入 from typeidea.custom_site import custom_site

上面用的是 reverse 方式来获取后台地址，用到了 admin 这个名称，因此需要调整 blog/admin.py 代码。

原代码：

def operator(self, obj):

    return format_html(

        '<a href="{}">编辑</a>',

        reverse('admin:blog_post_change', args=(obj.id,))

    )

operator.short_description = '操作'

修改为：

def operator(self, obj):

    return format_html(

        '<a href="{}">编辑</a>',

        reverse('cus_admin:blog_post_change', args=(obj.id,))

    )

operator.short_description = '操作'

接着，需要在 urls.py 文件中进行修改，其完整代码如下：

from django.conf.urls import url

from django.contrib import admin

from .custom_site import custom_site

urlpatterns = [

    url(r'^super_admin/', admin.site.urls),

    url(r'^admin/', custom_site.urls)

]

这样就有两套后台地址，一套用来管理用户，另一套用来管理业务。但这两套系统都是基于一套逻辑的用户系统，只是在 URL 上进行了划分。

抽取 Admin 基类

抽象 author 基类

先抽象出一个基类 BaseownerAdmin，这个类完成两件事：一是重写 save 方法，此时需要设置对象的 owner；二是重写 get_queryset 方法，让列表页在展示文章或者分类时只能展示当前用户的数据。代码如下：

from django.contrib import admin

class BaseOwnerAdmin(admin.ModelAdmin):

    """

    1. 用来自动补充文章、分类、标签、侧边栏、友链这些 Model的 owner 字段

    2. 用来针对 queryset 过滤当前用户的数据

    """

    exclude = ('owner', )

    def get_queryset(self, request):

        qs = super(BaseOwnerAdmin, self).get_queryset(request)

        return qs.filter(owner=request.user)

    def save_model(self, request, obj, form, change):

        obj.owner = request.user

        return super(BaseOwnerAdmin, self).save_model(request, obj, form, change)

把这段代码放到 base_admin.py 中，与 custom_site.py 同目录。之所以放这里是因为所有的 App 都需要用到。

有了这个基类，接下来需要做的就是改造那些需要隔离不同用户数据的管理页面，只需要让对应的 Admin 类继承这个基类即可。这里列一下完整的 blog/admin.py 代码：

from django.contrib import admin

from django.urls import reverse

from django.utils.html import format_html

from .adminforms import PostAdminForm

from .models import Post, Category, Tag

from typeidea.base_admin import BaseOwnerAdmin

from typeidea.custom_site import custom_site

class PostInline(admin.TabularInline):

    fields = ('title', 'desc')

    extra = 1   # 控制额外多几个

    model = Post

@admin.register(Category, site=custom_site)

class CategoryAdmin(BaseOwnerAdmin):

    inlines = [PostInline, ]

    list_display = ('name', 'status', 'is_nav', 'created_time', 'post_count')

    fields = ('name', 'status', 'is_nav')

    def post_count(self, obj):

        return obj.post_set.count()

    post_count.short_description = '文章数量'

@admin.register(Tag, site=custom_site)

class TagAdmin(BaseOwnerAdmin):

    list_display = ('name', 'status', 'created_time')

    fields = ('name', 'status')

class CategoryOwnerFilter(admin.SimpleListFilter):

    """

    自定义过滤器只展示当前用户分类

    """

    title = '分类过滤器'

    parameter_name = 'owner_category'

    def lookups(self, request, model_admin):

        return Category.objects.filter(owner=request.user).values_list('id', 'name')

    def queryset(self, request, queryset):

        category_id = self.value()

        if category_id:

            return queryset.filter(category_id=self.value())

        return queryset

@admin.register(Post, site=custom_site)

class PostAdmin(admin.ModelAdmin):

    form = PostAdminForm

    list_display = [

        'title', 'category', 'status',

        'created_time', 'owner', 'operator'

    ]

    list_display_links = []

    list_filter = [CategoryOwnerFilter]

    search_fields = ['title', 'category__name']

    actions_on_top = True

    actions_on_bottom = True

    # 编辑页面

    save_on_top = True

    exclude = ('owner', )

    # fields = (

    #     ('category', 'title'),

    #     'desc',

    #     'status',

    #     'content',

    #     'tag',

    # )

    fieldsets = (

        ('基础配置', {

            'description': '基础配置描述',

            'fields': (

                ('title', 'category'),

                'status',

            ),

        }),

        ('内容', {

            'fields': (

                'desc',

                'content',

            ),

        }),

        ('额外信息', {

            'classes': ('collapse', ),

            'fields': ('tag', ),

        })

    )

    # filter_horizontal = ('tag', )

    filter_vertical = ('tag', )

    def operator(self, obj):

        return format_html(

            '<a href="{}">编辑</a>',

            reverse('cus_admin:blog_post_change', args=(obj.id,))

        )

    operator.short_description = '操作'

    # class Media:

    #     css = {

    #         'all': ("https://cdn.bootcss.com/twitter-bootstrap/3.3.7/css/bootstrap.min.css", ),

    #     }

    #     js = ('https://cdn.bootcss.com/twitter-bootstrap/3.3.7/js/bootstrap.js', )

在 admin 页面上查看操作日志

# blog/admin.py

from django.contrib.admin.models import LogEntry

@admin.register(LogEntry, site=custom_site)

class LogEntryAdmin(admin.ModelAdmin):

    list_display = ['object_repr', 'object_id', 'action_flag', 'user', 'change_message']

巴特西