87 lines
2.1 KiB
Python
87 lines
2.1 KiB
Python
from django.contrib import admin
|
|
from .models import Team, TeamArea, TeamMember, TeamTask
|
|
from .email import add_added_membership_email, add_removed_membership_email
|
|
|
|
|
|
@admin.register(TeamTask)
|
|
class TeamTaskAdmin(admin.ModelAdmin):
|
|
list_display = [
|
|
'id',
|
|
'team',
|
|
'name',
|
|
'description',
|
|
]
|
|
|
|
|
|
@admin.register(Team)
|
|
class TeamAdmin(admin.ModelAdmin):
|
|
def get_responsible(self, obj):
|
|
return ", ".join([resp.get_full_name() for resp in obj.responsible])
|
|
get_responsible.short_description = 'Responsible'
|
|
|
|
list_display = [
|
|
'name',
|
|
'area',
|
|
'get_responsible',
|
|
'needs_members',
|
|
]
|
|
|
|
list_filter = [
|
|
'camp',
|
|
'needs_members',
|
|
]
|
|
|
|
|
|
@admin.register(TeamMember)
|
|
class TeamMemberAdmin(admin.ModelAdmin):
|
|
list_filter = [
|
|
'team',
|
|
'approved',
|
|
]
|
|
|
|
actions = ['approve_membership', 'remove_member']
|
|
|
|
def approve_membership(self, request, queryset):
|
|
teams_count = queryset.values('team').distinct().count()
|
|
updated = 0
|
|
|
|
for membership in queryset:
|
|
membership.approved = True
|
|
membership.save()
|
|
updated += 1
|
|
add_added_membership_email(membership)
|
|
|
|
self.message_user(
|
|
request,
|
|
'Membership(s) approved: Added {} user(s) to {} team(s).'.format(
|
|
updated,
|
|
teams_count
|
|
)
|
|
)
|
|
approve_membership.description = 'Approve membership.'
|
|
|
|
def remove_member(self, request, queryset):
|
|
teams_count = queryset.values('team').distinct().count()
|
|
updated = 0
|
|
|
|
for membership in queryset:
|
|
add_removed_membership_email(membership)
|
|
membership.delete()
|
|
updated += 1
|
|
|
|
self.message_user(
|
|
request,
|
|
'Removed {} user(s) from {} team(s).'.format(
|
|
updated,
|
|
teams_count
|
|
)
|
|
)
|
|
remove_member.description = 'Remove a user from the team.'
|
|
|
|
|
|
@admin.register(TeamArea)
|
|
class TeamAreaAdmin(admin.ModelAdmin):
|
|
list_filter = [
|
|
'camp'
|
|
]
|