02. xadmin的過濾器queryset()

需求: 每個老師都只能看到自己的課程

 

# models.py
from django.contrib.auth.models import AbstractUser class UserProfile(AbstractUser): ...... class Courses: teacher = models.ForeignKey(Teacher, on_delete=models.CASCADE) class Teacher: # 由於是中途才添加的user外鍵,所以當前teacher表已經存在數據,此時最好設置為SET_NULL
    user=models.OneToOneField(UserProfile, null=True, blank=True, on_delete=models.SET_NULL)

 

# adminx.py
import xadmin from apps.courses.models import Courses class CoursesAdmin: list_display = ['name', 'desc', ... ] search_fields = ['name', 'desc', ... ] list_filter = ['name', ...] # 重寫queryset()
    def queryset(self): # 取出當前Courses表單的所有對象
        qs = super().queryset() # 如果不是超級管理員,就對qs進行過濾
        if not self.request.user.is_superuser: # 對於OneToOne這種外鍵,可直接反向取user.teacher
            qs = qs.filter(teacher=self.request.user.teacher) return qs xadmin.site.register(Courses, CoursesAdmin)

 

 

 

 

---end---