vj4%dZddlZddlZddlZddlZddlZddlZddlmZddl m Z ddl Z ddl m Z ddlmZmZmZmZddlmZmZmZ dd lmZddlZd Zd ad ad aej<dedZ GddejBZ"gdZ#dZ$dZ%dZ&dZ'dZ(d"dZ)d#dZ*d#dZ+d$dZ,d$dZ-d#dZ.d#dZ/dZ0d Z1d!Z2y#d ZYpxYw)%zCommunication with the viewerN)Path)connect)Timer)is_topods_shapeis_toploc_location serialize loc_to_tq) get_state update_stateget_config_file) get_ipythonTFzws://127.0.0.1concecVdd}|t_tj|ty)Nc&|jd|dS)Nz:  )__name__)messagecategoryfilenamelinenofilelines a/workspace/moltex-experiments/.pixi/envs/default/lib/python3.12/site-packages/ocp_vscode/comms.pywarning_on_one_linez&warn_once..warning_on_one_line,s%..88)NN)warnings formatwarningwarn UserWarning)rrs r warn_oncer"+s91H MM';'rc,eZdZdZdZdZdZdZdZdZ dZ y ) MessageTypez Message typesr N) r __module__ __qualname____doc__DATACOMMANDUPDATESLISTENBACKENDBACKEND_RESPONSECONFIGrrr$r$8s) DGG FG Frr$) send_data send_command send_responseset_portget_portlistener is_pytestcFtjjddk(S)NOCP_VSCODE_PYTEST1)osenvirongetr5rrr<r<Os ::>>- .# 55rctjtjtj}|jd|j t |fdk(}|r|j |S)z#Check whether the port is listeningr r)socketAF_INET SOCK_STREAM settimeout connect_exget_hostclose)portsresults r port_checkrNSsR fnnf&8&89ALLO \\8:t, - 2F   Mrct|r-tjt|j dSt |r t |St|tjr |jStdt|d)zDefault JSON serializer.utf-8zObject of type z is not JSON serializable) rbase64 b64encoderdecoderr isinstanceenumEnumvalue TypeErrortype)objs rdefaultr[]sms #/66w?? C ~ C #yy/$s)4MNOOrcXtrytstttS)z Get the portr)r< INIT_DONEfind_and_set_portset_connection_fileCMD_PORTr5rrr:r:is{  OrctddS)z Get the hostr(N)CMD_URLr5rrrIrIts 12;rc|ad|aday)z Set the portzws://TN)r`rbr])rKhosts rr9r9ysHdVnGIrc | tsttt} t |ddd5t j |t}|tjk(rd|z}n||tjk(rd|z}nc|tjk(rd|z}nJ|tjk(rd |z}n1|tjk(rd |z}n|tjk(rd |z}dddt |dd |jd d5 t!t"d|d5}|j%t |ddt'|dz dz ddd5d}|tjk(r+>>&tT2tF||7S*)-BGGI)>FN8 W0 /:$-* %b *FE ._listens y(,u = NN= )'nn.G "jj1G',,.0 !1;3C3CD{{9-9")&/&( $+MMODAq K/KNa4G (-.KN%4'.  k.A.AB%(!"I1> = =sADC%DBC%$D% D. C>9D>DDDr5)rrs` rr;r;s8 Nrcd}ttjjdd}|dkDrt d|dn5|}|t d|d nt d rd }t d |d t |y) z Set the port and connection filec zd}tj}tj}|D]\}}t t |s&t d|dtt |dd@|jdg}|D])}|jt|st |}||Dcgc]\}}t t |s|}}}t|dk(r|d}|St|dkDrtddj|dt d tScc}}w) Nz#Found stale configuration for port z, deleting it.rootsr rz Multiple ports found (z, z) and the file is outside of any workspace folder of this VS Code instance: The right viewer cannot be auto detected, use set_port(port) in your code.z#Could not find port in config file )rcwdr rrNintrr rBis_relative_torz RuntimeErrorjoinr ) rK current_pathstatespstaterroot_portss r find_portz$find_and_set_port..find_port s7xxz ""$HAuc!f%;A3nMNSVT40IIgr*E..tDz:q6D <)/IgdA:c$i3HTEI5zQQx Ua".tyy/?.@Aaa ;O  D M$'9: ; TNrctrttdrtj}|jdd}t |dd5}t j|}dddtdr3td ttd|td tytd yyy#1swYXxYw) z@Set the connection file for Jupyter in the state file .ocpvscodekernel IPKernelAppconnection_filerrP)encodingN iopub_portzJupyter kernel runningz(Jupyter connection file path written to zJupyter kernel not responding) JCONSOLEhasattrrrropenr|loadrNrr r`r )rcffconnection_infos rr_r_8sGKM84%% ]]= )*; < "cG ,"iilO- ol3 4 * + #4b 9 <_=Nrs# * & <;#H    fk*( $,,  6 P <~7 ; : CD-` 3sHs  B>>C