code quality

This commit is contained in:
bkfox
2023-03-13 17:47:00 +01:00
parent 934817da8a
commit 112770eddf
162 changed files with 4798 additions and 4069 deletions

View File

@ -1,5 +1,3 @@
from copy import copy
from django.contrib import admin
from django.forms import ModelForm
from django.utils.translation import gettext_lazy as _
@ -14,20 +12,20 @@ class ScheduleInlineForm(ModelForm):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
if self.initial:
self.fields['date'].disabled = True
self.fields['frequency'].disabled = True
self.fields["date"].disabled = True
self.fields["frequency"].disabled = True
class ScheduleInline(admin.TabularInline):
model = Schedule
form = ScheduleInlineForm
readonly_fields = ('timezone',)
readonly_fields = ("timezone",)
extra = 1
class StreamInline(admin.TabularInline):
model = Stream
fields = ['delay', 'begin', 'end']
fields = ["delay", "begin", "end"]
extra = 1
@ -39,20 +37,23 @@ class ProgramAdmin(PageAdmin):
schedule.boolean = True
schedule.short_description = _("Schedule")
list_display = PageAdmin.list_display + ('schedule', 'station', 'active')
list_filter = PageAdmin.list_filter + ('station', 'active')
prepopulated_fields = {'slug': ('title',)}
search_fields = ('title',)
list_display = PageAdmin.list_display + ("schedule", "station", "active")
list_filter = PageAdmin.list_filter + ("station", "active")
prepopulated_fields = {"slug": ("title",)}
search_fields = ("title",)
inlines = [ScheduleInline, StreamInline]
def get_fieldsets(self, request, obj=None):
fields = super().get_fieldsets(request, obj)
if request.user.has_perm('aircox.program.scheduling'):
if request.user.has_perm("aircox.program.scheduling"):
fields = fields + [
(_('Program Settings'), {
'fields': ['active', 'station', 'sync'],
})
(
_("Program Settings"),
{
"fields": ["active", "station", "sync"],
},
)
]
return fields
@ -61,26 +62,32 @@ class ProgramAdmin(PageAdmin):
class ScheduleAdmin(admin.ModelAdmin):
def program_title(self, obj):
return obj.program.title
program_title.short_description = _('Program')
program_title.short_description = _("Program")
def freq(self, obj):
return obj.get_frequency_verbose()
freq.short_description = _('Day')
list_filter = ['frequency', 'program']
list_display = ['program_title', 'freq', 'time', 'timezone', 'duration',
'initial']
list_editable = ['time', 'duration', 'initial']
freq.short_description = _("Day")
list_filter = ["frequency", "program"]
list_display = [
"program_title",
"freq",
"time",
"timezone",
"duration",
"initial",
]
list_editable = ["time", "duration", "initial"]
def get_readonly_fields(self, request, obj=None):
if obj:
return ['program', 'date', 'frequency']
return ["program", "date", "frequency"]
else:
return []
@admin.register(Stream)
class StreamAdmin(admin.ModelAdmin):
list_display = ('id', 'program', 'delay', 'begin', 'end')
list_display = ("id", "program", "delay", "begin", "end")