mirror of https://github.com/Damillora/Altessimo
81 lines
2.7 KiB
Python
81 lines
2.7 KiB
Python
from django import forms
|
|
from django.contrib import admin
|
|
|
|
from .models import Song, OutsideSong
|
|
from artists.models import Artist
|
|
from idols.models import Idol
|
|
|
|
|
|
class SongForm(forms.ModelForm):
|
|
lyricist_str = forms.CharField(label='Lyricists', required=False)
|
|
composer_str = forms.CharField(label='Composers', required=False)
|
|
arranger_str = forms.CharField(label='Arrangers', required=False)
|
|
idols_str = forms.CharField(label='Idols', required=False)
|
|
|
|
class Meta:
|
|
model = Song
|
|
fields = [
|
|
'branch',
|
|
'title',
|
|
'romanized_title',
|
|
'lyricist_str',
|
|
'composer_str',
|
|
'arranger_str',
|
|
'idols_str',
|
|
'impression',
|
|
]
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.fields['impression'].widget.attrs['class'] = 'tm-textfield'
|
|
instance = kwargs.get("instance")
|
|
if instance:
|
|
self.fields['lyricist_str'].initial = ', '.join(
|
|
x.romanized_name for x in instance.lyricist.all())
|
|
self.fields['composer_str'].initial = ', '.join(
|
|
x.romanized_name for x in instance.composer.all())
|
|
self.fields['arranger_str'].initial = ', '.join(
|
|
x.romanized_name for x in instance.arranger.all())
|
|
self.fields['idols_str'].initial = ', '.join(
|
|
x.romanized_name for x in instance.idols.all())
|
|
|
|
|
|
class SongAdmin(admin.ModelAdmin):
|
|
form = SongForm
|
|
search_fields = ['romanized_title', 'title']
|
|
|
|
class Media:
|
|
js = (
|
|
'https://cdnjs.cloudflare.com/ajax/libs/tinymce/5.6.2/tinymce.min.js',
|
|
'tinymce-init.js'
|
|
)
|
|
|
|
def save_model(self, request, obj, form, change):
|
|
lyricist_str = form.cleaned_data.get('lyricist_str')
|
|
composer_str = form.cleaned_data.get('composer_str')
|
|
arranger_str = form.cleaned_data.get('arranger_str')
|
|
idols_str = form.cleaned_data.get('idols_str')
|
|
lyricist = Artist.objects.comma_to_qs(lyricist_str)
|
|
composer = Artist.objects.comma_to_qs(composer_str)
|
|
arranger = Artist.objects.comma_to_qs(arranger_str)
|
|
idols = Idol.objects.comma_to_qs(idols_str)
|
|
if not obj.id:
|
|
obj.save()
|
|
obj.lyricist.clear()
|
|
obj.lyricist.add(*lyricist)
|
|
obj.composer.clear()
|
|
obj.composer.add(*composer)
|
|
obj.arranger.clear()
|
|
obj.arranger.add(*arranger)
|
|
obj.idols.clear()
|
|
obj.idols.add(*idols)
|
|
obj.save()
|
|
|
|
|
|
class OutsideSongAdmin(admin.ModelAdmin):
|
|
search_fields = ['title']
|
|
|
|
|
|
admin.site.register(Song, SongAdmin)
|
|
admin.site.register(OutsideSong, OutsideSongAdmin)
|