bornhack-website/src/program/admin.py

123 lines
3.0 KiB
Python

from django.contrib import admin, messages
from django.core.exceptions import ValidationError
from .models import (
Event,
EventInstance,
EventLocation,
EventProposal,
EventTrack,
EventType,
Favorite,
Speaker,
SpeakerProposal,
Url,
UrlType,
)
@admin.register(SpeakerProposal)
class SpeakerProposalAdmin(admin.ModelAdmin):
def mark_speakerproposal_as_approved(self, request, queryset):
for sp in queryset:
sp.mark_as_approved(request)
mark_speakerproposal_as_approved.description = (
"Approve and create Speaker object(s)"
)
actions = ["mark_speakerproposal_as_approved"]
list_filter = ("camp", "proposal_status", "user")
def get_speakers_string(event_proposal):
return ", ".join(event_proposal.speakers.all().values_list("email", flat=True))
get_speakers_string.short_description = "Speakers"
@admin.register(EventProposal)
class EventProposalAdmin(admin.ModelAdmin):
def mark_eventproposal_as_approved(self, request, queryset):
for ep in queryset:
if not ep.speakers.all():
messages.error(
request, "Event cant be approved as it has no speaker(s)."
)
return False
else:
try:
ep.mark_as_approved(request)
except ValidationError as e:
messages.error(request, e)
return False
mark_eventproposal_as_approved.description = "Approve and create Event object(s)"
def get_speakers(self):
return
actions = ["mark_eventproposal_as_approved"]
list_filter = ("event_type", "proposal_status", "track", "user")
list_display = ["title", get_speakers_string, "event_type", "proposal_status"]
@admin.register(EventLocation)
class EventLocationAdmin(admin.ModelAdmin):
list_filter = ("camp",)
list_display = ("name", "camp")
@admin.register(EventTrack)
class EventTrackAdmin(admin.ModelAdmin):
list_filter = ("camp",)
list_display = ("name", "camp")
@admin.register(EventInstance)
class EventInstanceAdmin(admin.ModelAdmin):
list_display = ("event", "when", "location")
list_filter = ("event__track__camp", "event")
search_fields = ["event__title"]
@admin.register(EventType)
class EventTypeAdmin(admin.ModelAdmin):
pass
@admin.register(Speaker)
class SpeakerAdmin(admin.ModelAdmin):
list_filter = ("camp",)
readonly_fields = ["proposal"]
@admin.register(Favorite)
class FavoriteAdmin(admin.ModelAdmin):
raw_id_fields = ("event_instance",)
class SpeakerInline(admin.StackedInline):
model = Speaker.events.through
@admin.register(Event)
class EventAdmin(admin.ModelAdmin):
list_filter = ("track", "speakers")
list_display = ["title", "event_type"]
inlines = [SpeakerInline]
readonly_fields = ["proposal"]
@admin.register(UrlType)
class UrlTypeAdmin(admin.ModelAdmin):
pass
@admin.register(Url)
class UrlAdmin(admin.ModelAdmin):
pass