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
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
| from django.contrib import admin
| from django.contrib.auth.admin import UserAdmin as BaseUserAdmin
|
| from django.contrib.auth import get_user_model
| from django.utils.translation import gettext_lazy as _
|
| from lunarlink.models import Project
|
| # Register your models here.
| User = get_user_model()
|
|
| @admin.register(User)
| class UserAdmin(BaseUserAdmin):
| def accessible_projects(self, obj):
| user_groups = obj.groups.all()
| # 用多对多的关系来过滤出属于用户所在分组的项目。实际应用中可能会有更复杂的逻辑。
| return ", ".join(
| Project.objects.filter(groups__in=user_groups)
| .distinct()
| .values_list("name", flat=True)
| )
|
| accessible_projects.short_description = "可访问项目"
|
| list_display = (
| "username",
| "name",
| "is_active",
| "belong_groups",
| "accessible_projects",
| )
|
| # 编辑资料的时候显示的字段
| fieldsets = (
| (None, {"fields": ("username", "name", "password")}),
| (
| _("Permissions"),
| {
| "fields": (
| "is_active",
| "is_staff",
| "is_superuser",
| "groups",
| ),
| },
| ),
| )
|
| # 新增用户需要填写的字段
| add_fieldsets = (
| (
| None,
| {
| "classes": ("wide",),
| "fields": (
| "username",
| "name",
| "password1",
| "password2",
| "is_active",
| "is_staff",
| "groups",
| ),
| },
| ),
| )
| filter_horizontal = ("groups",)
|
| def belong_groups(self, obj):
| return ", ".join([g.name for g in obj.groups.all()])
|
| belong_groups.short_description = "所属分组"
|
|
