快捷導(dǎo)航

Django Admin 實現(xiàn)外鍵過濾的方法

更新時間：2017年09月29日 09:03:07 作者：ishouyong

下面小編就為大家?guī)硪黄狣jango Admin 實現(xiàn)外鍵過濾的方法。小編覺得挺不錯的，現(xiàn)在就分享給大家，也給大家做個參考。一起跟隨小編過來看看吧

說明和 Model

環(huán)境：

➜ python

Python 3.6.3 |Anaconda custom (x86_64)| (default, Oct 6 2017, 12:04:38)
[GCC 4.2.1 Compatible Clang 4.0.1 (tags/RELEASE_401/final)] on darwin
Type "help", "copyright", "credits" or "license" for more information.
>>> import django
>>> print(django.get_version())
2.0.1
>>>

2018年05月23日更新：

可以通過get_changeform_initial_data 函數(shù)來傳遞initial參數(shù).

# admin.py
@admin.register(Score)
class ScoreConfigAdmin(FilterUserAdmin):
  # fields = ('id','name')
  form = ScoreConfigAdminForm

  def get_changeform_initial_data(self, request):
    initial = super().get_changeform_initial_data(request)
    initial.update({'uid': request.user.id})
    return initial

# forms.py
class ScoreConfigAdminForm(forms.ModelForm):
  def __init__(self, *args, **kwargs):
    super().__init__(*args, **kwargs)
    if not kwargs.get('initial'):
      return
    self.uid = kwargs.get('initial').get('uid')

  class Meta:
    model = Score
    fields = '__all__'

有一個支持多用戶（使用 django admin）的 Blog，每一篇 Post 都需要記錄是誰發(fā)表的并且屬于那個 Blog。

user 與 Blog 的關(guān)系、 Blog 與 Post 有2種定義方式，一種是使用獨立關(guān)系表，另外一種是直接在 Model 中定義中使用外鍵。

后面一種的 model 定義如下：

from django.contrib.auth.models import User
from django.db import models

class Blog(models.Model):
  '''
  Blog
  '''
  id = models.AutoField(unique=True, primary_key = True, verbose_name="序號")
  name = models.CharField(max_length=255, blank=True, null=True, verbose_name="名稱")
  user = models.ForeignKey(User, on_delete=models.CASCADE)
  create_time = models.DateTimeField(verbose_name='添加時間', auto_now_add=True, blank=True)

  class Meta:
    verbose_name = 'Blog'
    verbose_name_plural = 'Blog管理'

  def __str__(self):
    return self.name

class Post(models.Model):
  '''
  Post 內(nèi)容
  '''
  id = models.AutoField(unique=True, primary_key = True, verbose_name="序號")
  title = models.CharField(max_length=255, blank=True, null=True, verbose_name="標(biāo)題")
  content = models.TextField(max_length=1024, blank=True, null=True, verbose_name="內(nèi)容")
  blog = models.ForeignKey(Blog, on_delete=models.CASCADE, verbose_name="所屬Blog")
  user = models.ForeignKey(User, on_delete=models.CASCADE)
  create_time = models.DateTimeField(verbose_name='添加時間', auto_now_add=True, blank=True)

  class Meta:
    verbose_name = '文章'
    verbose_name_plural = '文章管理'

  def __str__(self):
    return self.title

Admin 中實現(xiàn)

admin 中有2處，一處是 Blog 和 Post 列表中按 user 過濾，另外一處是新增 Post 時需要按當(dāng)前 user 過濾。完整代碼如下：

from django.contrib import admin
from django import forms

# Register your models here.
from django_summernote.admin import SummernoteModelAdmin
from .models import Team, Member, Activity, Score



from .models import Blog, Post
class FilterUserAdmin(admin.ModelAdmin):
  '''
  按所屬用戶過濾的 base, class
  '''
  def save_model(self, request, obj, form, change):
    # TODO 需要考慮不同用戶對同一數(shù)據(jù)進(jìn)行修改。
    obj.user = request.user
    obj.save()

  def get_queryset(self, request):
    # For Django < 1.6, override queryset instead of get_queryset
    qs = super(FilterUserAdmin, self).get_queryset(request) 
    # 不能加這個，加了這個會導(dǎo)致 superuser 更新普通用戶的數(shù)據(jù)。
    # if request.user.is_superuser:
    #   return qs
    return qs.filter(user=request.user)

  def has_change_permission(self, request, obj=None):
    has_class_permission = super(FilterUserAdmin, self).has_change_permission(request, obj)
    if not has_class_permission:
      return False
    if obj is not None and not request.user.is_superuser and request.user.id != obj.user.id:
      return False
    return True


class BlogConfigAdmin(FilterUserAdmin):
  list_display = ('id','name', 'create_time')
  exclude = ['user']
  list_per_page = 50

admin.site.register(Blog, BlogConfigAdmin)


class PostConfigAdmin(FilterUserAdmin):
  list_display = ('id','title', 'create_time')
  exclude = ['user']
  list_per_page = 50

  def render_change_form(self, request, context, add=False, change=False, form_url='', obj=None):
    # 新增 Post 時，相關(guān)聯(lián)的 Blog 需要過濾，關(guān)鍵就在下面這句。
    context['adminform'].form.fields['blog'].queryset = Team.objects.filter(user=request.user)
    return super(MemberConfigAdmin, self).render_change_form(request, context, add, change, form_url, obj)


admin.site.register(Post, PostConfigAdmin)

說2句

在render_change_form中下斷點，直接調(diào)試下會發(fā)現(xiàn)更多有趣的內(nèi)容。

以上這篇Django Admin 實現(xiàn)外鍵過濾的方法就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。

您可能感興趣的文章: