o tBhX @srddlmZddlmZmZmZmZmZmZddl m Z m Z  d de de deee efeefde fd d Zd S) )Enum)AnyDictMappingTypeUnioncast)GraphQLEnumType GraphQLSchemaFschemanamevaluesuse_enum_valuesc Cst|tr"t|tr"ttt|}|rdd|D}ndd|D}t|ts0tdt|||}|durAt d|dt|t sRtd|d t||j D]\}}z|||_ WqWt yqt d |d wd |jvr}|jd =dSdS) aUpdate in the schema the GraphQLEnumType corresponding to the given name. Example:: from enum import Enum class Color(Enum): RED = 0 GREEN = 1 BLUE = 2 update_schema_enum(schema, 'Color', Color) :param schema: a GraphQL Schema already containing the GraphQLEnumType type. :param name: the name of the enum in the GraphQL schema :param values: Either a Python Enum or a dict of values. The keys of the provided values should correspond to the keys of the existing enum in the schema. :param use_enum_values: By default, we configure the GraphQLEnumType to serialize to enum instances (ie: .parse_value() returns Color.RED). If use_enum_values is set to True, then .parse_value() returns 0. use_enum_values=True is the defaut behaviour when passing an Enum to a GraphQLEnumType. cSsi|]}|j|jqS)r value.0enumrrw/var/www/html/riverr-enterprise-integrations-main/venv/lib/python3.10/site-packages/gql/utilities/update_schema_enum.py *sz&update_schema_enum..cSsi|]}|j|qSr)r rrrrr,szInvalid type for enum values: NzEnum z not found in schema!z The type "z$" is not a GraphQLEnumType, it is a z Enum key "z" not found in provided values! _value_lookup) isinstancetype issubclassrrrr TypeErrorget_typeKeyErrorr r itemsr__dict__)r r r r schema_enum enum_name enum_valuerrrupdate_schema_enums0        r"N)F)rrtypingrrrrrrgraphqlr r strboolr"rrrrs