관리-도구
편집 파일: exposition.cpython-38.pyc
U .?�fd � @ sX d dl Z d dlmZ d dlZd dlmZ d dlZd dlZd dlm Z d dl Z d dl Z d dl Z d dl mZmZmZmZmZmZmZmZ d dlmZ d dlmZmZmZ d dl m Z m Z m Z m Z m!Z!m"Z" d d l#m$Z$m%Z%m&Z& d d l'm(Z' d d l)m*Z*m+Z+ d d l,m-Z- dZ.dZ/G dd� de �Z0dd� Z1e+dfe*e2ed�dd�Z3G dd� de%�Z4G dd� de e&�Z5d d � Z6dSe7e7e8ee7 ee7 e2e j9d �d d �Z:d!e+dddde j;dfe8e7e*ee7 ee7 ee7 ee7 e8e2ee&e j<f d"� d#d$�Z=e=Z>e+fe*e?d%�d&d'�Z@e7eee*ge?f e7f d(�d)d*�ZAe7e2d+�d,d-�ZBG d.d/� d/e�ZCe7e*dd0�d1d2�ZDe7e7eeE eee7e7f e?ee eFf eg df d3�d4d5�ZGe7e7eeE eee7e7f e?eg df d6�d7d8�ZHe7e7eeE eee7e7f e?eg df d6�d9d:�ZIdTe7e7eeE eee7e7f e?ee7 ee7 eg df d;�d<d=�ZJde jKdfe7e7eeE eee7e7f e?e7e7ee7 e8e2eg df d>� d?d@�ZLddAeHfe7e7e*eee7ef eeE eddB�dCdD�ZMddAeHfe7e7ee* eee7ef eeE eddB�dEdF�ZNddAeHfe7e7eee7ef eeE eddG�dHdI�ZOe7e7e7ee* eee7ef eeE eddJ�dKdL�ZPdMdN� ZQee7ef dO�dPdQ�ZRd dRlSmTZT dS )U� N)�closing)�BaseHTTPRequestHandler)�ThreadingMixIn)�Any�Callable�Dict�List�Optional�Sequence�Tuple�Union)� HTTPError)�parse_qs� quote_plus�urlparse)� BaseHandler� build_opener� HTTPHandler�HTTPRedirectHandler� HTTPSHandler�Request)� make_server�WSGIRequestHandler� WSGIServer� )� exposition)�CollectorRegistry�REGISTRY)�floatToGoString) �CONTENT_TYPE_LATEST�delete_from_gateway�generate_latest�instance_ip_grouping_key� make_asgi_app� make_wsgi_app�MetricsHandler�push_to_gateway�pushadd_to_gateway�start_http_server�start_wsgi_server�write_to_textfilez(text/plain; version=0.0.4; charset=utf-8c @ s e Zd ZdZdd� ZdS )�_PrometheusRedirectHandleraw Allow additional methods (e.g. PUT) and data forwarding in redirects. Use of this class constitute a user's explicit agreement to the redirect responses the Prometheus client will receive when using it. You should only use this class if you control or otherwise trust the redirect behavior involved and are certain it is safe to full transfer the original request (method and data) to the redirected URL. For example, if you know there is a cosmetic URL redirect in front of a local deployment of a Prometheus server, and all redirects are safe, this is the class to use to handle redirects in that case. The standard HTTPRedirectHandler does not forward request data nor does it allow redirected PUT requests (which Prometheus uses for some operations, for example `push_to_gateway`) because these cannot generically guarantee no violations of HTTP RFC 2616 requirements for the user to explicitly confirm redirects that could have unexpected side effects (such as rendering a PUT request non-idempotent or creating multiple resources not named in the original request). c C sl t |d|�� �}|dkr |dksB|dkr0|dksBt|j||||��t|�dd�|j|jd|jd �}||_ |S ) a� Apply redirect logic to a request. See parent HTTPRedirectHandler.redirect_request for parameter info. If the redirect is disallowed, this raises the corresponding HTTP error. If the redirect can't be determined, return None to allow other handlers to try. If the redirect is allowed, return the new request. This method specialized for the case when (a) the user knows that the redirect will not cause unacceptable side effects for any request method, and (b) the user knows that any request data should be passed through to the redirect. If either condition is not met, this should not be used. �method)�- �. �/ i3 )�GET�HEAD)r- r. r/ )�POST�PUT� z%20T)�headers�origin_req_host� unverifiable�data) �getattr� get_methodr �full_urlr �replacer5 r6 r8 r, ) �self�req�fp�code�msgr5 �newurl�mZ new_request� rD �J/opt/hc_python/lib/python3.8/site-packages/prometheus_client/exposition.py�redirect_requestA s �� �z+_PrometheusRedirectHandler.redirect_requestN)�__name__� __module__� __qualname__�__doc__rF rD rD rD rE r+ + s r+ c C s^ t |�\}}d|kr"| �|d �} || �}d|fg}|sTt|�rTt�|�}|�d� d||fS )z Bake output for metrics output.zname[]� Content-Type)zContent-Encoding�gzip�200 OK)�choose_encoderZrestricted_registry� gzip_acceptedrL �compress�append) �registry� accept_header�accept_encoding_header�params�disable_compression�encoder� content_type�outputr5 rD rD rE � _bake_outputb s rZ F)rR rV �returnc s � �fdd�}|S )z;Create a WSGI app which serves the metrics from a registry.c sf | � d�}| � d�}t| � dd��}| d dkr@d}dg}d }nt�|||� �\}}}|||� |gS ) NZ HTTP_ACCEPTZHTTP_ACCEPT_ENCODINGZ QUERY_STRING� Z PATH_INFOz /favicon.icorM )r\ r\ � )�getr rZ )�environZstart_responserS rT rU �statusr5 rY �rV rR rD rE �prometheus_appt s z%make_wsgi_app.<locals>.prometheus_apprD )rR rV rb rD ra rE r$ q s r$ c @ s e Zd ZdZdd� ZdS )�_SilentHandlerz(WSGI handler that does not log requests.c G s dS �z Log nothing.NrD �r= �format�argsrD rD rE � log_message� s z_SilentHandler.log_messageN)rG rH rI rJ rh rD rD rD rE rc � s rc c @ s e Zd ZdZdZdS )�ThreadingWSGIServerz Thread per request HTTP server.TN)rG rH rI rJ �daemon_threadsrD rD rD rE ri � s ri c C s8 t j| |t jt jd�}tt|��\}}}}}||d fS )z8Automatically select address family depending on address)�type�flagsr )�socket� getaddrinfo� SOCK_STREAM� AI_PASSIVE�next�iter)�address�port�infos�family�_ZsockaddrrD rD rE �_get_best_family� s rx )�certfile�keyfile�protocol�cafile�capath�client_auth_requiredr[ c C sD t j|d�}|dk s |dk r|z|�||� W q� tk rx } z.t|�}t|�} |d|�d|�d| � ���W 5 d}~X Y q�X nVz|jt jjd� W n@ tk r� } z"t|�}t|�} |d| � ���W 5 d}~X Y nX |r�t j |_ z|j | |d� W nN tk �r> } z.t|�}t|�} |d | �d |�d| � ���W 5 d}~X Y nX |S ) zLoad context supports SSL.�r{ Nz+Cannot load CA certificate chain from file z or directory z: )�purposez*Cannot load default CA certificate chain: �ry rz z$Cannot load server certificate file z or its private key file ) �ssl� SSLContext�load_verify_locations�IOErrorrk �str�load_default_certs�Purpose� CLIENT_AUTH� CERT_REQUIRED� verify_mode�load_cert_chain) ry rz r{ r| r} r~ Zssl_cxt�exc�exc_typerA rD rD rE � _get_ssl_ctx� s. . ,r� z0.0.0.0) rt �addrrR ry rz � client_cafile� client_capathr{ r~ r[ c C s� G dd� dt �} t|| �\| _}t|�} t|| | | td�} |rf|rft||||||�} | j| jdd�| _t j | j d�} d| _ | � � | | fS )z?Starts a WSGI server for prometheus metrics as a daemon thread.c @ s e Zd ZdZdS )z$start_wsgi_server.<locals>.TmpServerz<Copy of ThreadingWSGIServer to update address_family locallyN)rG rH rI rJ rD rD rD rE � TmpServer� s r� )Z handler_classT)� server_side)�target)ri rx �address_familyr$ r rc r� � wrap_socketrm � threading�Thread� serve_forever�daemon�start)rt r� rR ry rz r� r� r{ r~ r� ZappZhttpd�context�trD rD rE r) � s r) �rR r[ c C s� dd� }g }| � � D �]�}z�|j}|j}|dkr:|d }n>|dkrP|d }d}n(|dkr^d}n|d krld }n |d krxd }|�d �||j�dd��dd��� |�d|� d|� d�� i }|jD ]F}dD ].}|j|j| kr�|�|g ��||�� q�q�|�||�� q�W n: t k �rB } z| j �p&d|f | _ � W 5 d} ~ X Y nX t |� � �D ]T\}} |�d�|j||j�dd��dd��� |�d|j� |� d�� |� | � �qPqd�|��d�S )zHReturns the metrics from the registry in latest text format as a string.c S sx | j r,d�d�dd� t| j �� �D ���}nd}d}| jd k rXdtt| j�d �d��}| j� |� dt | j �� |� d �S ) Nz{{{0}}}�,c S s4 g | ],\}}d � ||�dd��dd��dd���qS )z{}="{}"�\�\\� �\n�"z\")rf r< ��.0�k�vrD rD rE � <listcomp>� s � �z8generate_latest.<locals>.sample_line.<locals>.<listcomp>r\ r4 i� �dr� ) �labelsrf �join�sorted�items� timestamp�int�float�namer �value)�lineZlabelstrr� rD rD rE � sample_line� s �� z$generate_latest.<locals>.sample_line�counterZ_total�info�_infoZgaugeZstatesetZgaugehistogramZ histogram�unknownZuntypedz # HELP {} {} r� r� r� r� z# TYPE r4 )Z_createdZ_gsumZ_gcount)r\ Nz# HELP {}{} {} z gauge r\ �utf-8)Zcollectr� rk rQ rf � documentationr< Zsamples� setdefault� Exceptionrg r� r� �extendr� �encode) rR r� rY ZmetricZmname�mtypeZ om_samples�s�suffix� exception�linesrD rD rE r! � sN � �r! )rS r[ c C sF | pd} | � d�D ]*}|� d�d �� dkrtjtjf S qttfS )Nr\ r� �;r z application/openmetrics-text)�split�strip� openmetricsr! r )rS �acceptedrD rD rE rN ) s � rN )rT r[ c C s< | pd} | � d�D ]$}|� d�d �� �� dkr dS qdS )Nr\ r� r� r rL TF)r� r� �lower)rT r� rD rD rE rO 2 s rO c @ sR e Zd ZU dZeZeed< dd�dd�Ze e dd�dd �Z e ee d �d d ��ZdS ) r% z2HTTP handler that gives metrics from ``REGISTRY``.rR N�r[ c C s� | j }| j�d�}| j�d�}tt| j�j�}t||||d�\}}}| �t |� d�d �� |D ]}| j |� q`| � � | j �|� d S )N�AcceptzAccept-EncodingFr4 r )rR r5 r^ r r �path�queryrZ Z send_responser� r� Z send_headerZ end_headers�wfile�write) r= rR rS rT rU r` r5 rY �headerrD rD rE �do_GET> s zMetricsHandler.do_GET)rf rg r[ c G s dS rd rD re rD rD rE rh M s zMetricsHandler.log_messager� c C s" t | j�}t|| tfd|i�}|S )zWReturns a dynamic MetricsHandler class tied to the passed registry. rR )r� rG rk �object)�clsrR �cls_nameZMyMetricsHandlerrD rD rE �factoryP s �zMetricsHandler.factory)rG rH rI rJ r rR r �__annotations__r� r� r rh � classmethodrk r� rD rD rD rE r% : s r% )r� rR r[ c C sj | � dt �� � dt�� j� �}t|d��}|�t|�� W 5 Q R X t jdkrZt � || � n t � || � dS )z�Write metrics to the given path. This is intended for use with the Node exporter textfile collector. The path must end in .prom for the textfile collector to process it.�.�wb�ntN) �os�getpidr� �current_thread�ident�openr� r! r� r< �rename)r� rR Ztmppath�frD rD rE r* ` s r* )�urlr, �timeoutr5 r8 � base_handlerr[ c s d d�� �����fdd� }|S )Nr� c sl t ��d�} �fdd�| _�D ]\}}| �||� q t� �j| �d�}|jdkrhtd|j� d|j� ���d S )N)r8 c s � S )NrD rD )r, rD rE �<lambda>z r] z/_make_handler.<locals>.handle.<locals>.<lambda>)r� i� z error talking to pushgateway: r4 )r r: � add_headerr r� r@ �OSErrorrA )�requestr� r� �resp�r� r8 r5 r, r� r� rD rE �handlex s z _make_handler.<locals>.handlerD )r� r, r� r5 r8 r� r� rD r� rE � _make_handlerp s r� )r� r, r� r5 r8 r[ c C s t | ||||t�S )z�Default handler that implements HTTP/HTTPS connections. Used by the push_to_gateway functions. Can be re-used by other handlers.)r� r �r� r, r� r5 r8 rD rD rE �default_handler� s r� c C s t | ||||t�S )a Handler that automatically trusts redirect responses for all HTTP methods. Augments standard HTTPRedirectHandler capability by permitting PUT requests, preserving the method upon redirect, and passing through all headers and data from the original request. Only use this handler if you control or trust the source of redirect responses you encounter when making requests via the Prometheus client. This handler will simply repeat the identical request, including same method and data, to the new redirect URL.)r� r+ r� rD rD rE � passthrough_redirect_handler� s r� )r� r, r� r5 r8 �username�passwordr[ c s � ������fdd�}|S )z�Handler that implements HTTP/HTTPS connections with Basic Auth. Sets auth headers using supplied 'username' and 'password', if set. Used by the push_to_gateway functions. Can be re-used by other handlers.c sX �dk rB�dk rB�� d�� �� � } t�| �}d| }��d|f� t����� �� dS )z1Handler that implements HTTP Basic Auth. N�:s Basic � Authorization)r� �base64� b64encoderQ r� )Z auth_valueZ auth_token� auth_header�r8 r5 r, r� r� r� r� rD rE r� � s z"basic_auth_handler.<locals>.handlerD )r� r, r� r5 r8 r� r� r� rD r� rE �basic_auth_handler� s r� ) r� r, r� r5 r8 ry rz r| r{ �insecure_skip_verifyr[ c C sd t j|d�} |dk r | �|� n| �� | r:d| _t j| _| j||d� t| d�} t | ||||| �S )aL Handler that implements an HTTPS connection with TLS Auth. The default protocol (ssl.PROTOCOL_TLS_CLIENT) will also enable ssl.CERT_REQUIRED and SSLContext.check_hostname by default. This can be disabled by setting insecure_skip_verify to True. Both this handler and the TLS feature on pushgateay are experimental.r NFr� )r� ) r� r� r� r� �check_hostname� CERT_NONEr� r� r r� ) r� r, r� r5 r8 ry rz r| r{ r� r� �handlerrD rD rE �tls_auth_handler� s r� � )�gateway�jobrR � grouping_keyr� r� r[ c C s t d| |||||� dS )aN Push metrics to the given pushgateway. `gateway` the url for your push gateway. Either of the form 'http://pushgateway.local', or 'pushgateway.local'. Scheme defaults to 'http' if none is provided `job` is the job label to be attached to all pushed metrics `registry` is an instance of CollectorRegistry `grouping_key` please see the pushgateway documentation for details. Defaults to None `timeout` is how long push will attempt to connect before giving up. Defaults to 30s, can be set to None for no timeout. `handler` is an optional function which can be provided to perform requests to the 'gateway'. Defaults to None, in which case an http or https request will be carried out by a default handler. If not None, the argument must be a function which accepts the following arguments: url, method, timeout, headers, and content May be used to implement additional functionality not supported by the built-in default handler (such as SSL client certicates, and HTTP authentication mechanisms). 'url' is the URL for the request, the 'gateway' argument described earlier will form the basis of this URL. 'method' is the HTTP method which should be used when carrying out the request. 'timeout' requests not successfully completed after this many seconds should be aborted. If timeout is None, then the handler should not set a timeout. 'headers' is a list of ("header-name","header-value") tuples which must be passed to the pushgateway in the form of HTTP request headers. The function should raise an exception (e.g. IOError) on failure. 'content' is the data which should be used to form the HTTP Message Body. This overwrites all metrics with the same job and grouping_key. This uses the PUT HTTP method.r3 N�� _use_gateway�r r rR r r� r� rD rD rE r& � s .r& c C s t d| |||||� dS )a" PushAdd metrics to the given pushgateway. `gateway` the url for your push gateway. Either of the form 'http://pushgateway.local', or 'pushgateway.local'. Scheme defaults to 'http' if none is provided `job` is the job label to be attached to all pushed metrics `registry` is an instance of CollectorRegistry `grouping_key` please see the pushgateway documentation for details. Defaults to None `timeout` is how long push will attempt to connect before giving up. Defaults to 30s, can be set to None for no timeout. `handler` is an optional function which can be provided to perform requests to the 'gateway'. Defaults to None, in which case an http or https request will be carried out by a default handler. See the 'prometheus_client.push_to_gateway' documentation for implementation requirements. This replaces metrics with the same name, job and grouping_key. This uses the POST HTTP method.r2 Nr r rD rD rE r' s r' )r r r r� r� r[ c C s t d| |d|||� dS )a� Delete metrics from the given pushgateway. `gateway` the url for your push gateway. Either of the form 'http://pushgateway.local', or 'pushgateway.local'. Scheme defaults to 'http' if none is provided `job` is the job label to be attached to all pushed metrics `grouping_key` please see the pushgateway documentation for details. Defaults to None `timeout` is how long delete will attempt to connect before giving up. Defaults to 30s, can be set to None for no timeout. `handler` is an optional function which can be provided to perform requests to the 'gateway'. Defaults to None, in which case an http or https request will be carried out by a default handler. See the 'prometheus_client.push_to_gateway' documentation for implementation requirements. This deletes metrics with the given job and grouping_key. This uses the DELETE HTTP method.�DELETENr )r r r r� r� rD rD rE r 3 s r )r, r r rR r r� r� r[ c C s� t |�}|jr|jdkr"d|� �}|�d�}dj|ftd|��� }d} | dkrb|d krZt}t|�} |d krni }|d�d d � t|� � �D ��7 }||| |d t fg| d �� d S ) N)�http�httpszhttp://�/z{}/metrics/{}/{}r r] r r\ c s s* | ]"\}}d j tt|�t|��� V qdS )z/{}/{}N)rf �_escape_grouping_keyr� r� rD rD rE � <genexpr>i s �z _use_gateway.<locals>.<genexpr>rK r� ) r �scheme�rstriprf r r r! r� r� r� r ) r, r r rR r r� r� Z gateway_urlr� r8 rD rD rE r P s, � �r c C sJ |dkr| d dfS d|kr:| d t �|�d���d�fS | t|�fS d S )Nr\ z@base64�=r r� )r� �urlsafe_b64encoder� �decoder )r� r� rD rD rE r s s r r� c C s\ t t�tjtj���>} tjdkr,| �d� n | �d� d| �� d iW 5 Q R � S Q R X dS )z>Grouping key with instance set to the IP Address of this host.�darwin)z10.255.255.255r )� localhostr �instancer N)r rm �AF_INET� SOCK_DGRAM�sys�platform�connect� getsockname)r� rD rD rE r" ~ s r" )r# )NNF)NN)Ur� � contextlibr rL Z http.serverr r� rm � socketserverr r� r r� �typingr r r r r r r r � urllib.errorr � urllib.parser r r �urllib.requestr r r r r r Zwsgiref.simple_serverr r r r� r rR r r �utilsr �__all__r r+ rZ �boolr$ rc ri rx r� r� r� r� �PROTOCOL_TLS_SERVERr� r) r( �bytesr! rN rO r% r* r� rk r� r� r� r� �PROTOCOL_TLS_CLIENTr� r& r' r r r r" Zasgir# rD rD rD rE �<module> s@ ( 7 �� +� � >" & � � � � � #� � &�� 5�� "�� � #