-
Notifications
You must be signed in to change notification settings - Fork 5
/
Copy pathadmin.py
31 lines (24 loc) · 1.09 KB
/
admin.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
from django.contrib import admin
from notice.models import Notice
from department.views import get_department
from department.models import Department
class NoticeAdmin(admin.ModelAdmin):
search_fields = ['title', 'department__name']
list_display = ['title', 'get_bref', 'pub_date', 'department']
def get_queryset(self, request):
qs = super(NoticeAdmin, self).get_queryset(request)
if request.user.is_superuser:
return qs
if request.user.has_perm('department.is_manager'):
return qs.filter(department=request.user.manager.m_department)
def formfield_for_foreignkey(self, db_field, request, **kwargs):
department = get_department(request)
if department is not None:
if db_field.name == 'department':
kwargs['initial'] = department
kwargs['queryset'] = Department.objects.filter(
name=department.name
)
return super(NoticeAdmin, self).formfield_for_foreignkey(
db_field, request, **kwargs)
admin.site.register(Notice, NoticeAdmin)