2

在我正在构建的应用程序中，我创建了一系列自定义模型管理器和查询集以拥有更高级别的 api。

当我执行管理操作时出现问题。传递给它的查询集似乎是一个通用查询集，我希望能够访问我的自定义查询集，以便能够使用我在其中创建的过滤功能。

这是动作：

def mark_payment_as_sent_action():
    ''' Admin action to mark payment as sent '''

    def mark_payment_as_sent(modeladmin, request, queryset):

        # #####################################################################
        # This is what I currently do
        payments = queryset.filter(status=models.Payment.S_PENDING)

        # This is what I want to do
        payments = queryset.pending()
        # #####################################################################

        # Do stuff with filtered payments

        return HttpResponseRedirect("...")

    mark_payment_as_sent.short_description = "Mark as sent"
    return mark_payment_as_sent

这些是自定义模型管理器的查询集：

class PaymentQuerySet(models.query.QuerySet):

    def pending(self):
        return self.filter(status=self.model.S_PENDING)

class PaymentManager(models.Manager):
    use_for_related_fields = True

    def get_query_set(self):
        return PaymentQuerySet(self.model)

    def pending(self, *args, **kwargs):
        return self.get_query_set().pending(*args, **kwargs)

最后是模型和管理类：

class Payment(models.Model):

    status = models.CharField(
        max_length=25,
        choices=((S_PENDING, 'Pending'), ...)
    )

    objects = managers.PaymentManager()

@admin.register(models.Payment)
class PaymentsAdmin(admin.ModelAdmin):

    actions = (
        admin_actions.mark_payment_as_sent_action(),
    )

关于如何在调用管理操作时告诉 Django 使用我的查询集的任何提示？

非常感谢。

4

1 回答 1

0

正如 noamk 所指出的，问题出在方法名称上。Django 将该get_query_set方法重命名为get_queryset.

现在它工作得很好。

class PaymentQuerySet(models.query.QuerySet):

    def pending(self):
        return self.filter(status=self.model.S_PENDING)

class PaymentManager(models.Manager):
    use_for_related_fields = True

    def get_queryset(self):
        return PaymentQuerySet(self.model)

    def pending(self):
        return self.get_queryset().pending()
于 2016-02-16T15:34:20.027 回答