o tBh@sdZddlZddlZddlmZmZddlmZddlm Z m Z m Z m Z m Z mZmZmZddlmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;mZ>m?Z?m@Z@mAZAmBZBmCZCmDZDddlEmFZFdd lGmHZHeIeJZKeLd ZMd e d e e9fd dZNde de d e e9fddZOddddd efddZPGdddZQGdddeZRGdddeRZSGdddeRZTGdd d eSeTZUGd!d"d"eUZVGd#d$d$eUZWGd%d&d&eUZXGd'd(d(ZYGd)d*d*ZZGd+d,d,Z[Gd-d.d.eZ\Gd/d0d0eRZ]Gd1d2d2eRZ^Gd3d4d4e\Z_Gd5d6d6e_e^Z`Gd7d8d8e`ZaGd9d:d:e\e]ZbGd;d<dYs  z4ast_from_serialized_value_untyped..cs*|]\}}|rtt|d|dVqdSrJ)namerJNr,r)rH field_name field_valuerKrKrLrM] fieldscss|]}t|VqdSrErFrHitemrKrKrLrMecs|]}|r|VqdSrErKrHnoderKrKrLrMfr[valuesrPz.0zCannot convert value to AST: .)r3r+ isinstanceritemstupler-rstrr'boolrintr%floatrendswithrr1 TypeErrorr@)rC field_items field_nodes maybe_nodesnodesrJrKrKrLrGIs4            rGrJtype_c sttr |jSt|r-tt|}|j}t|}t|t r+t dt |d|Sdur4t St urBt dt |dt |rttt|}|jttrottsofddD}tdd|D}t|dStSt|rdusttsdStt|}fd d|jD}td d|D}t|d St|r|}t|trt|rt|d S|turt|rt |d St!|d St"|St#d t |d)a This is a partial copy paste of the ast_from_value function in graphql-core utilities/ast_from_value.py Overwrite the if blocks that use recursion and add a new case to return a VariableNode when value is a DSLVariable Produce a GraphQL Value AST given a Python object. Raises a GraphQLError instead of returning None if we receive an Undefined of if we receive a Null value for a Non-Null type. z(Received Null value for a Non-Null type rbNz"Received Undefined value for type c3s|]}t|VqdSrE)ast_from_valuerY) item_typerKrLrMsz!ast_from_value..csr\rErKr]rKrKrLrMr[r_c3s0|]\}}|vr|t||jfVqdSrE)rqtype)rHrTfieldrPrKrLrMscsrNrOrRrSrKrKrLrMrVrWrPzUnexpected input type: )$rc DSLVariableset_typeast_variable_namer=r rof_typerqr+rr@r3r<rrrfrer'r:rrrXrdr-r; serializer9rr_re_integer_stringmatchr%r1rGrk) rJrp inner_type ast_valuemaybe_value_nodes value_nodesrlrmrCrK)rrrJrLrq{sX               rq operations DSLExecutableoperations_with_namecOspg|dd|DR}|D]\}}||_q|D]}t|ts-tdt|dqtdd|DdS)aGiven arguments instances of :class:`DSLExecutable` containing GraphQL operations or fragments, generate a Document which can be executed later in a gql client or a gql session. Similar to the :func:`gql.gql` function but instead of parsing a python string to describe the request, we are using operations which have been generated dynamically using instances of :class:`DSLField`, generated by instances of :class:`DSLType` which themselves originated from a :class:`DSLSchema` class. :param \*operations: the GraphQL operations and fragments :type \*operations: DSLQuery, DSLMutation, DSLSubscription, DSLFragment :param \**operations_with_name: the GraphQL operations with an operation name :type \**operations_with_name: DSLQuery, DSLMutation, DSLSubscription :return: a Document which can be later executed or subscribed by a :class:`Client `, by an :class:`async session ` or by a :class:`sync session ` :raises TypeError: if an argument is not an instance of :class:`DSLExecutable` :raises AttributeError: if a type has not been provided in a :class:`DSLFragment` css|]}|VqdSrErKrH operationrKrKrLrMszdsl_gql..zsOperations should be instances of DSLExecutable (DSLQuery, DSLMutation, DSLSubscription or DSLFragment). Received: rbcSsg|]}|jqSrK)executable_astrrKrKrL szdsl_gql..) definitions)r`rdrQrcrrkrsr)rrall_operationsrQrrKrKrLdsl_gqls$  rc@s0eZdZdZdefddZdeddfdd Zd S) DSLSchemazThe DSLSchema is the root of the DSL code. Attributes of the DSLSchema class are generated automatically with the `__getattr__` dunder method in order to generate instances of :class:`DSLType` schemacCs&t|tstdt|||_dS)aGInitialize the DSLSchema with the given schema. :param schema: a GraphQL Schema provided locally or fetched using an introspection query. Usually `client.schema` :type schema: GraphQLSchema :raises TypeError: if the argument is not an instance of :class:`GraphQLSchema` z1DSLSchema needs a schema as parameter. Received: N)rcr"rkrs_schema)selfrrKrKrL__init__s   zDSLSchema.__init__rQrDDSLTypecCsR|j|}|durtd|dt|ttfs$td|d|dt||S)NzType 'z' not found in the schema!zType "z (z`)" is not valid as an attribute of DSLSchema. Only Object types or Interface types are accepted.)rget_typeAttributeErrorrcr rr)rrQtype_defrKrKrL __getattr__#s  zDSLSchema.__getattr__N)__name__ __module__ __qualname____doc__r"rrfrrKrKrKrLr src@sTeZdZUdZeed<    dddZed dd efd d Z    dd dZ dS) DSLSelectorafDSLSelector is an abstract class which defines the :meth:`select ` method to select children fields in the query. Inherited by :class:`DSLRootFieldSelector `, :class:`DSLFieldSelector ` :class:`DSLFragmentSelector ` selection_setrX DSLSelectablefields_with_aliasDSLSelectableWithAliascOs,tdd|_|s |r|j|i|dSdS):meta private:rK) selectionsN)r0rselectrrXrrKrKrLr@s zDSLSelector.__init__rtrDcCtd)Nz:Any DSLSelector subclass must have a is_valid_field methodNotImplementedError)rrtrKrKrLis_valid_fieldKszDSLSelector.is_valid_fieldcOst||}|D]}t|tstdt|||s'td|d|qtdd|D}|j j ||j _ t d|d|dS) aSelect the fields which should be added. :param \*fields: fields or fragments :type \*fields: DSLSelectable :param \**fields_with_alias: fields or fragments with alias as key :type \**fields_with_alias: DSLSelectable :raises TypeError: if an argument is not an instance of :class:`DSLSelectable` :raises GraphQLError: if an argument is not a valid field z7Fields should be instances of DSLSelectable. Received: zInvalid field for z: css|]}|jVqdSrE) ast_field)rHrtrKrKrLrMssz%DSLSelector.select..zAdded fields: z in N) DSLFieldget_aliased_fieldsrcrrkrsrrrerrlogdebug)rrXr added_fieldsrtadded_selectionsrKrKrLrQs"  zDSLSelector.selectNrXrrr) rrrrr0__annotations__rrrgrrrKrKrKrLr3s    rc@sPeZdZUdZded<eeed<eed<ee ddZ  dd d Z dS)rzInterface for the root elements which can be executed in the :func:`dsl_gql ` function Inherited by :class:`DSLOperation ` and :class:`DSLFragment ` DSLVariableDefinitionsvariable_definitionsrQrcCr)8Generates the ast for :func:`dsl_gql `.z`. The fields arguments should be either be fragments or fields of root GraphQL types (Query, Mutation or Subscription) and correspond to the operation_type of this operation. :param \*fields: root fields or fragments :type \*fields: DSLSelectable :param \**fields_with_alias: root fields or fragments with alias as key :type \**fields_with_alias: DSLSelectable :raises TypeError: if an argument is not an instance of :class:`DSLSelectable` :raises AssertionError: if an argument is not a field which correspond to the operation type N)rQrrrrrrKrKrLrszDSLExecutable.__init__Nr) rrrrrr rfr0propertyrrrrKrKrKrLr{s   c@s"eZdZdZdddefddZdS)DSLRootFieldSelectoraClass used to define the :meth:`is_valid_field ` method for root fields for the :meth:`select ` method. Inherited by :class:`DSLOperation ` rtrrDcCst|tsJ|jj}t|tr$|jdvr|dkS|jdkr"|dkSd St|trd|jdus0J|jjj}d}|dkr?|j }n|dkrG|j }n|dkrN|j }|dur]t d|dd S|jj|jkSd S) aCheck that a field is valid for a root field. For operations, the fields arguments should be fields of root GraphQL types (Query, Mutation or Subscription) and correspond to the operation_type of this operation. the :code:`__typename` field can only be added to Query or Mutation. the :code:`__schema` and :code:`__type` field can only be added to Query. )__schema__typeQUERY __typename SUBSCRIPTIONNMUTATIONzRoot type of type z not found in the schema!F)rc DSLOperationoperation_typerQ DSLMetaFieldrdsl_type _dsl_schemar query_type mutation_typesubscription_typererror parent_type)rrtoperation_namer root_typerKrKrLrs2       z#DSLRootFieldSelector.is_valid_fieldN)rrrrrgrrKrKrKrLrrc@s:eZdZUdZeed<edefddZde fddZ dS) rzInterface for GraphQL operations. Inherited by :class:`DSLQuery `, :class:`DSLMutation ` and :class:`DSLSubscription ` rrDcCsBtdt|j|j|jd|jrdt|jdiniddiS)r)rrrrQrP directivesrK)r.r/rrrget_ast_definitionsrQr)rrKrKrLrszDSLOperation.executable_astcCsd|jjdS)N<>) __class__rrrKrKrL__repr__zDSLOperation.__repr__N) rrrrr/rrr.rrfrrKrKrKrLrs  rc@eZdZejZdS)DSLQueryN)rrrr/rrrKrKrKrLr rc@r) DSLMutationN)rrrr/rrrKrKrKrLrrrc@r)DSLSubscriptionN)rrrr/rrrKrKrKrLr rrc@sTeZdZdZdefddZdedefddZdeddfd d Z d e ddfd d Z dS)ruaZThe DSLVariable represents a single variable defined in a GraphQL operation Instances of this class are generated for you automatically as attributes of the :class:`DSLVariableDefinitions` The type of the variable is set by the :class:`DSLField` instance that receives it in the :meth:`args ` method. rQcCs0||_d|_tt|jdd|_d|_d|_dS)rNrPrQ)rQast_variable_typer6r)rw default_valuersrrQrKrKrLrs  zDSLVariable.__init__rprDcCsft|r t|trt||jdSt|tr t||jdSt|tt t fs*Jt t |j ddS)N)rsrPr)r>rcrr& to_ast_typerxrr*r!rrr(r)rQrrprKrKrLrs   zDSLVariable.to_ast_typecCs||_|||_|SrE)rsrrrrKrKrLrv-s zDSLVariable.set_typercCs ||_|SrE)r)rrrKrKrLdefault2szDSLVariable.defaultN) rrrrrfrrr2rrvrrrKrKrKrLru s  ruc@s@eZdZdZddZdeddfddZdeed ffd d Z d S) raThe DSLVariableDefinitions represents variable definitions in a GraphQL operation Instances of this class have to be created and set as the `variable_definitions` attribute of a DSLOperation instance Attributes of the DSLVariableDefinitions class are generated automatically with the `__getattr__` dunder method in order to generate instances of :class:`DSLVariable`, that can then be used as values in the :meth:`args ` method. cCs i|_dS)rN) variablesrrKrKrLrCs zDSLVariableDefinitions.__init__rQrDrucCs"||jvr t||j|<|j|SrE)rrurrKrKrLrGs  z"DSLVariableDefinitions.__getattr__.cCstdd|jDS)zp :meta private: Return a list of VariableDefinitionNodes for each variable with a type cssF|]}|jdurt|j|j|jdurdnt|j|jddVqdS)NrK)rsvariablerr)rsr5rrwrrq)rHvarrKrKrLrMRs    z=DSLVariableDefinitions.get_ast_definitions..)rerr`rrKrKrLrLs z*DSLVariableDefinitions.get_ast_definitionsN) rrrrrrfrr r5rrKrKrKrLr7s  rc@sJeZdZdZdeeefdefddZde ddfd d Z de fd d Z d S)raThe DSLType represents a GraphQL type for the DSL code. It can be a root type (Query, Mutation or Subscription). Or it can be any other object type (Human in the StarWars schema). Or it can be an interface type (Character in the StarWars schema). Instances of this class are generated for you automatically as attributes of the :class:`DSLSchema` Attributes of the DSLType class are generated automatically with the `__getattr__` dunder method in order to generate instances of :class:`DSLField` graphql_type dsl_schemacCs"||_||_td|ddS)aUInitialize the DSLType with the GraphQL type. .. warning:: Don't instantiate this class yourself. Use attributes of the :class:`DSLSchema` instead. :param graphql_type: the GraphQL type definition from the schema :param dsl_schema: reference to the DSLSchema which created this type Creating )N)_typerrr)rrrrKrKrLroszDSLType.__init__rQrDrcCsnt|}||jjvr|}|jj|}n||jjvr"|}|jj|}n td|d|jjdt||j||S)NzField z does not exist in type rb)rBrrXrrQr)rrQcamel_cased_nameformatted_namertrKrKrLrs  zDSLType.__getattr__cCsd|jjd|jdSNr r)rrrrrKrKrLrzDSLType.__repr__N) rrrrr r rrrrfrrrKrKrKrLr`s  rc@s\eZdZUdZeeeefed<e de dde e dfde dfdd Zde fd d Zd S) raMDSLSelectable is an abstract class which indicates that the subclasses can be used as arguments of the :meth:`select ` method. Inherited by :class:`DSLField `, :class:`DSLFragment ` :class:`DSLInlineFragment ` rrXrrrD)r.cCsg|dd|DRS)z :meta private: Concatenate all the fields (with or without alias) in a Tuple. Set the requested alias for the fields with alias. css|] \}}||VqdSrE)alias)rHrrtrKrKrLrMsz3DSLSelectable.get_aliased_fields..)rd)rXrrKrKrLrs  z DSLSelectable.get_aliased_fieldscCs t|jSrE)r?rrrKrKrL__str__s zDSLSelectable.__str__N)rrrrr rr$rr staticmethodrrrfr rrrKrKrKrLrs    rc@"eZdZdZdedefddZdS)DSLFragmentSelectora7Class used to define the :meth:`is_valid_field ` method for fragments for the :meth:`select ` method. Inherited by :class:`DSLFragment `, :class:`DSLInlineFragment ` rtrDcCst|ttfs Jt|ttfrdSt|tsJt|tr#|jdkS|j}|dus,J|j|jvr?|j|jj |j j kSdS)Check that a field is valid.TrNF) rc DSLFragmentDSLInlineFragmentrrrQrrXkeysrsrt)rrt fragment_typerKrKrLrs   z"DSLFragmentSelector.is_valid_fieldNrrrrrrgrrKrKrKrLrs rc@r)DSLFieldSelectorzClass used to define the :meth:`is_valid_field ` method for fields for the :meth:`select ` method. Inherited by :class:`DSLField `, rtrDcCst|tsJt|ttfrdSt|tsJt|tr!|jdkSt|jj}t|t t fs0dS|j|j vrC|j |jj|jjkSdS)rTrF) rcrrrrrQr7rtrsrr rXr)rrtrrKrKrLrs   zDSLFieldSelector.is_valid_fieldNrrKrKrKrLrrrc@s,eZdZUdZeed<deddfddZdS)rzwDSLSelectableWithAlias is an abstract class which indicates that the subclasses can be selected with an alias. rrrDcCst|d|j_|S)aSet an alias .. note:: You can also pass the alias directly at the :meth:`select ` method. :code:`ds.Query.human.select(my_name=ds.Character.name)` is equivalent to: :code:`ds.Query.human.select(ds.Character.name.alias("my_name"))` :param alias: the alias :type alias: str :return: itself rP)r)rr)rrrKrKrLrszDSLSelectableWithAlias.aliasN)rrrrrrrfrrKrKrKrLrs rc seZdZUdZeeefed<eed<e ed< dde deeefde de e fd d Z ed d ZdddZdddZde d efddZ     dfdd Zd e fddZZS)raTThe DSLField represents a GraphQL field for the DSL code. Instances of this class are generated for you automatically as attributes of the :class:`DSLType` If this field contains children fields, then you need to select which ones you want in the request using the :meth:`select ` method. rrrtNrQrrcCsF||_||_tt|dddd|_||_td|t |dS)aInitialize the DSLField. .. warning:: Don't instantiate this class yourself. Use attributes of the :class:`DSLType` instead. :param name: the name of the field :param parent_type: the GraphQL type definition from the schema of the parent type of the field :param field: the GraphQL field definition from the schema :param dsl_type: reference of the DSLType instance which created this field rPrK)rQ argumentsrrN) rrtrr)rrrrrr)rrQrrtrrKrKrLr"szDSLField.__init__cCs |jjjS)r)rrQrJrrKrKrLrQBs z DSLField.namerDcKs|jdi|S)NrKargsrkwargsrKrKrL__call__GrzDSLField.__call__c sRjjdusJjjtfdd|Dj_td|ddS)aSet the arguments of a field The arguments are parsed to be stored in the AST of this field. .. note:: You can also call the field directly with your arguments. :code:`ds.Query.human(id=1000)` is equivalent to: :code:`ds.Query.human.args(id=1000)` :param \**kwargs: the arguments (keyword=value) :return: itself :raises KeyError: if any of the provided arguments does not exist for this field. Nc3s4|]\}}tt|dt||jdVqdSrO)r r)rq _get_argumentrs)rHrQrJrrKrLrM]s z DSLField.args..zAdded arguments z in field r)rrrerdrrrrKrrLrJs z DSLField.argscCs2|jj|}|durtd|d|jd|S)zMethod used to return the GraphQLArgument definition of an argument from its name. :raises KeyError: if the provided argument does not exist for this field. Nz Argument z does not exist in rb)rtrgetKeyError)rrQargrKrKrLriszDSLField._get_argumentrXrrrc tj|i||j|j_|SgCalling :meth:`select ` method with corrected typing hints superrrrrrrKrLrws zDSLField.selectcCs"d|jjd|jjd|jdS)Nrrz::r)rrrrQrrKrKrLrs"zDSLField.__repr__rE)rDr)rXrrrrDr)rrrrr r rrrrrfr rrrrQrrrrrr __classcell__rKrKrrLrs8        rc sleZdZdZedeeeeeedeeeedde e ediddd Z de ffd d Z ZS) rzDSLMetaField represents a GraphQL meta-field for the DSL code. meta-fields are reserved field in the GraphQL type system prefixed with "__" two underscores and used for introspection. meta_field__Schema__TyperQ)rpr)rrrrWcsFz|jj|}Wntytd|dwt||j|dS)ziInitialize the meta-field. :param name: the name between __typename, __schema or __type zInvalid meta-field ""N) meta_typerXrrrr)rrQrtrrKrLrs  zDSLMetaField.__init__)rrrrr rr#r r8rrrrfrrrKrKrrLrs  rcszeZdZUdZeeefed<eed<    ddd Z     dfd d Z d e d dfddZ d e fddZZS)rzADSLInlineFragment represents an inline fragment for the DSL code.rrrXrrrcOs8td|tdd|_tj|g|Ri|dS)a#Initialize the DSLInlineFragment. :param \*fields: new children fields :type \*fields: DSLSelectable (DSLField, DSLFragment or DSLInlineFragment) :param \**fields_with_alias: new children fields with alias as key :type \**fields_with_alias: DSLField rrKrN)rrr$rrrrrKrKrLrs zDSLInlineFragment.__init__rDcrrrrrrKrLrs zDSLInlineFragment.selecttype_conditioncCs$|j|_tt|jjdd|j_|S)z2Provides the GraphQL type of this inline fragment.rPr)rr(r)rQrr rr rKrKrLons   zDSLInlineFragment.oncCsBd}z |d|jj7}Wn tyYnwd|jj|dS)Nz on rr)rrQrrr)r type_inforKrKrLrs zDSLInlineFragment.__repr__r)rXrrrrDr)rrrrr r rrr$rrrr rfrrrKrKrrLrs$    rcseZdZUdZeeeefed<e ed<e ed<de fddZ e de fdd Z  dfdd ZdeddfddZe defddZde fddZZS)rzADSLFragment represents a named GraphQL fragment for the DSL code.rrrQcCs*t|||_d|_td|dS)zkInitialize the DSLFragment. :param name: the name of the fragment :type name: str Nr)rrrQrrrrrKrKrLrs zDSLFragment.__init__rDcCstdd}t|jd|_|S)zast_field property will generate a FragmentSpreadNode with the provided name. Note: We need to ignore the type because of `issue #4125 of mypy `_. rKr rP)rr)rQ)r spread_noderKrKrLrs zDSLFragment.ast_fieldrXrrrcs(|jdur tdtj|i||S)rN=Missing type condition. Please use .on(type_condition) method)rrrrrrrKrLrs zDSLFragment.selectr cCs |j|_|S)zProvides the GraphQL type of this fragment. :param type_condition: the provided type :type type_condition: DSLType )rr rKrKrLr szDSLFragment.oncCs~|jdusJ|jdurtd|j}t|dkr i}nd|i}tdtt|jjdd|j d|t|jddd S) zGenerates the ast for :func:`dsl_gql `. :raises AttributeError: if a type has not been provided NrrrrPr)r rrK)rQr) rQrrrrlenrr(r)r)rfragment_variable_definitionsvariable_definition_kwargsrKrKrLr"s(     zDSLFragment.executable_astcCsd|jjd|jdSr)rrrQrrKrKrLrGrzDSLFragment.__repr__)rXrrrrDr)rrrrr r r rrrrfrrrrrr rrrrrKrKrrLrs*    $r)drloggingreabcrrmathrtypingrrrrr r r r graphqlr rrrrrrrrrrrrrrrrrrr r!r"r#r$r%r&r'r(r)r*r+r,r-r.r/r0r1r2r3r4r5r6r7r8r9r:r;r<r=r>r?graphql.pyutilsr@utilsrB getLoggerrrcompilerzrGrqrrrrrrrrrrurrrrrrrrrrrKrKrKrLsN ( 5   2[ 4)H38*)5#!!t#8