B .l\9@sddlmZddlZddlmZddlZddlZddlZddlZddl m Z m Z ddl m Z ddlmZddlmZyPdd lmZmZdd lmZdd lmZmZmZdd lmZdd lmZmZWnZek rdd l mZmZdd l!mZdd l"mZmZmZddl#mZmZmZYnXe$dZ%ej&dkZ'efddZ(Gddde Z)defddZ*efddZ+ddZ,GdddeZ-GdddeeZ.defd d!Z/d"d#Z0d$d%Z1d3d&d'Z2dd(e1fd)d*Z3dd(e1fd+d,Z4dd(e1fd-d.Z5d/d0Z6d1d2Z7dS)4)unicode_literalsN)closing) make_serverWSGIRequestHandler) exposition)REGISTRY)floatToGoString)BaseHTTPRequestHandler HTTPServer)ThreadingMixIn) build_openerRequest HTTPHandler) quote_plus)parse_qsurlparse)rrrz(text/plain; version=0.0.4; charset=utf-8)csfdd}|S)z;Create a WSGI app which serves the metrics from a registry.c sjt|dd}}t|d\}}d|kr<||d}||}td}td|fg}||||gS)NZ QUERY_STRINGZ HTTP_ACCEPTzname[]z200 OKz Content-type)rgetchoose_encoderrestricted_registrystr) environZstart_responseparamsrencoder content_typeoutputZstatusheaders)registry;lib/python3.7/site-packages/prometheus_client/exposition.pyprometheus_app's z%make_wsgi_app..prometheus_appr")r!r$r")r!r# make_wsgi_app%s r%c@seZdZdZddZdS)_SilentHandlerz(WSGI handler that does not log requests.cGsdS)z Log nothing.Nr")selfformatargsr"r"r# log_message9sz_SilentHandler.log_messageN)__name__ __module__ __qualname____doc__r*r"r"r"r#r&6sr&rcCs8t|}t|||td}tj|jd}d|_|dS)z?Starts a WSGI server for prometheus metrics as a daemon thread.)Z handler_class)targetTN)r%rr& threadingThread serve_foreverdaemonstart)portaddrr!Zapphttpdtr"r"r#start_wsgi_server=s r9c Csdd}g}xv|D]h}y|j}|j}|dkr>|d}n>|dkrT|d}d}n(|dkrbd}n|d krpd }n |d kr|d }|d ||jdddd|d||i}xR|jD]H}xBdD],}|j|j|kr||g||PqW|||qWWn:t k rF} z| j p*d|f| _ Wdd} ~ XYnXx8t | D](\}} |d|j|| | qVWqWd|dS)zHReturns the metrics from the registry in latest text format as a string.cSsn|jr,ddddt|jD}nd}d}|jdk rVdtt|jd}d|j|t |j |S) Nz{{{0}}},c Ss4g|],\}}d||ddddddqS)z {0}="{1}"\z\\ z\n"z\")r(replace).0kvr"r"r# Lsz8generate_latest..sample_line..rz {0:d}iz{0}{1} {2}{3} ) labelsr(joinsorteditems timestampintfloatnamer value)sZlabelstrrGr"r"r# sample_lineIs z$generate_latest..sample_lineZcounterZ_totalinfoZ_infoZgaugeZstatesetZgaugehistogramZ histogramunknownZuntypedz# HELP {0} {1} r;z\\r<z\nz# TYPE {0} {1} )Z_createdZ_gsumZ_gcount)rNz# TYPE {0}{1} gauge rzutf-8)ZcollectrJtypeappendr(Z documentationr>Zsamples setdefault Exceptionr)rErFextendrDencode) r!rMrZmetricZmnameZmtypeZ om_samplesrLsuffixZ exceptionlinesr"r"r#generate_latestFsD   rXcCsF|pd}x4|dD]&}|dddkrtjtjfSqWttfS)Nrr:;rzapplication/openmetrics-text)splitstrip openmetricsrXCONTENT_TYPE_LATEST)Z accept_headerZacceptedr"r"r#rs  rc@s0eZdZdZeZddZddZeddZ dS) MetricsHandlerz2HTTP handler that gives metrics from ``REGISTRY``.cCs|j}tt|jj}t|jd\}}d|kr@||d}y ||}Wn| ddYnX| d| d|| |j |dS)NZAcceptzname[]izerror generating metric outputz Content-Type)r!rrpathZqueryrr rrZ send_errorZ send_responseZ send_headerZ end_headersZwfilewrite)r'r!rrrrr"r"r#do_GETs    zMetricsHandler.do_GETcGsdS)z Log nothing.Nr")r'r(r)r"r"r#r*szMetricsHandler.log_messagecCs"t|j}t||tfd|i}|S)zWReturns a dynamic MetricsHandler class tied to the passed registry. r!)rr+rPobject)clsr!Zcls_nameZMyMetricsHandlerr"r"r#factorys   zMetricsHandler.factoryN) r+r,r-r.rr!rbr* classmethodrer"r"r"r#r^s r^c@seZdZdZdZdS)_ThreadingSimpleServerzThread per request HTTP server.TN)r+r,r-r.Zdaemon_threadsr"r"r"r#rgsrgcCs8t|}t||f|}tj|jd}d|_|dS)z?Starts an HTTP server for prometheus metrics as a daemon thread)r/TN)r^rergr0r1r2r3r4)r5r6r!ZCustomMetricsHandlerr7r8r"r"r#start_http_servers  rhc CsLd|ttjf}t|d}|t|WdQRXt||dS)zWrite 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.z%s.%s.%swbN) osgetpidr0Zcurrent_threadZidentopenrarXrename)r`r!Ztmppathfr"r"r#write_to_textfiles rocsfdd}|S)zDefault handler that implements HTTP/HTTPS connections. Used by the push_to_gateway functions. Can be re-used by other handlers.csltd}fdd|_xD]\}}|||q Wttj|d}|jdkrhtd|j|j dS)N)datacsS)Nr"r")methodr"r#sz1default_handler..handle..)timeoutiz%error talking to pushgateway: {0} {1}) rZ get_methodZ add_headerr rrlcodeIOErrorr(msg)Zrequestr@rAZresp)rpr rqrsurlr"r#handles  zdefault_handler..handler")rwrqrsr rprxr")rpr rqrsrwr#default_handlers rycsfdd}|S)zHandler 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.csXdk rBdk rBdd}t|}d|}d|gtdS)z1Handler that implements HTTP Basic Auth. Nz{0}:{1}zutf-8sBasic Z Authorization)r(rUbase64Z b64encoderQry)Z auth_valueZ auth_tokenZ auth_header)rpr rqpasswordrsrwusernamer"r#rxs  z"basic_auth_handler..handler")rwrqrsr rpr|r{rxr")rpr rqr{rsrwr|r#basic_auth_handlers r}cCstd||||||dS)aNPush 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.ZPUTN) _use_gateway)gatewayjobr! grouping_keyrshandlerr"r"r#push_to_gateways)rcCstd||||||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.ZPOSTN)r)rrr!rrsrr"r"r#pushadd_to_gateway!srcCstd||d|||dS)aDelete 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.DELETEN)r)rrrrsrr"r"r#delete_from_gateway;src Cst|}|jrtr&|jdkr&d|}d|t|}d} |dkrJt|} |dkrVi}|dddt|D7}||||d t fg| d dS) N)ZhttpZhttpsz http://{0}z{0}/metrics/job/{1}rrcss.|]&\}}dtt|tt|VqdS)z/{0}/{1}N)r(rr)r?r@rAr"r"r# `sz_use_gateway..z Content-Type)rwrqrsr rp) rZschemePYTHON26_OR_OLDERr(rrXrDrErFr]) rqrrr!rrsrZ gateway_urlrwrpr"r"r#rSs rc Cs:tttjtj}|dd|diSQRXdS)z>Grouping key with instance set to the IP Address of this host.)Z localhostrinstancerN)rsocketZAF_INETZ SOCK_DGRAMZconnectZ getsockname)rLr"r"r#instance_ip_grouping_keyis r)NN)8Z __future__rrz contextlibrrjrsysr0Zwsgiref.simple_serverrrr\rr!rZutilsr ZBaseHTTPServerr r Z SocketServerr Zurllib2r rrZurllibrrr ImportErrorZ http.serverZ socketserverZurllib.requestZ urllib.parserr] version_inforr%r&r9rXrr^rgrhroryr}rrrrrr"r"r"r#sR           = '     +