bornhack-website/src/teams/admin.py

from django.contrib import admin

from camps.utils import CampPropertyListFilter

from .email import add_added_membership_email, add_removed_membership_email
from .models import Team, TeamMember, TeamShift, TeamTask


@admin.register(TeamTask)
class TeamTaskAdmin(admin.ModelAdmin):
    list_display = ["id", "team", "name", "description"]


class TeamMemberInline(admin.TabularInline):
    model = TeamMember


@admin.register(Team)
class TeamAdmin(admin.ModelAdmin):
    def get_responsible(self, obj):
        return ", ".join(
            [resp.profile.public_credit_name for resp in obj.responsible_members.all()]
        )

    get_responsible.short_description = "Responsible"

    list_display = [
        "name",
        "camp",
        "get_responsible",
        "needs_members",
        "public_irc_channel_name",
        "public_irc_channel_bot",
        "public_irc_channel_managed",
        "private_irc_channel_name",
        "private_irc_channel_bot",
        "private_irc_channel_managed",
    ]

    list_filter = [
        CampPropertyListFilter,
        "needs_members",
        "public_irc_channel_bot",
        "public_irc_channel_managed",
        "private_irc_channel_bot",
        "private_irc_channel_managed",
    ]
    inlines = [TeamMemberInline]


@admin.register(TeamMember)
class TeamMemberAdmin(admin.ModelAdmin):
    list_filter = [CampPropertyListFilter, "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(TeamShift)
class TeamShiftAdmin(admin.ModelAdmin):
    list_filter = ["team"]