Django：在Graphene中重写Django的选择输出

在本文中，我们将介绍如何在Graphene中重写Django模型中定义的选择字段的输出。对于使用Django和Graphene构建API的开发人员来说，这是一个常见的需求。我们将通过示例来演示如何在Graphene中自定义选择字段的输出。

1. Django模型中的选择字段

在Django模型中，选择字段代表一系列可用选项，通常用于限制用户可以选择的值。例如，考虑一个表示订单的模型，其中有一个支付状态字段，可以选择已支付、未支付或部分支付。在这种情况下，我们可以使用Django的ChoiceField来定义这个字段，并指定可用的选项。

下面是一个示例订单模型定义：

from django.db import models

class Order(models.Model):
    PAYMENT_STATUS_CHOICES = (
        ('paid', '已支付'),
        ('unpaid', '未支付'),
        ('partial', '部分支付')
    )

    payment_status = models.CharField(
        max_length=10,
        choices=PAYMENT_STATUS_CHOICES,
        default='unpaid'
    )

在上述示例中，我们定义了一个订单模型，其中payment_status字段是一个选择字段，它可以选择”已支付”、”未支付”或”部分支付”。

2. 使用Graphene定义模型类型

现在我们将使用Graphene来定义与上面订单模型对应的类型。在Graphene中，我们可以使用自定义标量类型来自定义选择字段的输出。

下面是使用Graphene定义订单类型的示例代码：

from graphene import ObjectType, String
from graphene_django import DjangoObjectType

from .models import Order

class OrderType(DjangoObjectType):
    payment_status = String()

    class Meta:
        model = Order

    def resolve_payment_status(self, info):
        return dict(Order.PAYMENT_STATUS_CHOICES).get(self.payment_status)

在这个示例中，我们定义了一个OrderType，用于表示订单类型。我们手动添加了一个名为payment_status的字段。然后，我们在resolve_payment_status()方法中返回该字段的值。

3. 定义GraphQL查询

现在我们已经定义了自定义的OrderType，我们需要定义与之对应的GraphQL查询。我们将使用Graphene的ObjectType基类来定义查询类型，并在其中包含一个返回所有订单的字段。

下面是一个简单的GraphQL查询示例，其中返回所有订单的字段为”orders”：

from graphene import ObjectType, List
from .types import OrderType
from .models import Order

class Query(ObjectType):
    orders = List(OrderType)

    def resolve_orders(self, info, **kwargs):
        return Order.objects.all()

在上述示例中，我们定义了一个Query类，它包含一个名为”orders”的字段，该字段的类型是一个列表，其中包含了OrderType类型的对象。在resolve_orders()方法中，我们只需返回数据库中的所有订单即可。

4. 测试GraphQL查询

现在我们已经定义了模型类型、查询和解析器，我们可以通过运行GraphQL查询来测试我们的代码。

假设我们有三个订单数据，其中一个订单状态为”paid”，另一个为”unpaid”，最后一个为”partial”，现在我们可以运行以下GraphQL查询来获取订单数据：

query {
  orders {
    paymentStatus
  }
}

上述查询将返回一个包含所有订单的列表，并包括每个订单的paymentStatus字段。

输出示例：

{
  "data": {
    "orders": [
      {
        "paymentStatus": "已支付"
      },
      {
        "paymentStatus": "未支付"
      },
      {
        "paymentStatus": "部分支付"
      }
    ]
  }
}