為現有的石墨烯增加價值。枚舉

Question

有沒有辦法為現有的graphene.Enum添加一個值？

我想使用 graphene- sqlalchemy sort_enum()功能並添加我自己的附加功能 - TOTAL_COUNT_ASC和TOTAL_COUNT_DESC將啟用按總數排序，但我可以弄清楚如何做到這一點。 直接將字段添加到 graphene.Enum 不起作用：

SqlModel.sort_enum().TOTAL_COUNT_ASC = "TOTAL_COUNT_ASC"

謝謝，

Answer 1

我發現的最好方法是：

from graphene_sqlalchemy import utils

# Take the original Enum
ExtendedValues  = [(m.name, m.value) for m in SqlModel.sort_enum()._meta.enum]
# Add the new values with EnumValue
ExtendedValues += [("TOTAL_COUNT_ASC", utils.EnumValue('TOTAL_COUNT_ASC', sqlalchemy.func.count())), ('TOTAL_COUNT_DESC', utils.EnumValue('TOTAL_COUNT_DESC', sqlalchemy.func.count().desc()))]
# Create the extended graphene enum
ExtendedEnum = graphene.Enum("ExtendedEnum", ExtendedValues)