66 lines
2.4 KiB
Python
66 lines
2.4 KiB
Python
|
from django.contrib import admin
|
||
|
|
from .models import *
|
||
|
|
|
||
|
|
admin.site.site_title = '博客管理后台'
|
||
|
|
admin.site.site_header = '博客管理'
|
||
|
|
|
||
|
|
|
||
|
|
@admin.register(ArticleTag)
|
||
|
|
class ArticleTagAdmin(admin.ModelAdmin):
|
||
|
|
list_display = ['id', 'tag', 'user']
|
||
|
|
|
||
|
|
# 根据当前用户名设置数据访问权限
|
||
|
|
def get_queryset(self, request):
|
||
|
|
qs = super().get_queryset(request)
|
||
|
|
return qs.filter(user_id=request.user.id)
|
||
|
|
|
||
|
|
# 新增或修改数据时，设置外键可选值
|
||
|
|
def formfield_for_foreignkey(self, db_field, request, **kwargs):
|
||
|
|
if db_field.name == 'user':
|
||
|
|
id = request.user.id
|
||
|
|
kwargs["queryset"] = MyUser.objects.filter(id=id)
|
||
|
|
return super().formfield_for_foreignkey(db_field, request, **kwargs)
|
||
|
|
|
||
|
|
|
||
|
|
@admin.register(ArticleInfo)
|
||
|
|
class ArticleInfoAdmin(admin.ModelAdmin):
|
||
|
|
list_display = ['author', 'title', 'content', 'articlephoto', 'created', 'updated']
|
||
|
|
|
||
|
|
# 根据当前用户名设置数据访问权限
|
||
|
|
def get_queryset(self, request):
|
||
|
|
qs = super().get_queryset(request)
|
||
|
|
return qs.filter(author_id=request.user.id)
|
||
|
|
|
||
|
|
# 新增或修改数据时，设置外键可选值
|
||
|
|
def formfield_for_manytomany(self, db_field, request, **kwargs):
|
||
|
|
if db_field.name == 'article_tag':
|
||
|
|
id = request.user.id
|
||
|
|
kwargs["queryset"] = ArticleTag.objects.filter(user_id=id)
|
||
|
|
return super().formfield_for_manytomany(db_field, request, **kwargs)
|
||
|
|
|
||
|
|
# 新增或修改数据时，设置外键可选值
|
||
|
|
def formfield_for_foreignkey(self, db_field, request, **kwargs):
|
||
|
|
if db_field.name == 'author':
|
||
|
|
id = request.user.id
|
||
|
|
kwargs["queryset"] = MyUser.objects.filter(id=id)
|
||
|
|
return super().formfield_for_foreignkey(db_field, request, **kwargs)
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
@admin.register(Comment)
|
||
|
|
class CommentAdmin(admin.ModelAdmin):
|
||
|
|
list_display = ['article', 'commentator', 'content', 'created']
|
||
|
|
|
||
|
|
# 根据当前用户名设置数据访问权限
|
||
|
|
def get_queryset(self, request):
|
||
|
|
qs = super().get_queryset(request)
|
||
|
|
return qs.filter(article__author__id=request.user.id)
|
||
|
|
|
||
|
|
# 新增或修改数据时，设置外键可选值
|
||
|
|
def formfield_for_foreignkey(self, db_field, request, **kwargs):
|
||
|
|
if db_field.name == 'article':
|
||
|
|
id = request.user.id
|
||
|
|
kwargs["queryset"] = Comment.objects.filter(article__author__id=id)
|
||
|
|
return super().formfield_for_foreignkey(db_field, request, **kwargs)
|