spradio-server-django/savepointradio/radio/admin.py

254 lines
7.9 KiB
Python
Raw Normal View History

from django.contrib import admin, messages
from django.db import models
from django.forms import TextInput
from inline_actions.admin import InlineActionsModelAdminMixin
from .actions import change_items, publish_items, remove_items
2019-06-03 19:03:23 +00:00
from .models import Album, Artist, Game, Song, Store
from profiles.exceptions import MakeRequestError
from profiles.models import RadioProfile
class ArtistInline(admin.TabularInline):
model = Song.artists.through
verbose_name = 'artist'
verbose_name_plural = 'artists'
extra = 0
2019-06-03 19:03:23 +00:00
class StoreInline(admin.TabularInline):
model = Song.stores.through
verbose_name = 'data store'
verbose_name_plural = 'data stores'
extra = 0
@admin.register(Album)
class AlbumAdmin(admin.ModelAdmin):
# Detail List display
list_display = ('title', '_is_enabled', '_is_published')
search_fields = ['title']
2018-01-14 20:21:04 +00:00
actions = ['publish_albums']
# Edit Form display
readonly_fields = ('created_date', 'modified_date')
fieldsets = (
('Album Disabling', {
'classes': ('collapse',),
'fields': ('disabled', 'disabled_date', 'disabled_reason')
}),
('Main', {
'fields': ('title', 'published_date')
}),
('Stats', {
'classes': ('collapse',),
'fields': ('created_date', 'modified_date')
})
)
2018-01-14 20:21:04 +00:00
def publish_albums(self, request, queryset):
publish_items(request, queryset)
publish_albums.short_description = "Publish selected albums"
@admin.register(Artist)
class ArtistAdmin(admin.ModelAdmin):
# Detail List display
list_display = ('first_name',
'alias',
'last_name',
'_is_enabled',
'_is_published')
search_fields = ['first_name', 'alias', 'last_name']
2018-01-14 20:21:04 +00:00
actions = ['publish_artists']
# Edit Form display
readonly_fields = ('created_date', 'modified_date')
fieldsets = (
('Artist Disabling', {
'classes': ('collapse',),
'fields': ('disabled', 'disabled_date', 'disabled_reason')
}),
('Main', {
'fields': ('first_name', 'alias', 'last_name', 'published_date')
}),
('Stats', {
'classes': ('collapse',),
'fields': ('created_date', 'modified_date')
})
)
2018-01-14 20:21:04 +00:00
def publish_artists(self, request, queryset):
publish_items(request, queryset)
publish_artists.short_description = "Publish selected artists"
@admin.register(Game)
class GameAdmin(admin.ModelAdmin):
# Detail List display
list_display = ('title', '_is_enabled', '_is_published')
search_fields = ['title']
2018-01-14 20:21:04 +00:00
actions = ['publish_games']
# Edit Form display
readonly_fields = ('created_date', 'modified_date')
fieldsets = (
('Game Disabling', {
'classes': ('collapse',),
'fields': ('disabled', 'disabled_date', 'disabled_reason')
}),
('Main', {
'fields': ('title', 'published_date')
}),
('Stats', {
'classes': ('collapse',),
'fields': ('created_date', 'modified_date')
})
)
2018-01-14 20:21:04 +00:00
def publish_games(self, request, queryset):
publish_items(request, queryset)
publish_games.short_description = "Publish selected games"
2019-06-03 19:03:23 +00:00
@admin.register(Store)
class StoreAdmin(admin.ModelAdmin):
# Detail List display
list_display = ('iri',
'mime_type',
'file_size',
2019-07-03 15:57:46 +00:00
'length',
'_replaygain')
2019-06-03 19:03:23 +00:00
search_fields = ['iri']
# Edit Form display
readonly_fields = ('created_date', 'modified_date')
fieldsets = (
('Main', {
2019-07-03 15:57:46 +00:00
'fields': ('iri',
'mime_type',
'file_size',
'length',
'track_gain',
'track_peak')
2019-06-03 19:03:23 +00:00
}),
('Stats', {
'classes': ('collapse',),
'fields': ('created_date', 'modified_date')
})
)
@admin.register(Song)
class SongAdmin(InlineActionsModelAdminMixin, admin.ModelAdmin):
formfield_overrides = {
models.TextField: {'widget': TextInput(attrs={'size': 160, })},
}
# Detail List display
list_display = ('title',
'game',
'album',
'artist_list',
'_is_enabled',
'_is_published',
'_is_requestable')
search_fields = ['title']
actions = ['publish_songs',
'add_game', 'remove_game',
'add_album', 'remove_album',
'add_artists', 'remove_artists']
# Edit Form display
exclude = ('artists',)
2019-06-03 19:03:23 +00:00
readonly_fields = ('last_played',
'num_played',
'created_date',
'modified_date',
'next_play')
fieldsets = (
('Song Disabling', {
'classes': ('collapse',),
'fields': ('disabled', 'disabled_date', 'disabled_reason')
}),
('Main', {
'fields': ('song_type',
'title',
2019-06-03 19:03:23 +00:00
'published_date',
'active_store')
}),
('Stats', {
'classes': ('collapse',),
'fields': ('created_date',
'modified_date',
'last_played',
'num_played',
2019-06-03 19:03:23 +00:00
'next_play')
}),
('Album', {
'fields': ('album',)
}),
('Game', {
'fields': ('game',)
})
)
2019-06-03 19:03:23 +00:00
inlines = [ArtistInline, StoreInline]
inline_actions = ['request_song']
2019-06-03 19:03:23 +00:00
def formfield_for_foreignkey(self, db_field, request, **kwargs):
if db_field.name == 'active_store':
2019-06-03 19:03:23 +00:00
kwargs['queryset'] = Store.objects.filter(
song__pk=request.resolver_match.kwargs['object_id']
)
return super().formfield_for_foreignkey(db_field, request, **kwargs)
# Admin Actions
def artist_list(self, obj):
return ', '.join([a.full_name for a in obj.artists.all()])
def add_album(self, request, queryset):
return change_items(request, queryset, 'album', 'add_album')
add_album.short_description = "Add album to selected items"
def remove_album(self, request, queryset):
return remove_items(request, queryset, 'album', 'remove_album')
remove_album.short_description = "Remove album from selected items"
2018-01-13 05:39:13 +00:00
def add_artists(self, request, queryset):
return change_items(request, queryset, 'artists', 'add_artists',
m2m='artist')
add_artists.short_description = "Add artists to selected items"
def remove_artists(self, request, queryset):
return change_items(request, queryset, 'artists', 'remove_artists',
m2m='artist', remove=True)
remove_artists.short_description = "Remove artists from selected items"
def add_game(self, request, queryset):
return change_items(request, queryset, 'game', 'add_game')
add_game.short_description = "Add game to selected items"
def remove_game(self, request, queryset):
return remove_items(request, queryset, 'game', 'remove_game')
remove_game.short_description = "Remove game from selected items"
2018-01-14 20:21:04 +00:00
def publish_songs(self, request, queryset):
publish_items(request, queryset)
publish_songs.short_description = "Publish selected songs"
# Inline Admin Actions
def request_song(self, request, obj, parent_obj=None):
profile = RadioProfile.objects.get(user=request.user)
try:
profile.make_request(obj)
except MakeRequestError as e:
return messages.error(
request,
'Unable to request the song: {}'.format(str(e))
)
return messages.success(request, 'Successfully queued song.')
request_song.short_description = 'Request Song'