o i e!@sdZddlZddlZddlmZddlmZddlmZddlm Z ddlm Z ddl m Z dd l mZd d Zd d ZddZddZ    dddZ dddZ dddZdS)z;Communicating tracebacks and source code with debug server.N)debug_service_pb2) debug_pb2)common)debug_service_pb2_grpc) source_utils)gfile) tfprof_loggercCst|}t|_||_|j|_|j|_ z%t |d}|j | WdWdS1s3wYWdStyDYdSw)Nr)rStatsocket gethostnamehost file_path mtime_nsec last_modifiedlengthbytesOpenlinesextendread splitlinesIOError)rsource_file_proto file_statfrX/var/www/myenv/lib/python3.10/site-packages/tensorflow/python/debug/lib/source_remote.py_load_debugged_source_files  & rcCs||vr t|||<||SN)len)string string_to_idrrr _string_to_id+s r#c Cszi}d|d<|D]}|\}}}}|jjjt|||t||t||dq|j}|D]\} } | dur6| nd|| <q,dS)zFormat a traceback stack for a `CallTraceback` proto. Args: origin_stack: The stack list as returned by `traceback.extract_stack()`. call_traceback_proto: A `CallTraceback` proto whose fields are to be populated. rN)file_idlineno function_idline_id) origin_stacktracesaddr#origin_id_to_stringitems) r)call_traceback_protor"framerr% func_name line_text id_to_stringkeyvaluerrr_format_origin_stack1s r5csLt}|D]}|jD]}||jq qfdd|D}dd|D}|S)a4Extract source file paths outside TensorFlow Python library. Args: code_defs: An iterable of `CodeDef` protos, i.e., an iterable of stack traces. id_to_string: A proto map from integer ids to strings. Returns: An iterable of source file paths outside the TensorFlow Python library. c3s|]}|VqdSrr).0r$r2rr WszC_source_file_paths_outside_tensorflow_py_library..css(|]}t|st|r|VqdSr)rguess_is_tensorflow_py_libraryrExists)r6rrrrr8Xs )setr*r+r$) code_defsr2file_idscode_deftrace non_tf_filesrr7r0_source_file_paths_outside_tensorflow_py_libraryHs  rAFTcCs0t|ts|g}dd|D}|rtjjntjj}|r"tj|ddnd}tj||||r.|jndd}t |||rpt } | t dd|j jD|j j| t |jg|jg} | D]} t} t| | j| | q\|D]#} d d g}tj| |d }t|}|||r| D]} || qqrdS) arSend the tracebacks of a TensorFlow execution call. To gRPC debug server(s). This applies to graph execution (`tf.Session.run()`) calls and eager execution calls. If `send_source`, also sends the underlying source files outside the TensorFlow library. Args: destinations: gRPC destination addresses, a `str` or a `list` of `str`s, e.g., "localhost:4242". If a `list`, gRPC requests containing the same `CallTraceback` proto payload will be sent to all the destinations. origin_stack: The traceback stack for the origin of the execution call. For graph execution, this is the traceback of the `tf.Session.run()` invocation. For eager execution, this is the traceback of the Python line that executes the eager operation. is_eager_execution: (`bool`) whether an eager execution call (i.e., not a `tf.Session.run` or derived methods) is being sent. call_key: The key of the execution call, as a string. For graph execution, this is a string describing the feeds, fetches (and targets) names of the `tf.Session.run` call. For eager execution, this is ignored. graph: A Python `tf.Graph` object (i.e., *not* a `tf.compat.v1.GraphDef`), which contains op tracebacks, if applicable. send_source: Whether the source files involved in the op tracebacks but outside the TensorFlow library are to be sent. cSs.g|]}|tjr|ttjdn|qSr) startswithrGRPC_URL_PREFIXr )r6destrrr s  z)_send_call_tracebacks..F)add_trainable_varN) call_typecall_keygraph_traceback graph_versioncss|]}|jVqdSr)r>)r6 log_entryrrrr8sz(_send_call_tracebacks..)zgrpc.max_receive_message_length)zgrpc.max_send_message_lengthrL)options) isinstancelistr CallTracebackEAGER_EXECUTIONGRAPH_EXECUTIONrmerge_default_with_oplogversionr5r;updaterArI log_entriesr2r)r,rDebuggedSourceFilesr source_filesr+appendgrpcinsecure_channelrEventListenerStubSendTracebacksSendSourceFiles) destinationsr)is_eager_executionrHgraph send_sourcerGrIcall_tracebacksource_file_pathsdebugged_source_filesrrX destinationno_max_message_sizeschannelstubrrr_send_call_tracebacks^sd          rjcCst||d|||ddS)aSend the tracebacks of a graph execution call to debug server(s). Args: destinations: gRPC destination addresses, a `str` or a `list` of `str`s, e.g., "localhost:4242". If a `list`, gRPC requests containing the same `CallTraceback` proto payload will be sent to all the destinations. run_key: A string describing the feeds, fetches (and targets) names of the `tf.Session.run` call. origin_stack: The traceback of the `tf.Session.run()` invocation. graph: A Python `tf.Graph` object (i.e., *not* a `tf.compat.v1.GraphDef`), which contains op tracebacks. send_source: Whether the source files involved in the op tracebacks but outside the TensorFlow library are to be sent. F)r`rHrarbNrj)r_run_keyr)rarbrrrsend_graph_tracebackss rmcCst||d|ddS)aSend the tracebacks of an eager execution call to debug server(s). Args: destinations: gRPC destination addresses, a `str` or a `list` of `str`s, e.g., "localhost:4242". If a `list`, gRPC requests containing the same origin_stack: The traceback of the eager operation invocation. send_source: Whether the source files involved in the op tracebacks but outside the TensorFlow library are to be sent. T)r`rbNrk)r_r)rbrrrsend_eager_tracebackss  rn)FNNT)T)__doc__r rZtensorflow.core.debugrtensorflow.core.protobufrtensorflow.python.debug.librrrtensorflow.python.platformrtensorflow.python.profilerrrr#r5rArjrmrnrrrrs.         R