from django.contrib import admin from django import forms from .models import Artist from categories.models import Category # Register your models here. class ArtistForm(forms.ModelForm): categories_str = forms.CharField(label='Category', required=False) aliases_str = forms.CharField(label='Aliases', required=False) class Meta: model = Artist fields = [ 'name', 'romanized_name', 'slug', 'categories_str', 'aliases_str', 'about_artist', 'about_music', ] def __init__(self, *args, **kwargs): super().__init__(*args, **kwargs) self.fields['about_artist'].widget.attrs['class'] = 'tm-textfield' self.fields['about_music'].widget.attrs['class'] = 'tm-textfield' instance = kwargs.get("instance") if instance: self.fields['categories_str'].initial = ", ".join(x.name for x in instance.category.all() ) self.fields['aliases_str'].initial = ", ".join(x.romanized_name for x in instance.aliases.all() ) class ArtistAdmin(admin.ModelAdmin): form = ArtistForm search_fields = ['romanized_name','name'] 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): categories_str = form.cleaned_data.get('categories_str') categories = Category.objects.comma_to_qs(categories_str) aliases_str = form.cleaned_data.get('aliases_str') aliases = Artist.objects.comma_to_qs(aliases_str) if not obj.id: obj.save() obj.category.clear() obj.category.add(*categories) obj.aliases.clear() obj.aliases.add(*aliases) obj.save() admin.site.register(Artist, ArtistAdmin)