Django generic create view . Можем ли мы добавить условия для сохранения данных из формы?

Вот как выглядит мой код, и я хочу добавить некоторое условие в представление создания заемщика, например, если метод stock для книги возвращает 0, то не перечислять эту книгу в поле при создании нового заемщика или, если это невозможно, по крайней мере, выдать ошибку при добавлении заемщика к этой книге.

models.py:

class Book(models.Model):
    id = models.UUIDField(primary_key=True, unique=True,
                          default=uuid.uuid4, editable=False)
    title = models.CharField(max_length=200)
    author = models.CharField(max_length=100)
    summary = models.TextField(
        max_length=1000, help_text="Enter a brief description of the book")
    isbn = models.CharField('ISBN', max_length=13,
                            help_text='13 Character https://www.isbn-international.org/content/what-isbn')
    genre = models.ManyToManyField(
        Genre, help_text="Select a genre for this book")
    language = models.ForeignKey(
        'Language', on_delete=models.SET_NULL, null=True)
    total_copies = models.IntegerField()
    pic = models.ImageField(blank=True, null=True, upload_to='books')

    def stock(self):
        total_copies = self.total_copies
        available_copies = total_copies - \
            Borrower.objects.filter(book=self).count()
        if available_copies > 0:
            return available_copies

        else:
            return 0

    
    def __str__(self):
        return self.title


class Borrower(models.Model):
    id = models.UUIDField(primary_key=True, unique=True,
                          default=uuid.uuid4, editable=False)
    student = models.ForeignKey('Account', on_delete=models.CASCADE)
    book = models.ForeignKey('Book', on_delete=models.CASCADE)
    issue_date = models.DateField(
        null=True, blank=True, help_text='YYYY-MM-DD', default=date.today)
    return_date = models.DateField(
        null=True, blank=True, help_text='YYYY-MM-DD')

    def __str__(self):
        return self.student.name.title()+" borrowed "+self.book.title.title()

    def fine(self):
        today = date.today()
        fine = 0
        if self.return_date <= today:
            fine += 5 * (today - self.return_date).days
        return fine

views.py:

class BorrowerView(LoginRequiredMixin, ListView):
    model=Borrower
    context_object_name='borrowers'
    template_name = 'library/borrower_list.html'


    def get_context_data(self, **kwargs):
        context=super().get_context_data(**kwargs)
        if self.request.user.is_admin or self.request.user.is_superuser:
            context['borrowers']=context['borrowers']
        else:
            context['borrowers']=context['borrowers'].filter(student = self.request.user.id)
        return context



class BorrowerCreate(LoginRequiredMixin, UserAccessMixin, CreateView):
    model=Borrower
    permission_required= 'borrowers.add_borrowers'
    fields='__all__'
    success_url=reverse_lazy('library:borrower-list')
    
    def form_valid(self, form):
        form.instance.user=self.request.user
        return super(BorrowerCreate, self).form_valid(form)



class BorrowerDetail(LoginRequiredMixin,  DetailView):
    model=Borrower()
    context_object_name='borrower'
    template_name='library/borrower.html'

Ист: Вместо того чтобы определять новый метод, который вы назвали stock, почему бы не добавить stock как поле, вместо того чтобы делать запрос к базе данных.

2-й: вызов класса внутри того же класса - не лучший способ делать запросы внутри класса.

3-й: Чтобы добавить условие при добавлении объекта, необходимо переопределить метод сохранения объекта следующим образом.

def save(self, *args, **kwargs):
    # do_something() here.....
    # then
    return super().save(*args, **kwargs)

Приведенный выше код позволит вам выполнить любое действие перед сохранением объекта.

Другой способ сделать это - внутри функции form_valid следующим образом.

def form_valid(self, form):
    instance = form.save(commit=False)
    # commit = False will make sure that the form is not saved
    # then you can now query the database and check conditions like
    if Borrower.object.all().count() > 0:
       instance.save()
       messages.success(self.request, _("saved successfully")
    else:
       messages.error(self.request, _("Error")
    return redirect("URL")
Вернуться на верх