o eZh[ @sUddlZddlZddlZddlZddlZddlmZddlmZm Z m Z m Z m Z m Z mZmZmZmZmZmZddlmZmZddlmZmZmZmZmZddlmZmZm Z m!Z!m"Z"m#Z#ddl$m%Z%m&Z&m'Z'm(Z(m)Z)dd l*m+Z+m,Z,dd l-m.Z.dd l/m0Z0m1Z1m2Z2dd l3m4Z4dd l5m6Z6m7Z7e)e(e%e'dZ8e e9ee&fe:d<hdZ;de6e4fZGdddeZ?GdddZ@de e"deAdeAdeAdeeddf ddZBde e"ddfdd ZCdS)!N)python_version) AnyCallable CollectionDictListMapping NamedTupleOptionalTupleTypeUnioncast)Lockwarn_stacklevel)ConnectionErrorConnectionTimeout SniffingErrorTransportErrorTransportWarning)DEFAULTApiResponseMeta DefaultType HttpHeaders NodeConfig SniffOptions)AiohttpHttpNodeBaseNodeHttpxAsyncHttpNodeRequestsHttpNodeUrllib3HttpNode)NodePool NodeSelector)OpenTelemetrySpan)DEFAULT_SERIALIZERS SerializerSerializerCollection) __version__)client_meta_versionresolve_default)urllib3requestsZaiohttpZ httpxasyncNODE_CLASS_NAMES>Niiiiiietzelastic_transport.transportc@seZdZUeed<eed<dS)TransportApiResponsemetabodyN)__name__ __module__ __qualname__r__annotations__rr6r6S/var/www/html/lang_env/lib/python3.10/site-packages/elastic_transport/_transport.pyr/Us  r/c+@seZdZdZeedddddddddddddd ddefd eed e e e e fd e ed e dee e e efdeedeedeee efde dedeede de de de deededeeddge eeeeffde dee e ff(dd Zdeeeeeeed!d"e d#e d$eed%e ee efefde eefde eeefde e efd&e eeefd'e eee e fd(fefd)e eefd*efd+d,Zd4d-e d*dfd.d/Zd5d0d1Zd-e d*e fd2d3Z dS)6 Transportz Encapsulation of transport-related to logic. Handles instantiation of the individual nodes as well as creating a node pool to hold them. Main interface is the :meth:`elastic_transport.Transport.perform_request` method. TNzapplication/json)iiiiFg?g$@ node_configs node_classnode_pool_classrandomize_nodes_in_poolnode_selector_classdead_node_backoff_factormax_dead_node_backoff serializersdefault_mimetype max_retriesretry_on_statusretry_on_timeoutsniff_on_startsniff_before_requestssniff_on_node_failure sniff_timeoutmin_delay_between_sniffingsniff_callbackr meta_headerclient_meta_servicecCst|tr!|tvrdtt}td|d|dt|}t|j }t|j }||krDtd|r7dndd|r>d nd d t || |||d |d t t fdt t ff|_tttttft|dd}|ru|j|f7_t|ts~td||_t}|r||t|| d|_| |_| |_| |_i}|dur||d<|dur||d<|dur||d<||f||d||_| |_||_ ||_!||_"||_#t$|_%||_&d|_'| r|(ddSdS)a :arg node_configs: List of 'NodeConfig' instances to create initial set of nodes. :arg node_class: subclass of :class:`~elastic_transport.BaseNode` to use or the name of the Connection (ie 'urllib3', 'requests') :arg node_pool_class: subclass of :class:`~elastic_transport.NodePool` to use :arg randomize_nodes_in_pool: Set to false to not randomize nodes within the pool. Defaults to true. :arg node_selector_class: Class to be used to select nodes within the :class:`~elastic_transport.NodePool`. :arg dead_node_backoff_factor: Exponential backoff factor to calculate the amount of time to timeout a node after an unsuccessful API call. :arg max_dead_node_backoff: Maximum amount of time to timeout a node after an unsuccessful API call. :arg serializers: optional dict of serializer instances that will be used for deserializing data coming from the server. (key is the mimetype) :arg max_retries: Maximum number of retries for an API call. Set to 0 to disable retries. Defaults to ``0``. :arg retry_on_status: set of HTTP status codes on which we should retry on a different node. defaults to ``(429, 502, 503, 504)`` :arg retry_on_timeout: should timeout trigger a retry on different node? (default ``False``) :arg sniff_on_start: If ``True`` will sniff for additional nodes as soon as possible, guaranteed before the first request. :arg sniff_on_node_failure: If ``True`` will sniff for additional nodees after a node is marked as dead in the pool. :arg sniff_before_requests: If ``True`` will occasionally sniff for additional nodes as requests are sent. :arg sniff_timeout: Timeout value in seconds to use for sniffing requests. Defaults to 1 second. :arg min_delay_between_sniffing: Number of seconds to wait between calls to :meth:`elastic_transport.Transport.sniff` to avoid sniffing too frequently. Defaults to 10 seconds. :arg sniff_callback: Function that is passed a :class:`elastic_transport.Transport` and :class:`elastic_transport.SniffOptions` and should do node discovery and return a list of :class:`elastic_transport.NodeConfig` instances. :arg meta_header: If set to False the ``X-Elastic-Client-Meta`` HTTP header won't be sent. Defaults to True. :arg client_meta_service: Key-value pair for the service field of the client metadata header. Defaults to the service key-value for Elastic Transport. z', 'z Unknown option for node_class: 'z'. Available options are: ''zSpecified 'node_class' iszis notz async, should be asyncsyncz instead)r:rFrGrHrKpytZ_CLIENT_META_HTTP_CLIENTNz"'meta_header' must be of type bool)rBr>r?r@)r;Zrandomize_nodesgT)) isinstancestrr-joinsortedkeys ValueErrorinspectiscoroutinefunctionperform_requestvalidate_sniffing_optionsr)rr(_transport_client_metarr r getattrbool TypeErrorrLr%copyupdater'rArCrDrE node_poolZ_sniff_on_start_sniff_before_requests_sniff_on_node_failure_sniff_timeout_sniff_callbackr_sniffing_lock_min_delay_between_sniffing_last_sniffed_atsniff)selfr:r;r<r=r>r?r@rArBrCrDrErFrGrHrIrJrKrLrMoptionsZis_transport_asyncZ is_node_asyncZhttp_client_metaZ _serializersZnode_pool_kwargsr6r6r7__init__bs D        zTransport.__init__)r1headersrCrDrErequest_timeout client_meta otel_spanmethodtargetr1rprqrr.rsreturnc Cs6|turt} nt|} t||j}t||j}t||j}t| td} |jr:ddd|j t| dD| d<|durVd| vrFt d|j j || dd } | | nd} g} t|d D]7}|jrk|d d }d }d}|j}t}zI| |j|j|j||j||| | |d }td ||j||jjt|f|dkr|j |j|jj}nd}|jj|vrd}t |j|}Wnt!yD}zttd ||j|dt|ft"|t#r|}d}n t"|t$rd}d}|r|j%||j&rz|d Wn t!yYnw|r||kr,|dur&|WYd}~St'| |_(tj)d|||d| *|WYd}~q`d}~wwd|jjkrSdks[n|jjt+vrb|j,|n|j%||j&r~z|d Wn t!y}Ynw|r||krt |j|St)d|jj||q`dS)a@ Perform the actual request. Retrieve a node from the node pool, pass all the information to it's perform_request method and return the data. If an exception was raised, mark the node as failed and retry (up to ``max_retries`` times). If the operation was successful and the node used was previously marked as dead, mark it as live, resetting it's failure count. :arg method: HTTP method to use :arg target: HTTP request target :arg body: body of the request, will be serialized using serializer and passed to the node :arg headers: Additional headers to send with the request. :arg max_retries: Maximum number of retries before giving up on a request. Set to ``0`` to disable retries. :arg retry_on_status: Collection of HTTP status codes to retry. :arg retry_on_timeout: Set to true to retry after timeout errors. :arg request_timeout: Amount of time to wait for a response to fail with a timeout error. :arg client_meta: Extra client metadata key-value pairs to send in the client meta header. :arg otel_span: OpenTelemetry span used to add metadata to the span. :returns: Tuple of the :class:`elastic_transport.ApiResponseMeta` with the deserialized response. N,css"|] \}}|d|VqdS)=Nr6.0kvr6r6r7 2s  z,Transport.perform_request..r6zx-elastic-client-metaz content-typezr_extrasZconnections_per_noderNcsi|] \}}|vr||qSr6r6ryZ exempt_attrsr6r7 "s z7warn_if_varying_node_config_options..zDetected NodeConfig instances with different options. It's recommended to keep all options except for 'host' and 'port' the same for sniffing to work reliably.)category stacklevel) dataclassesasdictitemswarningswarnrr)r:Zmatch_attr_dictr attr_dictr6rr7rs   r)DrrZloggingrrplatformrtypingrrrrrrr r r r r r_compatrr _exceptionsrrrrrZ_modelsrrrrrr_noderrrr r!Z _node_poolr"r#Z_otelr$Z _serializerr%r&r'_versionr(Z client_utilsr)r*r-rUr5rr getLoggerrr/r8r`r]rr6r6r6r7sV  8     .