00af109e2f
* add flake8 to pre-commit config, and fixup many things to make flake8 happy * add isort and sort all imports, add to pre-commit and requirements
95 lines
2.5 KiB
Python
95 lines
2.5 KiB
Python
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"]
|