Graphene-SQLAlchemy中SQLAlchemy Enum字段映射为GraphQL字符串的问题咨询

这绝对不是预期行为——你遇到的是graphene-sqlalchemy对SQLAlchemy枚举类型的默认处理逻辑导致的问题。默认情况下，它会把SQLAlchemy的Enum字段直接映射为GraphQL的String类型，并且返回的是Python枚举实例的字符串表示（比如CoverTitleColor.YELLOW），而不是你期望的枚举值（比如"yellow"）或者自定义的GraphQL枚举类型。

根本原因 #

graphene-sqlalchemy的内置类型映射中，并没有自动将SQLAlchemy关联的Python Enum类转换为对应的GraphQL枚举类型。它只会把SQLAlchemy的Enum字段当作普通字符串处理，所以才会出现内省识别为String、返回值是枚举实例字符串的情况。

解决方案 #

你需要手动创建对应的GraphQL枚举类型，并在TypeBook中覆盖该字段的定义，同时处理解析逻辑。以下是两种可行的方案：

方案1：手动定义GraphQL枚举并覆盖字段 #

1. 首先创建与你的Python枚举对应的GraphQL枚举类型：

import graphene
from your_module import CoverTitleColor

class CoverTitleColorGraphQLEnum(graphene.Enum):
    YELLOW = CoverTitleColor.YELLOW.value
    RED = CoverTitleColor.RED.value
    BLUE = CoverTitleColor.BLUE.value
    ORANGE = CoverTitleColor.ORANGE.value
    WHITE = CoverTitleColor.WHITE.value
    PURPLE = CoverTitleColor.PURPLE.value

2. 在TypeBook中覆盖cover_title_color字段，指定使用上面的GraphQL枚举，并实现解析方法：

class TypeBook(SQLAlchemyObjectType):
    # 覆盖字段，指定为自定义GraphQL枚举（添加NonNull保证非空）
    cover_title_color = graphene.Field(graphene.NonNull(CoverTitleColorGraphQLEnum))

    class Meta:
        model = Book

    @staticmethod
    def resolve_cover_title_color(root, info):
        # 从SQLAlchemy模型的枚举实例中提取对应的值
        return root.cover_title_color.value

这样配置后，GraphQL内省会识别coverTitleColor为自定义的枚举类型，查询返回的结果也会是"yellow"、"red"这样的枚举值。

方案2：用SQLAlchemy TypeDecorator统一处理枚举 #

如果你希望在SQLAlchemy层面就统一枚举的序列化/反序列化逻辑，可以自定义一个TypeDecorator：

1. 定义自定义类型：

from sqlalchemy import types
from your_module import CoverTitleColor

class CoverTitleColorType(types.TypeDecorator):
    impl = types.String
    cache_ok = True

    def process_bind_param(self, value, dialect):
        # 写入数据库时，把枚举实例转成字符串值
        if isinstance(value, CoverTitleColor):
            return value.value
        return value

    def process_result_value(self, value, dialect):
        # 从数据库读取时，把字符串转回枚举实例
        if value is not None:
            return CoverTitleColor(value)
        return value

2. 修改Book模型的字段定义：

cover_title_color = sqlalchemy.Column(
    CoverTitleColorType, nullable=False
)

3. 再按照方案1的方式定义GraphQL枚举并覆盖字段，这样既保证了数据库存储的是字符串值，又能让GraphQL正确识别为枚举类型。

验证效果 #

配置完成后，重新执行GraphQL内省查询，coverTitleColor的类型会显示为你自定义的枚举类型（比如CoverTitleColorGraphQLEnum），查询返回的结果也会是枚举的实际值，而不是枚举实例的字符串表示。

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