Xc@`sdZddlmZddlmZddlmZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlmZddlmZddlmZddlZdd lmZdd lmZdd lmZdd lmZd ZdejZdejZdej Z!dej"Z#dZ$dej%Z&idd6dd6dd6dd6Z'dZ(dZ)de*fdYZ+dej,fdYZ-dS(sTensorBoard server handler logic. TensorboardHandler contains all the logic for serving static files off of disk and for handling the API calls to endpoints like /tags that require information about loaded events. i(tabsolute_import(tdivision(tprint_functionN(turllib(txrange(t text_format(tlogging(tresource_loader(tevent_accumulator(t float_wrappers/runst/s/individualImages image/bmptbmps image/giftgifs image/jpegtjpegs image/pngtpngsapplication/octet-streamcC`s"tjd|}tj|tS(N(timghdrtwhattNonet_IMGHDR_TO_MIMETYPEtgett_DEFAULT_IMAGE_MIMETYPE(tencoded_image_stringt image_type((sl/tmp/pip-build-UG86a1/tensorflow/tensorflow-0.6.0.data/purelib/tensorflow/tensorboard/tensorboard_handler.pyt_content_type_for_image@st _OutputFormatcB`seZdZdZdZRS(sAn enum used to list the valid output formats for API calls. Not all API calls support all formats (for example, only scalars and compressed histograms support CSV). tjsontcsv(t__name__t __module__t__doc__tJSONtCSV(((sl/tmp/pip-build-UG86a1/tensorflow/tensorflow-0.6.0.data/purelib/tensorflow/tensorboard/tensorboard_handler.pyREstTensorboardHandlercB`seZdZdZdZdZddZddZddZdZ d Z d Z d Z d Z d ZdZdZdZdZdZdZRS(sHandler class for use with BaseHTTPServer.HTTPServer. This is essentially a thin wrapper around calls to an EventMultiplexer object as well as serving files off disk. cG`s ||_tjj||dS(N(t _multiplexertBaseHTTPServertBaseHTTPRequestHandlert__init__(tselft multiplexertargs((sl/tmp/pip-build-UG86a1/tensorflow/tensorflow-0.6.0.data/purelib/tensorflow/tensorboard/tensorboard_handler.pyR$Vs cC`srg}xet|D]W\}}|ji|jd6|jd6|jd6|jd6|j|||d6qW|S(slBuilds a JSON-serializable object with information about run_images. Args: run_images: A list of event_accumulator.ImageValueEvent objects. run: The name of the run. tag: The name of the tag the images all belong to. Returns: A list of dictionaries containing the wall time, step, URL, width, and height for each image. t wall_timetsteptwidththeighttquery(t enumeratetappendR(R)R*R+t_query_for_individual_image(R%t run_imagestrunttagtresponsetindext run_image((sl/tmp/pip-build-UG86a1/tensorflow/tensorflow-0.6.0.data/purelib/tensorflow/tensorboard/tensorboard_handler.pyt_image_response_for_run\s      cC`sItjjtj}tjj|}tjj||g}||kS(sCheck path is safe (stays within current directory). This is for preventing directory-traversal attacks. Args: path: The path to check for safety. Returns: True if the given path stays within the current directory, and false if it would escape to a higher directory. E.g. _path_is_safe('index.html') returns true, but _path_is_safe('../../../etc/password') returns false. (tostpathtabspathtcurdirt commonprefix(R%R8tbaset absolute_pathtprefix((sl/tmp/pip-build-UG86a1/tensorflow/tensorflow-0.6.0.data/purelib/tensorflow/tensorboard/tensorboard_handler.pyt _path_is_safeus icC`stj}tjd|dd}|j||j|j}|j||jd||jdt||jdd|j |j j|dS( sWrites the given content as gzip response using the given content type. Args: content: The content to respond with. content_type: The mime type of the content. code: The numeric HTTP status code to use. tfileobjtmodetws Content-TypesContent-LengthsContent-EncodingtgzipN( tStringIORCtGzipFiletwritetclosetgetvaluet send_responset send_headertlent end_headerstwfile(R%tcontentt content_typetcodetouttft gzip_content((sl/tmp/pip-build-UG86a1/tensorflow/tensorflow-0.6.0.data/purelib/tensorflow/tensorboard/tensorboard_handler.pyt_send_gzip_responses      cC`sitjtj|}|j||jdd|jdt||j|jj |dS(sWrites out the given object as JSON using the given HTTP status code. This also replaces special float values with stringified versions. Args: obj: The object to respond with. code: The numeric HTTP status code to use. s Content-Typesapplication/jsonsContent-LengthN( RtdumpsR tWrapSpecialFloatsRIRJRKRLRMRF(R%tobjRPtoutput((sl/tmp/pip-build-UG86a1/tensorflow/tensorflow-0.6.0.data/purelib/tensorflow/tensorboard/tensorboard_handler.pyt_send_json_responses   cC`sQ|j||jdd|jdt||j|jj|dS(s.Writes out the given string, which represents CSV data. Unlike _send_json_response, this does *not* perform the CSV serialization for you. It only sets the proper headers. Args: serialized_csv: A string containing some CSV data. code: The numeric HTTP status code to use. s Content-Typestext/csvsContent-LengthN(RIRJRKRLRMRF(R%tserialized_csvRP((sl/tmp/pip-build-UG86a1/tensorflow/tensorflow-0.6.0.data/purelib/tensorflow/tensorboard/tensorboard_handler.pyt_send_csv_responses  cC`s|jd}|jd}|jj||}|jdtjkrtj}tj|}|jdddg|j ||j |j n |j |dS(s9Given a tag and single run, return array of ScalarEvents.R2R1tformats Wall timetSteptValueN( RR!tScalarsRRRDRtwritertwriterowt writerowsR[RHRY(R%t query_paramsR2R1tvaluest string_ioR`((sl/tmp/pip-build-UG86a1/tensorflow/tensorflow-0.6.0.data/purelib/tensorflow/tensorboard/tensorboard_handler.pyt_serve_scalarss  cC`s|jdd}|dkr2|jdddSy|jj|}Wntk ri|jddSXtj|}|j |ddS(s?Given a single run, return the graph definition in json format.R1is!query parameter "run" is requiredNis text/plain( RRt send_errorR!tGrapht ValueErrorRIRtMessageToStringRT(R%RcR1tgrapht graph_pbtxt((sl/tmp/pip-build-UG86a1/tensorflow/tensorflow-0.6.0.data/purelib/tensorflow/tensorboard/tensorboard_handler.pyt _serve_graphs   cC`sD|jd}|jd}|jj||}|j|dS(s@Given a tag and single run, return an array of histogram values.R2R1N(RR!t HistogramsRY(R%RcR2R1Rd((sl/tmp/pip-build-UG86a1/tensorflow/tensorflow-0.6.0.data/purelib/tensorflow/tensorboard/tensorboard_handler.pyt_serve_histogramssc C`sK|jd}|jd}|jj||}|jdtjkr:tj}tj|}ddg}|rt|dj }x/t |D]} |d| d| g7}qWn|j |xW|D]O} | j | j g} x'| j D]} | | j| jg7} qW|j | qW|j|jn |j|d S( sEGiven a tag and single run, return an array of compressed histograms.R2R1R\s Wall timeR]isEdge %d basis pointss Edge %d valueN(RR!tCompressedHistogramsRRRDRR`RKtcompressed_histogram_valuesRRaR(R)t rank_in_bpstvalueR[RHRY( R%RcR2R1tcompressed_histogramsReR`theaderst bucket_counttitcompressed_histogramtrowRs((sl/tmp/pip-build-UG86a1/tensorflow/tensorflow-0.6.0.data/purelib/tensorflow/tensorboard/tensorboard_handler.pyt_serve_compressed_histogramss&    cC`sY|jd}|jd}|jj||}|j|||}|j|dS(szGiven a tag and list of runs, serve a list of images. Note that the images themselves are not sent; instead, we respond with URLs to the images. The frontend should treat these URLs as opaque and should not try to parse information about them or generate them itself, as the format may change. Args: query_params: The query parameters as a dict. R2R1N(RR!tImagesR6RY(R%RcR2R1timagesR3((sl/tmp/pip-build-UG86a1/tensorflow/tensorflow-0.6.0.data/purelib/tensorflow/tensorboard/tensorboard_handler.pyt _serve_imagess cC`s|jd}|jd}t|jd}|jj|||}|j}t|}|jd|jd||jdt||j |j j |dS(sServes an individual image.R2R1R4is Content-TypesContent-LengthN( RtintR!R{RRRIRJRKRLRMRF(R%RcR2R1R4timageRRO((sl/tmp/pip-build-UG86a1/tensorflow/tensorflow-0.6.0.data/purelib/tensorflow/tensorboard/tensorboard_handler.pyt _serve_images    cC`s+tjji|d6|d6|d6}|S(sBuilds a URL for accessing the specified image. This should be kept in sync with _serve_image. Note that the URL is *not* guaranteed to always return the same image, since images may be unloaded from the reservoir as new images come in. Args: run: The name of the run. tag: The tag. index: The index of the image. Negative values are OK. Returns: A string representation of a URL that will load the index-th sampled image in the given run with the given tag. R1R2R4(Rtparset urlencode(R%R1R2R4t query_string((sl/tmp/pip-build-UG86a1/tensorflow/tensorflow-0.6.0.data/purelib/tensorflow/tensorboard/tensorboard_handler.pyR/#s   cC`s|j|jjdS(sReturn a JSON object about runs and tags. Returns a mapping from runs to tagType to list of tags for that run. Returns: {runName: {images: [tag1, tag2, tag3], scalars: [tagA, tagB, tagC], histograms: [tagX, tagY, tagZ]}} N(RYR!tRuns(R%tunused_query_params((sl/tmp/pip-build-UG86a1/tensorflow/tensorflow-0.6.0.data/purelib/tensorflow/tensorboard/tensorboard_handler.pyt _serve_runs:s cC`s|jddS(s9Serves the index page (i.e., the tensorboard app itself).s/dist/index.htmlN(t_serve_static_file(R%R((sl/tmp/pip-build-UG86a1/tensorflow/tensorflow-0.6.0.data/purelib/tensorflow/tensorboard/tensorboard_handler.pyt _serve_indexFscC`s|jddS(s)Serves the JavaScript for the index page.s /dist/app.jsN(R(R%R((sl/tmp/pip-build-UG86a1/tensorflow/tensorflow-0.6.0.data/purelib/tensorflow/tensorboard/tensorboard_handler.pyt _serve_jsJscC`s|jd}|j|s?tjd||jddS|jdrftjjd|}ntjjd|}yt j |}Wn/t k rtjd||jddSX|j d t j|d pd }|jd ||j|jj|dS( sServes the static file located at the given path. Args: path: The path of the static file, relative to the tensorboard/ directory. R spath %s not safe, sending 404iNtexternals../t tensorboardspath %s not found, sending 404iisapplication/octet-streams Content-Type(tlstripR?RtinfoRgt startswithR7R8tjoinRt load_resourcetIOErrorRIt mimetypest guess_typeRJRLRMRF(R%R8tcontentstmimetype((sl/tmp/pip-build-UG86a1/tensorflow/tensorflow-0.6.0.data/purelib/tensorflow/tensorboard/tensorboard_handler.pyRNs&     cC`s4tj|j}|j}|jdr7|d }ni |jt6|jt6|jt6|j t 6|j t 6|j t6|jt6|jd6|jd6}||kr#tj|j}xZ|D]R}t||}|dkr|jdd||fdS||d ||sH