o i.!@s.dZddlZddlZddlZddlZddlmZz ddlZddlZWn e y-dZYnwz ddl m Z m Z Wn e yEd\Z Z Ynwzddl Z Wn e yWdZ YnwddlmZmZmZmZedZdd Zd d ZGd d d eZGdddeZGdddeZGdddeZddZdS)z Available HTTP Client for Datadog API client. Priority: 1. `requests` 3p module 2. `urlfetch` 3p module - Google App Engine only N)Lock)urlfetchurlfetch_errors)NN) ProxyError ClientError HTTPError HttpTimeoutz datadog.apicCs4ddlm}dj|jtttdS)Nrversionz:datadogpy/{version} (python {pyver}; os {os}; arch {arch}))r pyverosarch) datadogr format __version__platformpython_versionsystemlowermachiner rK/home/ubuntu/.local/lib/python3.10/site-packages/datadog/api/http_client.py_get_user_agent_header*s   rcCs d|_|S)zZPython3: remove context from chained exceptions to prevent leaking API keys in tracebacks.N) __cause__)excrrr_remove_context5src@seZdZdZeddZdS) HTTPClientz[ An abstract generic HTTP client. Subclasses must implement the `request` methods. c Cstd)a Main method to be implemented by HTTP clients. The returned data structure has the following fields: * `content`: string containing the response from the server * `status_code`: HTTP status code returned by the server Can raise the following exceptions: * `ClientError`: server cannot be contacted * `HttpTimeout`: connection timed out * `HTTPError`: unexpected HTTP response code z-Must be implemented by HTTPClient subclasses.)NotImplementedError) clsmethodurlheadersparamsdatatimeoutproxiesverify max_retriesrrrrequest@szHTTPClient.requestN)__name__ __module__ __qualname____doc__ classmethodr(rrrrr;src@s&eZdZdZdZeZeddZdS) RequestClientz HTTP client based on 3rd party `requests` module, using a single session. This allows us to keep the session alive to spare some execution time. Nc CshzJ|j*|jdur't|_tjj| d} |jd| |jjdt iWdn1s1wY|jj ||||||||d} | W| Stj j y`} ztt ||| d} ~ wtjyt} ztt||| d} ~ wtj jytt|||tj jy} z| jjdvrn tt| jj| jWYd} ~ | Sd} ~ wtytdw)N)r'zhttps:// User-Agent)r!r"r#r$r%r&iiiiiizYour installed version of `requests` library seems not compatible withDatadog's usage. We recommend upgrading it ('pip install -U requests').If you need help or have any question, please contact support@datadoghq.com) _session_lock_sessionrequestsSessionadapters HTTPAdaptermountr!updaterr(raise_for_status exceptionsrrConnectionErrorrTimeoutrrresponse status_codereason TypeError) rrr r!r"r#r$r%r&r' http_adapterresulterrrr(ZsD       zRequestClient.request) r)r*r+r,r2rr1r-r(rrrrr.Qs r.c@s(eZdZdZeddZeddZdS)URLFetchClientzC HTTP client based on Google App Engine `urlfetch` module. c  Cs|rdnd} dj|t|d} t|} t| d<ztj| || | ||dd} || W| Stj yA}zt |||d}~wt j yNt |||w)zh Wrapper around `urlfetch.fetch` method. TO IMPLEMENT: * `max_retries` TFz{url}?{params})r r"r/)r rr!validate_certificatedeadlinepayloadfollow_redirectsN)rurllib urlencodecopydeepcopyrrfetchraise_on_status DownloadErrorrrDeadlineExceededErrorr)rrr r!r"r#r$r%r&r'rEurl_with_params newheadersrBrCrrrr(s,      zURLFetchClient.requestcCs*|j}|ddkr|dvrdSt|dS)3 Raise on HTTP status code errors. dr0N)r>r)rrBr>rrrrNs  zURLFetchClient.raise_on_statusN)r)r*r+r,r-r(rNrrrrrDs  %rDc@s2eZdZdZdZeZeddZeddZ dS) Urllib3Clientz: HTTP client based on 3rd party `urllib3` module. Nc Cs,zB|j|jdurtj| ||rdndd|_Wdn1s!wYt|} t| d<|jj||||| d} || W| Stj j yX} zt t ||| d} ~ wtj j ym} zt t ||| d} ~ wtj jy} zt t||| d} ~ wtj jy} zt t| d} ~ ww)z Wrapper around `urllib3.PoolManager.request` method. This method will raise exceptions for HTTP status codes that are not 2xx. N CERT_REQUIRED CERT_NONE)retriesr$ cert_reqsr/)bodyfieldsr!) _pool_lock_poolurllib3 PoolManagerrKrLrr(rNr:rr MaxRetryErrorr TimeoutErrorrr) rrr r!r"r#r$r%r&r'rRr=rCrrrr(s<        zUrllib3Client.requestcCs2|j}|dks |dkr|dvrt||jdSdS)rSi,r0N)statusrr?)rr=r>rrrrNs  zUrllib3Client.raise_on_status) r)r*r+r,r^rr]r-r(rNrrrrrVs  rVcCsBtr tdtStrtrtdtStrtdtSt d)z] Resolve an appropriate HTTP client based the defined priority and user environment. z!Use `requests` based HTTP client.z!Use `urlfetch` based HTTP client.z Use `urllib3` based HTTP client.z[Datadog API client was unable to resolve a HTTP client. Please install `requests` library.) r3logdebugr.rrrDr_rV ImportErrorrrrrresolve_http_clients   rh)r,rKloggingrrI threadingrr3requests.adaptersrggoogle.appengine.apirrr_datadog.api.exceptionsrrrr getLoggerrerrobjectrr.rDrVrhrrrrs>         09 4