code quality

aircox/admin/episode.py

@@ -1,78 +1,83 @@
+from adminsortable2.admin import SortableAdminBase
 from django.contrib import admin
 from django.forms import ModelForm
 from django.utils.translation import gettext as _
-from adminsortable2.admin import SortableAdminBase
-
-from ..models import Episode, Diffusion
 
+from ..models import Diffusion, Episode
 from .page import PageAdmin
 from .sound import SoundInline, TrackInline
 
 
 class DiffusionBaseAdmin:
-    fields = ('type', 'start', 'end', 'schedule')
-    readonly_fields = ('schedule',)
+    fields = ("type", "start", "end", "schedule")
+    readonly_fields = ("schedule",)
 
     def get_readonly_fields(self, request, obj=None):
         fields = super().get_readonly_fields(request, obj)
-        if not request.user.has_perm('aircox_program.scheduling'):
-            fields = fields + ('program', 'start', 'end')
+        if not request.user.has_perm("aircox_program.scheduling"):
+            fields = fields + ("program", "start", "end")
         return [field for field in fields if field in self.fields]
 
 
 @admin.register(Diffusion)
 class DiffusionAdmin(DiffusionBaseAdmin, admin.ModelAdmin):
     def start_date(self, obj):
-        return obj.local_start.strftime('%Y/%m/%d %H:%M')
-    start_date.short_description = _('start')
+        return obj.local_start.strftime("%Y/%m/%d %H:%M")
+
+    start_date.short_description = _("start")
 
     def end_date(self, obj):
-        return obj.local_end.strftime('%H:%M')
-    end_date.short_description = _('end')
+        return obj.local_end.strftime("%H:%M")
 
-    list_display = ('episode', 'start_date', 'end_date', 'type', 'initial')
-    list_filter = ('type', 'start', 'program')
-    list_editable = ('type',)
-    ordering = ('-start', 'id')
+    end_date.short_description = _("end")
 
-    fields = ('type', 'start', 'end', 'initial', 'program', 'schedule')
-    readonly_fields = ('schedule',)
+    list_display = ("episode", "start_date", "end_date", "type", "initial")
+    list_filter = ("type", "start", "program")
+    list_editable = ("type",)
+    ordering = ("-start", "id")
+
+    fields = ("type", "start", "end", "initial", "program", "schedule")
+    readonly_fields = ("schedule",)
 
 
 class DiffusionInline(DiffusionBaseAdmin, admin.TabularInline):
     model = Diffusion
-    fk_name = 'episode'
+    fk_name = "episode"
     extra = 0
 
     def has_add_permission(self, request, obj):
-        return request.user.has_perm('aircox_program.scheduling')
+        return request.user.has_perm("aircox_program.scheduling")
 
 
 class EpisodeAdminForm(ModelForm):
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
-        self.fields['parent'].required = True
+        self.fields["parent"].required = True
 
 
 @admin.register(Episode)
 class EpisodeAdmin(SortableAdminBase, PageAdmin):
     form = EpisodeAdminForm
     list_display = PageAdmin.list_display
-    list_filter = tuple(f for f in PageAdmin.list_filter
-                        if f != 'pub_date') + ('diffusion__start', 'pub_date')
-    search_fields = PageAdmin.search_fields + ('parent__title',)
+    list_filter = tuple(
+        f for f in PageAdmin.list_filter if f != "pub_date"
+    ) + (
+        "diffusion__start",
+        "pub_date",
+    )
+    search_fields = PageAdmin.search_fields + ("parent__title",)
     # readonly_fields = ('parent',)
 
     inlines = [TrackInline, SoundInline, DiffusionInline]
 
-    def add_view(self, request, object_id, form_url='', context=None):
+    def add_view(self, request, object_id, form_url="", context=None):
         context = context or {}
-        context['init_app'] = True
-        context['init_el'] = '#inline-tracks'
+        context["init_app"] = True
+        context["init_el"] = "#inline-tracks"
         return super().change_view(request, object_id, form_url, context)
 
-    def change_view(self, request, object_id, form_url='', context=None):
+    def change_view(self, request, object_id, form_url="", context=None):
         context = context or {}
-        context['init_app'] = True
-        context['init_el'] = '#inline-tracks'
+        context["init_app"] = True
+        context["init_el"] = "#inline-tracks"
         return super().change_view(request, object_id, form_url, context)