261 lines
9.8 KiB
Python
261 lines
9.8 KiB
Python
from django.contrib import admin
|
|
from django.utils.safestring import mark_safe
|
|
from django.utils.translation import ugettext_lazy as _
|
|
from django.contrib.admin import SimpleListFilter
|
|
|
|
# Register your models here.
|
|
from .models.locale import Language, Continent, Country, SPR, City
|
|
from .models.profiles import (
|
|
UserProfile,
|
|
Organization,
|
|
OrgTeamRequest,
|
|
Team,
|
|
Member,
|
|
Category,
|
|
Topic,
|
|
Sponsor,
|
|
)
|
|
from .models.search import Searchable
|
|
from .models.events import (
|
|
Place,
|
|
Event,
|
|
EventComment,
|
|
EventSeries,
|
|
EventPhoto,
|
|
CommonEvent,
|
|
Attendee,
|
|
)
|
|
from .models.speakers import (
|
|
Speaker,
|
|
Talk,
|
|
Presentation,
|
|
SpeakerRequest,
|
|
)
|
|
|
|
admin.site.register(Language)
|
|
admin.site.register(Continent)
|
|
admin.site.register(Country)
|
|
|
|
from django.db.models import Count
|
|
class NumberOfEventsFilter(SimpleListFilter):
|
|
title = _('Event Count')
|
|
parameter_name = 'event_count'
|
|
|
|
def lookups(self, request, model_admin):
|
|
return (
|
|
('0', '0'),
|
|
('1', '1'),
|
|
('2', '2 - 9'),
|
|
('10', '10 - 99'),
|
|
('100', '100+'),
|
|
('>0', '> 0')
|
|
)
|
|
|
|
def queryset(self, request, queryset):
|
|
if self.value() == '0':
|
|
return queryset.annotate(num_events=Count('event')).filter(num_events=0)
|
|
if self.value() == '>0':
|
|
return queryset.annotate(num_events=Count('event')).filter(num_events__gte=0)
|
|
if self.value() == '1':
|
|
return queryset.annotate(num_events=Count('event')).filter(num_events=1)
|
|
if self.value() == '2':
|
|
return queryset.annotate(num_events=Count('event')).filter(num_events__gte=2, num_events__lte=9)
|
|
if self.value() == '10':
|
|
return queryset.annotate(num_events=Count('event')).filter(num_events__gte=10, num_events__lte=99)
|
|
if self.value() == '100':
|
|
return queryset.annotate(num_events=Count('event')).filter(num_events__gte=100)
|
|
|
|
class NumberOfMembersFilter(SimpleListFilter):
|
|
title = _('Member Count')
|
|
parameter_name = 'member_count'
|
|
|
|
def lookups(self, request, model_admin):
|
|
return (
|
|
('0', '0'),
|
|
('1', '1'),
|
|
('2', '2 - 9'),
|
|
('10', '10 - 99'),
|
|
('100', '100+'),
|
|
('>0', '> 0')
|
|
)
|
|
|
|
def queryset(self, request, queryset):
|
|
if self.value() == '0':
|
|
return queryset.annotate(num_events=Count('member')).filter(num_events=0)
|
|
if self.value() == '>0':
|
|
return queryset.annotate(num_events=Count('member')).filter(num_events__gt=0)
|
|
if self.value() == '1':
|
|
return queryset.annotate(num_events=Count('member')).filter(num_events=1)
|
|
if self.value() == '2':
|
|
return queryset.annotate(num_events=Count('member')).filter(num_events__gte=2, num_events__lte=9)
|
|
if self.value() == '10':
|
|
return queryset.annotate(num_events=Count('member')).filter(num_events__gte=10, num_events__lte=99)
|
|
if self.value() == '100':
|
|
return queryset.annotate(num_events=Count('member')).filter(num_events__gte=100)
|
|
|
|
class NumberOfAttendeesFilter(SimpleListFilter):
|
|
title = _('Attendee Count')
|
|
parameter_name = 'attendee_count'
|
|
|
|
def lookups(self, request, model_admin):
|
|
return (
|
|
('0', '0'),
|
|
('1', '1'),
|
|
('2', '2 - 9'),
|
|
('10', '10 - 99'),
|
|
('100', '100+'),
|
|
('>0', '> 0')
|
|
)
|
|
|
|
def queryset(self, request, queryset):
|
|
if self.value() == '0':
|
|
return queryset.annotate(num_events=Count('attendee')).filter(num_events=0)
|
|
if self.value() == '>0':
|
|
return queryset.annotate(num_events=Count('attendee')).filter(num_events__gt=0)
|
|
if self.value() == '1':
|
|
return queryset.annotate(num_events=Count('attendee')).filter(num_events=1)
|
|
if self.value() == '2':
|
|
return queryset.annotate(num_events=Count('attendee')).filter(num_events__gte=2, num_events__lte=9)
|
|
if self.value() == '10':
|
|
return queryset.annotate(num_events=Count('attendee')).filter(num_events__gte=10, num_events__lte=99)
|
|
if self.value() == '100':
|
|
return queryset.annotate(num_events=Count('attendee')).filter(num_events__gte=100)
|
|
|
|
class SPRAdmin(admin.ModelAdmin):
|
|
raw_id_fields = ('country',)
|
|
list_filter =('country',)
|
|
search_fields = ('name', 'country__name')
|
|
admin.site.register(SPR, SPRAdmin)
|
|
|
|
class CityAdmin(admin.ModelAdmin):
|
|
raw_id_fields = ('spr',)
|
|
list_display = ('name', 'spr', 'latitude', 'longitude', 'population')
|
|
list_filter =('spr__country',)
|
|
search_fields = ('name', 'spr__name')
|
|
admin.site.register(City, CityAdmin)
|
|
|
|
class ProfileAdmin(admin.ModelAdmin):
|
|
raw_id_fields = ('city',)
|
|
list_display = ('user', 'realname', 'city', 'web_url', 'send_notifications', 'do_not_track')
|
|
list_filter = ('send_notifications', 'do_not_track', 'user__last_login')
|
|
admin.site.register(UserProfile, ProfileAdmin)
|
|
|
|
class OrgAdmin(admin.ModelAdmin):
|
|
list_display = ('name', 'slug', 'team_count', 'owner_profile', 'site')
|
|
def team_count(self, org):
|
|
return org.teams.all().count()
|
|
team_count.short_description = 'Teams'
|
|
admin.site.register(Organization, OrgAdmin)
|
|
|
|
class OrgRequestAdmin(admin.ModelAdmin):
|
|
list_display = ('organization', 'team', 'request_origin', 'requested_by', 'requested_date', 'accepted_by', 'joined_date')
|
|
list_filter = ('organization', 'request_origin')
|
|
admin.site.register(OrgTeamRequest, OrgRequestAdmin)
|
|
|
|
class SponsorAdmin(admin.ModelAdmin):
|
|
list_display = ('name', 'web_url')
|
|
admin.site.register(Sponsor, SponsorAdmin)
|
|
|
|
class TeamAdmin(admin.ModelAdmin):
|
|
raw_id_fields = ('country', 'spr', 'city', 'owner_profile', 'admin_profiles', 'contact_profiles', 'sponsors')
|
|
list_display = ('__str__', 'active', 'member_count', 'event_count', 'owner_profile', 'created_date', 'access')
|
|
list_filter = ('access', 'organization', NumberOfMembersFilter, NumberOfEventsFilter, ('country',admin.RelatedOnlyFieldListFilter), 'active')
|
|
ordering = ('-created_date',)
|
|
|
|
def member_count(self, team):
|
|
return team.members.all().count()
|
|
member_count.short_description = 'Members'
|
|
def event_count(self, team):
|
|
return team.event_set.all().count()
|
|
event_count.short_description = 'Events'
|
|
admin.site.register(Team, TeamAdmin)
|
|
|
|
class SearchableAdmin(admin.ModelAdmin):
|
|
list_display = ('event_url', 'start_time', 'federation_node', 'federation_time')
|
|
list_filter = ('federation_node',)
|
|
ordering = ('-start_time',)
|
|
admin.site.register(Searchable, SearchableAdmin)
|
|
|
|
class PlaceAdmin(admin.ModelAdmin):
|
|
raw_id_fields = ('city',)
|
|
admin.site.register(Place, PlaceAdmin)
|
|
|
|
class EventAdmin(admin.ModelAdmin):
|
|
raw_id_fields = ('place', 'created_by', 'sponsors')
|
|
list_display = ('__str__', 'attendee_count', 'start_time', 'created_by', 'created_time')
|
|
list_filter = ('created_time', 'start_time', NumberOfAttendeesFilter, ('team__country',admin.RelatedOnlyFieldListFilter))
|
|
ordering = ('-start_time',)
|
|
def attendee_count(self, event):
|
|
return event.attendees.all().count()
|
|
attendee_count.short_description = 'Attendees'
|
|
admin.site.register(Event, EventAdmin)
|
|
|
|
class EventPhotoAdmin(admin.ModelAdmin):
|
|
raw_id_fields = ('event',)
|
|
list_display = ('title', 'event', 'view')
|
|
def view(self, photo):
|
|
return mark_safe('<a href="%s" target="_blank"><img src="%s" height="90px"></a>' % (photo.src.url, photo.thumbnail.url))
|
|
view.short_description = 'Photo'
|
|
admin.site.register(EventPhoto, EventPhotoAdmin)
|
|
|
|
class EventCommentAdmin(admin.ModelAdmin):
|
|
raw_id_fields = ('event', 'author')
|
|
list_display = ('event', 'author', 'status', 'created_time')
|
|
admin.site.register(EventComment, EventCommentAdmin)
|
|
|
|
class CommonEventAdmin(admin.ModelAdmin):
|
|
raw_id_fields = ('place', 'city', 'spr', 'country')
|
|
list_display = ('__str__', 'participant_count', 'organization', 'start_time', 'country', 'spr', 'city')
|
|
ordering = ('-start_time',)
|
|
def participant_count(self, event):
|
|
return event.participating_events.all().count()
|
|
participant_count.short_description = 'Participants'
|
|
admin.site.register(CommonEvent, CommonEventAdmin)
|
|
|
|
class EventSeriesAdmin(admin.ModelAdmin):
|
|
raw_id_fields = ('place', 'team')
|
|
list_display = ('__str__', 'instance_count', 'team', 'start_time', 'last_time')
|
|
ordering = ('-last_time',)
|
|
def instance_count(self, series):
|
|
return series.instances.all().count()
|
|
instance_count.short_description = 'Instances'
|
|
admin.site.register(EventSeries, EventSeriesAdmin)
|
|
|
|
class MemberAdmin(admin.ModelAdmin):
|
|
list_display = ('__str__', 'role', 'joined_date')
|
|
list_filter = ('role', 'team', 'joined_date')
|
|
admin.site.register(Member, MemberAdmin)
|
|
|
|
class AttendeeAdmin(admin.ModelAdmin):
|
|
list_display = ('__str__', 'role', 'status', 'joined_date', 'last_reminded')
|
|
list_filter = ('role', 'status', 'joined_date')
|
|
admin.site.register(Attendee, AttendeeAdmin)
|
|
|
|
class CategoryAdmin(admin.ModelAdmin):
|
|
list_display = ('name', 'slug', 'image')
|
|
exclude = ('slug', )
|
|
def image(self, obj):
|
|
return (mark_safe('<img src="%s" title="%s" height="64px" />' % (obj.img_url, obj.name)))
|
|
image.short_description = 'Image'
|
|
admin.site.register(Category, CategoryAdmin)
|
|
|
|
class TopicAdmin(admin.ModelAdmin):
|
|
list_display = ('name', 'slug', 'category')
|
|
list_filter = ('category',)
|
|
exclude = ('slug', )
|
|
admin.site.register(Topic, TopicAdmin)
|
|
|
|
class SpeakerAdmin(admin.ModelAdmin):
|
|
list_display = ('title', 'user', 'avatar')
|
|
admin.site.register(Speaker, SpeakerAdmin)
|
|
|
|
class TalkAdmin(admin.ModelAdmin):
|
|
list_display = ('title', 'speaker', 'category')
|
|
list_filter = ('category',)
|
|
admin.site.register(Talk, TalkAdmin)
|
|
|
|
class PresentationAdmin(admin.ModelAdmin):
|
|
list_display = ('talk', 'status', 'event')
|
|
list_filter = ('status',)
|
|
admin.site.register(Presentation, PresentationAdmin)
|
|
|