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
|
from django.contrib import admin
from .utils import (
get_invitation_admin_add_form,
get_invitation_admin_change_form,
get_invitation_model,
)
Invitation = get_invitation_model()
InvitationAdminAddForm = get_invitation_admin_add_form()
InvitationAdminChangeForm = get_invitation_admin_change_form()
@admin.register(Invitation)
class InvitationAdmin(admin.ModelAdmin):
list_display = ("email", "sent", "accepted")
raw_id_fields = ("inviter",)
def get_form(self, request, obj=None, **kwargs):
if obj:
kwargs["form"] = InvitationAdminChangeForm
else:
kwargs["form"] = InvitationAdminAddForm
kwargs["form"].user = request.user
kwargs["form"].request = request
return super().get_form(request, obj, **kwargs)
|
