From 3afc5c7a3a6350cd0df13a3eb90e84ade03d36df Mon Sep 17 00:00:00 2001
From: RecursiveGreen <recursive.green@gmail.com>
Date: Mon, 3 Jun 2019 15:03:23 -0400
Subject: [PATCH] Implemented Store into admin pages.

---
 savepointradio/radio/admin.py | 50 +++++++++++++++++++++++++++++------
 1 file changed, 42 insertions(+), 8 deletions(-)

diff --git a/savepointradio/radio/admin.py b/savepointradio/radio/admin.py
index e5fefa4..2a10ede 100644
--- a/savepointradio/radio/admin.py
+++ b/savepointradio/radio/admin.py
@@ -3,7 +3,7 @@ from django.db import models
 from django.forms import TextInput
 
 from .actions import change_items, publish_items, remove_items
-from .models import Album, Artist, Game, Song
+from .models import Album, Artist, Game, Song, Store
 
 
 class ArtistInline(admin.TabularInline):
@@ -13,6 +13,13 @@ class ArtistInline(admin.TabularInline):
     extra = 0
 
 
+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
@@ -101,6 +108,28 @@ class GameAdmin(admin.ModelAdmin):
     publish_games.short_description = "Publish selected games"
 
 
+@admin.register(Store)
+class StoreAdmin(admin.ModelAdmin):
+    # Detail List display
+    list_display = ('iri',
+                    'mime_type',
+                    'file_size',
+                    'length')
+    search_fields = ['iri']
+
+    # Edit Form display
+    readonly_fields = ('created_date', 'modified_date')
+    fieldsets = (
+        ('Main', {
+            'fields': ('iri', 'mime_type', 'file_size', 'length')
+        }),
+        ('Stats', {
+            'classes': ('collapse',),
+            'fields': ('created_date', 'modified_date')
+        })
+    )
+
+
 @admin.register(Song)
 class SongAdmin(admin.ModelAdmin):
     formfield_overrides = {
@@ -123,8 +152,7 @@ class SongAdmin(admin.ModelAdmin):
 
     # Edit Form display
     exclude = ('artists',)
-    readonly_fields = ('length',
-                       'last_played',
+    readonly_fields = ('last_played',
                        'num_played',
                        'created_date',
                        'modified_date',
@@ -137,8 +165,8 @@ class SongAdmin(admin.ModelAdmin):
         ('Main', {
             'fields': ('song_type',
                        'title',
-                       'path',
-                       'published_date')
+                       'published_date',
+                       'current_store')
         }),
         ('Stats', {
             'classes': ('collapse',),
@@ -146,8 +174,7 @@ class SongAdmin(admin.ModelAdmin):
                        'modified_date',
                        'last_played',
                        'num_played',
-                       'next_play',
-                       'length')
+                       'next_play')
         }),
         ('Album', {
             'fields': ('album',)
@@ -156,7 +183,14 @@ class SongAdmin(admin.ModelAdmin):
             'fields': ('game',)
         })
     )
-    inlines = [ArtistInline]
+    inlines = [ArtistInline, StoreInline]
+
+    def formfield_for_foreignkey(self, db_field, request, **kwargs):
+        if db_field.name == 'current_store':
+            kwargs['queryset'] = Store.objects.filter(
+                song__pk=request.resolver_match.kwargs['object_id']
+            )
+        return super().formfield_for_foreignkey(db_field, request, **kwargs)
 
     def artist_list(self, obj):
         return ', '.join([a.full_name for a in obj.artists.all()])