from .models import Post, Category
from .decorators import action_form


class PostCategoryForm(forms.Form):
    title = 'Update category for the selected posts'
    myfile = forms.FileField()
    category = forms.ModelChoiceField(queryset=Category.objects.all())


@admin.register(Post)
class PostAdmin(admin.ModelAdmin):
    actions = ['change_category']
  
    @action_form(PostCategoryForm)
    def change_category(self, request, queryset, form):
        category = form.cleaned_data['category']
        return queryset.update(category=category)