AFPy/PonyConf
AFPy/PonyConf
7
2
Fork 3
Code Issues 10 Pull Requests 1 Projects Releases Wiki Activity

improving proposals admin

Browse Source
This commit is contained in:
Élie Bouttier 2016-10-18 22:40:56 +02:00
parent 21b53f3091
commit 0ae9b5cb4b
1 changed files with 34 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
proposals/admin.py
View File

@ -1,34 +1,51 @@
from django.contrib import admin from django.contrib import admin
from django.contrib.sites.shortcuts import get_current_site
from proposals.models import Conference, Talk, Topic, Track, Event from proposals.models import Conference, Talk, Topic, Track, Event
from planning.models import Room
class TalkAdmin(admin.ModelAdmin): class SiteAdminMixin:
# Disable add button in django admin has it is too dangerous exclude = ('site',)
# (it is easy to obtain incoherent data due to site framework)
def has_add_permission(self, request): def get_queryset(self, request):
return False return super().get_queryset(request).filter(site=get_current_site(request))
# Filter for 'on site' topics, tracks and events
def save_model(self, request, obj, form, change):
if not change:
obj.site = get_current_site(request)
super().save_model(request, obj, form, change)
class TalkAdmin(SiteAdminMixin, admin.ModelAdmin):
def get_form(self, request, obj=None, **kwargs): def get_form(self, request, obj=None, **kwargs):
form = super(TalkAdmin, self).get_form(request, obj, **kwargs) form = super(TalkAdmin, self).get_form(request, obj, **kwargs)
# in fact, obj should never be none as 'add' button is disabled site = get_current_site(request)
if obj: form.base_fields['topics'].queryset = Topic.objects.filter(site=site)
form.base_fields['topics'].queryset = Topic.objects.filter(site=obj.site) form.base_fields['track'].queryset = Track.objects.filter(site=site)
form.base_fields['track'].queryset = Track.objects.filter(site=obj.site) form.base_fields['event'].queryset = Event.objects.filter(site=site)
form.base_fields['event'].queryset = Event.objects.filter(site=obj.site) form.base_fields['room'].queryset = Room.objects.filter(site=site)
return form return form
class TopicAdmin(admin.ModelAdmin): class TopicAdmin(SiteAdminMixin, admin.ModelAdmin):
# Filter for 'on site' tracks
def get_form(self, request, obj=None, **kwargs): def get_form(self, request, obj=None, **kwargs):
form = super().get_form(request, obj, **kwargs) form = super().get_form(request, obj, **kwargs)
if obj: site = get_current_site(request)
form.base_fields['track'].queryset = Track.objects.filter(site=obj.site) form.base_fields['track'].queryset = Track.objects.filter(site=site)
return form return form
class TrackAdmin(SiteAdminMixin, admin.ModelAdmin):
pass
class EventAdmin(SiteAdminMixin, admin.ModelAdmin):
pass
admin.site.register(Conference) admin.site.register(Conference)
admin.site.register(Topic, TopicAdmin) admin.site.register(Topic, TopicAdmin)
admin.site.register(Track) admin.site.register(Track, TrackAdmin)
admin.site.register(Talk, TalkAdmin) admin.site.register(Talk, TalkAdmin)
admin.site.register(Event) admin.site.register(Event, EventAdmin)