o ci.@s8ddlZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl m Z ddlmZddlZddlmZddlmZmZddlmZeeZejfddZGd d d eZGd d d e ZGd dde Z      dddZ!edddZ"ddZ#ddZ$ddZ%ddZ&dS)N)Pdb)Callable) ray_constants)_internal_kv_del_internal_kv_put) DeveloperAPIcCst||d|dS)N)file)printflush)messagestderrr A/home/ubuntu/.local/lib/python3.10/site-packages/ray/util/rpdb.py_crys  rc@sLeZdZddZeddZddZedfdd Z edfd d Z d S) _LF2CRLF_FileWrappercsp|_d|_j|_j|_j|_j|_j|_j|_t dr2fdd|_ dSj |_ dS)Nrwencodingcs|jjddS)Nreplace)errors)sendallencoder)data connectionfhr r,sz/_LF2CRLF_FileWrapper.__init__..) rmakefilestreamreadreadline readlinescloser filenohasattr_sendr)selfrr rr__init__"s  z_LF2CRLF_FileWrapper.__init__cCs|jjSN)rrr%r r rr2sz_LF2CRLF_FileWrapper.encodingcCs |jSr')r__iter__r(r r rr)6s z_LF2CRLF_FileWrapper.__iter__z ? cCs|d|}||dS)Nz )subr$)r%rnl_rexr r rwrite9s z_LF2CRLF_FileWrapper.writecCs|D]}|||qdSr')r,)r%linesr+liner r r writelines=sz_LF2CRLF_FileWrapper.writelinesN) __name__ __module__ __qualname__r&propertyrr)recompiler,r/r r r rr!s rc@s6eZdZdZdegdffddZddZeZZdS)_PdbWrapz/Wrap PDB to run a custom exit hook on continue. exit_hookNcCs||_t|dSr') _exit_hookrr&)r%r7r r rr&Esz_PdbWrap.__init__cC|t||Sr')r8r do_continuer%argr r rr:I z_PdbWrap.do_continue) r0r1r2__doc__rr&r:do_cdo_contr r r rr6Bs  r6c@sveZdZdZdZ  dddZddZdd Zd d ZeZ Z d d Z e Z Z dddZdddZddZddZdS) _RemotePdbac This will run pdb as a ephemeral telnet service. Once you connect no one else can connect. On construction this object will block execution till a client has connected. Based on https://github.com/tamentis/rpdb I think ... To use this:: RemotePdb(host="0.0.0.0", port=4444).set_trace() Then run: telnet 127.0.0.1 4444 NFcCsR||_||_||_ttjtj|_|jtjtj d|j ||f||_ dS)NT) _breakpoint_uuid_quiet_patch_stdstreamssocketAF_INET SOCK_STREAM_listen_socket setsockopt SOL_SOCKET SO_REUSEADDRbind _ip_address)r%breakpoint_uuidhostport ip_addresspatch_stdstreamsquietr r rr&]s  z_RemotePdb.__init__cCs|jstd|j|jdf|jd|j\}}|js(td|t||_t j |d|j|jdgdg|_ |j rWdD]}|j |tt|ftt||jqB|t_dS)Nz>RemotePdb session open at %s:%s, use 'ray debug' to connect...z#RemotePdb accepted connection from tabzray.*) completekeystdinstdoutskip)r rX __stderr__ __stdout__rW __stdin__)rCrrMrH getsocknamelistenacceptrhandlerr&backuprDappendgetattrsyssetattrrAactive_instance)r%raddressnamer r rr^ns0   z_RemotePdb.listencCsJ|jr |js td|j|jD] \}}tt||q|jdt_dS)NzRestoring streams: %s ...) rarCrrerdr`r!rArf)r%rhrr r r __restores   z_RemotePdb.__restorecCr9r')_RemotePdb__restorerdo_quitr;r r rrkr=z_RemotePdb.do_quitcCs ||jjt||Sr')rjr`rr!rr:r;r r rr:s  z_RemotePdb.do_continuec CsZ|dur tj}z t||WdSty,}z|jtjkr!WYd}~dSd}~wwr')rd _getframef_backr set_traceIOErrorerrno ECONNRESET)r%frameexcr r rrns  z_RemotePdb.set_tracec Cs^ztd}|t|d|WdSty.}z|jtjkr#WYd}~dSd}~ww)N)rdexc_inforesetr interactionrorprq)r% tracebacktrsr r r post_mortems  z_RemotePdb.post_mortemcCs\|jtjjj_tdt i}t d |j|t j d||jjt||S)z7remote Skip into the next remote call. job_idzRAY_PDB_CONTINUE_{} namespace)rBray_privateworker global_workerdebugger_breakpointjsondumpsget_runtime_context get_job_idrformatrKV_NAMESPACE_PDBrjr`rr!rr:)r%r<rr r r do_remotes    z_RemotePdb.do_remotecCs.|jtjjj_||jj t ||S)z?get Skip to where the current task returns to. ) rBr~rrrdebugger_get_breakpointrjr`rr!rr:r;r r rdo_gets    z_RemotePdb.do_get)FFr')r0r1r2r>rfr&r^rjrkdo_qdo_exitr:r?r@rnrzrrr r r rrAPs   !   rAFc Csd|r |rJdd}n |durtjdd}|dur#ttjdd}|dur0ttjdd }|s7tj}|r@tj j j j }nd }t ||||||d }|j}d ||d } ttd} tj| | j| jdtjtttt t!t"t#d } t$d|t%&| dt'j(d|)t*d|t'j(d|S)z5 Opens a remote PDB on first available port. z.Cannot specify both host and debugger_externalz0.0.0.0NREMOTE_PDB_HOSTz 127.0.0.1REMOTE_PDB_PORT0REMOTE_PDB_QUIET localhost)rNrOrPrQrRrSz{}:{}rTrt ) proctitle pdb_addressfilenamelinenorx timestampr{node_id worker_idactor_idtask_idz RAY_PDB_{}T) overwriter}r|)+osenvirongetintbooluuiduuid4hexr~rrrnode_ip_addressrArHr]rinspectgetouterframes currentframe_raylet getproctitlerrjoinrxformat_exceptionrdrutimerr get_node_id get_worker_id get_actor_id get_task_idrrrrrr^r) rOrPrRrSrNdebugger_externalrQrdbsocknamerparentframeinforr r r_connect_ray_pdbs`          rcCstjdddkrtjj|StjdddkrBtjjj j dkrDt j }tdddd|r1|ndtjjj jd}|j|ddSdSdS) zsInterrupt the flow of the program and drop into the Ray debugger. Can be used within a Ray task or actor. RAY_DEBUG1legacyNF)rOrPrRrSrNr)rr)rrrr~util ray_debugpyrnrrrrrdrlrmrdecoderay_debugger_external)rNrrrr r rrns   rncCsntjdddkrtjjStjdddkr5tddtjj _ dd}t |}t j}||dSdS) zThe breakpoint hook to use for the driver. This disables Ray driver logs temporarily so that the PDB console is not spammed: https://github.com/ray-project/ray/issues/18172 rrrz-*** Temporarily disabling Ray worker logs ***FcSstddtjj_dS)Nz#*** Re-enabling Ray worker logs ***T)r r~rr_worker_logs_enabledr r r renable_logging=sz)_driver_set_trace..enable_loggingN)rrrr~rrrnr rrrr6rdrlrm)rpdbrrr r r_driver_set_trace1s   rcCstjdddkS)NRAY_DEBUG_POST_MORTEMrr)rrrr r r r$_is_ray_debugger_post_mortem_enabledFsrcCsDtjdddkrtjjStddddtjj j j d}| dS)NrrF)rOrPrRrSr) rrrr~rr _post_mortemrrrrrrz)rr r rrJs   rc Cstjdkr ddl}ttjtj}|||f tjdkrAt|gggdd}|r3| tj |s@tj s@| tj n ttj |ggg\}}}|D]*}||krn| d}|s`dStj |tj qPtj } || qPq)Nwin32rTrTi)rdplatformmsvcrtrErFrGconnectselectkbhitrbrWisattyrecvrXr,rr rsendr) rOrPrs ready_to_read write_sockets error_socketssockrmsgr r r_connect_pdb_clientXs2         r)NNFNNFr')'rprrloggingrr4rrErdrrxrrrtypingrr~ ray._privaterray.experimental.internal_kvrrray.util.annotationsr getLoggerr0logrZrobjectrr6rArrnrrrrr r r rsJ     !  @