o tBh>@sddlZddlZddlZddlmZmZmZmZmZm Z m Z m Z ddl Z ddl mZmZmZddlmZmZddlmZddlmZddlmZddlmZd d lmZd d lmZd d l m!Z!m"Z"m#Z#m$Z$e%e&Z'GdddeZ(dS)N)Any CollectionDictListOptionalTupleTypeUnion) DocumentNodeExecutionResult print_ast) HTTPAdapterRetry)AuthBase)RequestsCookieJar)MultipartEncoder) Transport)GraphQLRequest) extract_files)TransportAlreadyConnectedTransportClosedTransportProtocolErrorTransportServerErrorc@s^eZdZUdZejfZeee dfe d<dZ ddddddddd e f d e d e ee e fd e eee e fefd e edede edeee fdede dedeede fddZddZ     d@dede ee e fde e de ede ee e fded efd!d"Z  dAd#eede ede ee e fd eefd$d%Zd&ee e fd efd'd(Zd)e d dfd*d+Zd)eee e fd dfd,d-Z d)eee e fd dfd.d/Z!d#eed)eee e fd dfd0d1Z"d2e d3e d dfd4d5Z#d6e$j%d e fd7d8Z&  dAd#eede ede ee e fd ee e ffd9d:Z'd;ed ee e ffdd?Z)dS)BRequestsHTTPTransportz:ref:`Sync Transport ` used to execute GraphQL queries on remote servers. The transport uses the requests library to send HTTP POST requests. . file_classes)iiiiiNTrPOSTg?urlheaderscookiesauthuse_jsontimeoutverifyretriesmethodretry_backoff_factorretry_status_forcelistkwargsc KsX||_||_||_||_||_||_||_||_| |_| |_ | |_ | |_ d|_ d|_ dS)aInitialize the transport with the given request parameters. :param url: The GraphQL server URL. :param headers: Dictionary of HTTP Headers to send with the :class:`Request` (Default: None). :param cookies: Dict or CookieJar object to send with the :class:`Request` (Default: None). :param auth: Auth tuple or callable to enable Basic/Digest/Custom HTTP Auth (Default: None). :param use_json: Send request body as JSON instead of form-urlencoded (Default: True). :param timeout: Specifies a default timeout for requests (Default: None). :param verify: Either a boolean, in which case it controls whether we verify the server's TLS certificate, or a string, in which case it must be a path to a CA bundle to use. (Default: True). :param retries: Pre-setup of the requests' Session for performing retries :param method: HTTP method used for requests. (Default: POST). :param retry_backoff_factor: A backoff factor to apply between attempts after the second try. urllib3 will sleep for: {backoff factor} * (2 ** ({number of previous retries})) :param retry_status_forcelist: A set of integer HTTP status codes that we should force a retry on. A retry is initiated if the request method is in allowed_methods and the response status code is in status_forcelist. (Default: [429, 500, 502, 503, 504]) :param kwargs: Optional arguments that ``request`` takes. These can be seen at the `requests`_ source code or the official `docs`_ .. _requests: https://github.com/psf/requests/blob/master/requests/api.py .. _docs: https://requests.readthedocs.io/en/master/ N)rrr r!r"default_timeoutr$r%r&r'r(r)sessionresponse_headers) selfrrr r!r"r#r$r%r&r'r(r)r.m/var/www/html/riverr-enterprise-integrations-main/venv/lib/python3.10/site-packages/gql/transport/requests.py__init__%s- zRequestsHTTPTransport.__init__cCsb|jdur-t|_|jdkr)tt|j|j|jddd}dD] }|j||qdSdSt d)Nr)totalbackoff_factorstatus_forcelistallowed_methods) max_retries)zhttp://zhttps://zTransport is already connected) r+requestsSessionr%r rr'r(mountr)r-adapterprefixr.r.r/connectcs    zRequestsHTTPTransport.connectFdocumentvariable_valuesoperation_name extra_args upload_filesreturncs||jstdt|}d|i}|r||d<|j|j|j|p|j|jd} |r|dus,Jt||j d\} | |d<t |} t d| d d tD} fd d tD} t | }t d || |d }| D]"\}}t|d|}t|dd}|dur||f||<ql|||f||<qlt|d}|| d<| dduri| d<ni| d| d<|j| dd<n|r||d<|jrdnd}|| |<t tjrt dt || |j|r| ||jj|j|jfi| }|j|_dtjdt fdd}z| }t tjrt d|j!Wnt"y||dYnwd|vr.d|vr.||dt#|$d|$d|$ddS) aExecute GraphQL query. Execute the provided document AST against the configured remote server. This uses the requests library to perform a HTTP POST request to the remote server. :param document: GraphQL query as AST Node object. :param variable_values: Dictionary of input parameters (Default: None). :param operation_name: Name of the operation that shall be executed. Only required in multi-operation documents (Default: None). :param timeout: Specifies a default timeout for requests (Default: None). :param extra_args: additional arguments to send to the requests post method :param upload_files: Set to True if you want to put files in the variable values :return: The result of execution. `data` is the result of executing the query, `errors` is null if no errors occurred, and is a non-empty array if an error occurred. Transport is not connectedquery operationNamerr!r r#r$N) variablesrrFz operations %scSsi|] \}}t||gqSr.str.0ipathr.r.r/ sz1RequestsHTTPTransport.execute..csi|] \}}t||qSr.rGrIfilesr.r/rMsz file_map %s) operationsmapname content_type)fieldsdatarz Content-Typejson>>> %srespreasonc SsVz|Wntjy}z tt||jj|d}~ww|j}td|d|)Nz(Server did not return a GraphQL result: : ) raise_for_statusr6 HTTPErrorrrHresponse status_codetextr)rXrYe result_textr.r.r/raise_response_errors z;RequestsHTTPTransport.execute..raise_response_error<<< %sNot a JSON answererrors$No "data" or "errors" keys in answer extensionsrerUrg)%r+rr rr!r r*r$rrrVdumpslogdebug enumerateitemsgetattrrrSr" isEnabledForloggingINFOinfoupdater)requestr&rr,r6ResponserHr_ Exceptionr get)r-r<r=r>r#r?r@ query_strpayload post_argsnulled_variable_valuesoperations_strfile_map file_streams file_map_strrTkfrRrSrUdata_keyr]rbresultr.rNr/executews                zRequestsHTTPTransport.executereqscsjstdjjjjfi|||}|j_|} | || | |fdd|DS)aExecute multiple GraphQL requests in a batch. Execute the provided requests against the configured remote server. This uses the requests library to perform a HTTP POST request to the remote server. :param reqs: GraphQL requests as a list of GraphQLRequest objects. :param timeout: Specifies a default timeout for requests (Default: None). :param extra_args: additional arguments to send to the requests post method :return: A list of results of execution. For every result `data` is the result of executing the query, `errors` is null if no errors occurred, and is a non-empty array if an error occurred. rBcg|]}|qSr.)_answer_to_execution_result)rJanswerr-r.r/ ;z7RequestsHTTPTransport.execute_batch..) r+rrtr&r_build_batch_post_argsrr,_extract_response_validate_answer_is_a_list)_validate_num_of_answers_same_as_requests _validate_every_answer_is_a_dict)_validate_data_and_errors_keys_in_answers)r-rr#r?r]answersr.rr/ execute_batchs      z#RequestsHTTPTransport.execute_batchrcCs t|d|d|ddS)NrerUrgrh)r rw)r-rr.r.r/r=s z1RequestsHTTPTransport._answer_to_execution_resultresultscCs"t|ts|t|ddSdS)NzAnswer is not a list) isinstancelist_raise_invalid_resultrH)r-rr.r.r/rDs z0RequestsHTTPTransport._validate_answer_is_a_listcCs.|D]}d|vrd|vr|t|dqdS)NrerUrf)rrHr-rrr.r.r/rKsz?RequestsHTTPTransport._validate_data_and_errors_keys_in_answerscCs(|D]}t|ts|t|dqdS)NzNot every answer is dict)rdictrrHrr.r.r/rUs  z6RequestsHTTPTransport._validate_every_answer_is_a_dictcCs(t|t|kr|t|ddSdS)NzInvalid answer length)lenrrH)r-rrr.r.r/rZs z?RequestsHTTPTransport._validate_num_of_answers_same_as_requestsrarYcCstd|d|)Nz.Server did not return a valid GraphQL result: rZ)r)r-rarYr.r.r/resz+RequestsHTTPTransport._raise_invalid_resultr]c Csz||}ttjrtd|jW|SW|Stj y2}z t t ||j j |d}~wtyD|t |jdY|Sw)Nrcrd)r[rVrjrorprqrrr_r6r\rrHr]r^rvr)r-r]rr`r.r.r/rls   z'RequestsHTTPTransport._extract_responsecsjjj|p jjd}jrdnd}fdd|D||<ttj r2t dt ||| j|r?| ||S)NrErVrUcrr.) _build_data)rJreqrr.r/rrz@RequestsHTTPTransport._build_batch_post_args..rW)rr!r r*r$r"rjrorprqrrrVrirsr))r-rr#r?rzrr.rr/r|s   z,RequestsHTTPTransport._build_batch_post_argsrcCs6t|j}d|i}|jr|j|d<|jr|j|d<|S)NrCrDrF)r r<r>r=)r-rrxryr.r.r/rs   z!RequestsHTTPTransport._build_datacCs|jr |jd|_dSdS)z2Closing the transport by closing the inner sessionN)r+closerr.r.r/rs  zRequestsHTTPTransport.close)NNNNF)NN)*__name__ __module__ __qualname____doc__ioIOBaserrrr__annotations___default_retry_codesrHrrr rrboolintfloatrr0r;r r rrrrrrrrrrr6rurrrrr.r.r.r/rs       > " (       r))rrVrptypingrrrrrrrr r6graphqlr r r requests.adaptersr r requests.authrrequests.cookiesr#requests_toolbelt.multipart.encoderr gql.transportrgraphql_requestrutilsr exceptionsrrrr getLoggerrrjrr.r.r.r/s (