Song reporting and user stats

marietje/songs/admin.py

@@ -1,20 +1,33 @@
 from django.contrib import admin
 
-from .models import Song
+from .models import ReportNote, Song
 
 
+class ReportNoteInline(admin.StackedInline):
+    model = ReportNote
+    extra = 0
+
 @admin.register(Song)
 class SongAdmin(admin.ModelAdmin):
-    list_display = ('artist', 'title', 'user_name')
+    list_display = ('artist', 'title', 'user_name', 'reports')
     search_fields = ('artist', 'title', 'user__name')
+    inlines = [ReportNoteInline]
+
+    def reports(self, song):
+        return ReportNote.objects.filter(song=song).count()
 
     @staticmethod
-    def user_name(obj):
+    def user_name(song):
         try:
-            return obj.user.name
+            return song.user.name
         except AttributeError:
             return '<unknown>'
 
     @staticmethod
     def get_readonly_fields(request, obj=None):
-        return [] if request.user.is_superuser else ['hash']
+        return [] if request.user.is_superuser else ['hash']
+
+@admin.register(ReportNote)
+class ReportNoteAdmin(admin.ModelAdmin):
+    list_display = ('song', 'note', 'user')
+    search_fields = ('song__artist', 'song__title', 'user__name')