Django 在 CreateView 和 UpdateView 中调用自定义管理器方法

Question

your expert knowledge of Django demanded:smile: So here is the problem. I created a custom create_number manager method. I wanted to call this on form submit when user fill in number.

class Number(models.Model):
    user = models.ForeignKey(User, related_name="numbers")
    number = models.PositiveIntegerField()
    verification_code = models.CharField(max_length=6)
    is_verified = models.BooleanField(default=False)
    objects = NumberQueryset.as_manager()

class NumberQueryset(models.QuerySet):
    def create_number(self, number, user):
        if number_validator(number):
             if self.filter(is_verified=True, user=user, number=number).exists():
                  raise ValidationError("Number already verified")
             else:
                  verification_code = generate_random()
                  self.create(user=user, number=number, verification_code=verification_code, operator=get_operator(number), valid_until = timezone.now()+timezone.timedelta(seconds=VALID_UNTIL_CONSTANT))

So in a views.py I tried using CreateView and created a form with following. Now user redirects to Verify view and fill in verification code, if verification code is expired how can i show error on that page?

class NumberRegister(LoginRequiredMixin, CreateView):
    model = Number
    fields = ['number']

    def form_valid(self, form):
        if form.is_valid:
           Number.objects.create_number(number=form.cleaned_data['number', user=self.request.user .....)
           redirect(reverse('verify', kwargs={'number': form.cleaned_data['number']}))


class Verify(LoginRequiredMixin, UpdateView):

Answer 1

You have to add a custom manager like below

class NumberManager(models.Manager):

    def get_queryset(self):
        return NumberQueryset(self.model, using=self._db)

and on bottom of your model call it like so

objects = NumberManager()

and call it in your views.