Django与Graphene集成Enum时始终返回null问题求助

我之前在Django 2.x搭配Graphene 2.x的项目里也踩过这个Enum适配的坑，结合你描述的情况——值能正常存到SQLite也能读取，但Graphene持续报错，大概率是Graphene对原生Python Enum的类型映射没跟上Django的处理逻辑。毕竟Django 2.0+已经原生支持Python Enum作为字段选项，但Graphene 2.x的自动类型推导还没覆盖这种场景。

核心问题分析 #

Django本身可以通过choices参数把Python Enum映射到CharField，完成数据的持久化和读取，但Graphene在生成GraphQL schema时，无法自动识别模型字段关联的Python Enum，会把它当成普通的字符串字段处理，或者找不到对应的GraphQL Enum类型，从而抛出类型不匹配的错误。

具体解决方案 #

下面是针对你的环境（Django 2.0.3、Graphene 2.0.1）的分步解决方法：

1. 确保模型层的Enum定义正确

首先确认你的models.py里Enum和字段的定义是规范的，比如用CharField配合Enum的选项：

# models.py
from django.db import models
from enum import Enum

# 定义Python Enum
class TaskStatus(Enum):
    PENDING = "pending"
    COMPLETED = "completed"
    CANCELLED = "cancelled"

# 定义Django模型
class Task(models.Model):
    title = models.CharField(max_length=100)
    status = models.CharField(
        max_length=20,
        # 将Enum转换为Django需要的choices格式
        choices=[(e.value, e.name) for e in TaskStatus],
        default=TaskStatus.PENDING.value
    )

2. 在GraphQL Schema中手动映射Graphene Enum

Graphene 2.x不会自动生成对应Python Enum的GraphQL类型，所以需要我们手动定义，并绑定到对应的模型字段上：

# schema.py
import graphene
from graphene_django import DjangoObjectType
from .models import Task, TaskStatus

# 定义与Python Enum对应的Graphene Enum
class GrapheneTaskStatus(graphene.Enum):
    PENDING = TaskStatus.PENDING.value
    COMPLETED = TaskStatus.COMPLETED.value
    CANCELLED = TaskStatus.CANCELLED.value

    # 可选：添加从Python Enum到Graphene Enum的转换方法
    @classmethod
    def from_enum(cls, enum_instance):
        return cls(enum_instance.value)

# 定义模型对应的GraphQL类型，指定status字段使用自定义的Graphene Enum
class TaskType(DjangoObjectType):
    # 手动指定字段类型为我们定义的Graphene Enum
    status = GrapheneTaskStatus()

    class Meta:
        model = Task
        fields = "__all__"

# 定义查询入口
class Query(graphene.ObjectType):
    all_tasks = graphene.List(TaskType)

    def resolve_all_tasks(self, info):
        return Task.objects.all()

# 生成Schema
schema = graphene.Schema(query=Query)

3. 验证解决方案

启动Django服务后，在GraphQL Playground中执行查询：

query {
  allTasks {
    id
    title
    status
  }
}

此时应该能正常返回包含枚举值的结果，不会再抛出类型相关的错误。

额外说明 #

如果你的项目中有多个Enum需要处理，也可以封装一个通用的转换函数，避免重复代码。另外，若后续升级到Graphene 3.x+，这个问题会得到改善——新版本的Graphene对Django模型中的Enum支持更完善，可能不需要手动映射。

内容的提问来源于stack exchange，提问作者Christopher Z