!ddlmZddlZddlZddlZddlZddlmZmZm Z ddl Z ddl Z ddl m Z mZmZmZdZdZdZdZdZGd d Zej.d d gZd ZdZy)) ApiExceptionN) WebSocketABNF enableTrace) urlencode quote_plusurlparse urlunparseceZdZdZddZddZddZdZddZddZ dd Z dd Z dd Z dd Z d ZdZdZddZddZdZy)WSClientctdg}d|_i|_d|_|rd|vr|j d|dz|rd|vr|j d|dzn|j d|j dr[|j rOtj|jxstjd }|j"|j|d <nd tji}|jr|j|d <|jr|j|d<t!|d|_|j"j%||d|_y )a2A websocket client with support for channels. Exec command uses different channels for different streams. for example, 0 is stdin, 1 is stdout and 2 is stderr. Some other API calls like port forwarding can forward different pods' streams to different channels. F authorizationzauthorization: %szsec-websocket-protocolzsec-websocket-protocol: %sz)sec-websocket-protocol: v4.channel.k8s.iozwss://) cert_reqsca_certsNcheck_hostnamercertfilekeyfile)ssloptskip_utf8_validation)headerT)r _connected _channels_allappend startswith verify_sslssl CERT_REQUIRED ssl_ca_certcertifiwhereassert_hostname CERT_NONE cert_filekey_filersockconnect)self configurationurlheadersrssl_optss .lib/third_party/kubernetes/stream/ws_client.py__init__zWSClient.__init__%sC FDODNDI?g- mm''/*BBC+w6 mm04567 mm?@ ~~hM$<$<((#//B7==?h  & & 2%2%B%B!"s}}-h*44hz)22hyFDIIIc&)DOcb|j|||jvr|j|Sy)zRPeek a channel and return part of the input, empty string otherwise. timeoutr)updater)r-channelr7s r2 peek_channelzWSClient.peek_channelQs0  KKK $.. ^^G $$ r4c||jvr|j||}n|j|}||jvr |j|=|S)zRead data from a channel.)rr:)r-r9r7rets r2 read_channelzWSClient.read_channel[sKdnn$   gw /c NN7 #c$.. .. ! Jr4Nc | td}tj}|jrtj|z |kr||jvrS|j|}d|vr@|j d}|d|}||dzd}|r||j|<|S|j|=|S|j |tjz |z|jrtj|z |kryyyy)zRead a line from a channel.Ninf r r6)floattimeis_openrfindr8)r-r9r7startdataindexr<s r2readline_channelzWSClient.readline_channelese g IIKE ,,.TYY[507: DNN "~~g& 4<))D/%Ve #eaij!$ &*DNN7 #*w'* kk7TYY[058k: ,,.TYY[507:.:.r4cR|jjt||zy)zWrite data to a channel.N)r+sendchr)r-r9rFs r2 write_channelzWSClient.write_channelxsIINN3w<$&'r4c0|jt|S)z$Same as peek_channel with channel=1.r6)r:STDOUT_CHANNELr-r7s r2 peek_stdoutzWSClient.peek_stdout|   ^W  ==r4c0|jt|S)z$Same as read_channel with channel=1.r6)r=rNrOs r2 read_stdoutzWSClient.read_stdoutrQr4c0|jt|S)z(Same as readline_channel with channel=1.r6)rHrNrOs r2readline_stdoutzWSClient.readline_stdout   AAr4c0|jt|S)z$Same as peek_channel with channel=2.r6)r:STDERR_CHANNELrOs r2 peek_stderrzWSClient.peek_stderrrQr4c0|jt|S)z$Same as read_channel with channel=2.r6)r=rXrOs r2 read_stderrzWSClient.read_stderrrQr4c0|jt|S)z(Same as readline_channel with channel=2.r6)rHrXrOs r2readline_stderrzWSClient.readline_stderrrVr4c:|j}d|_i|_|S)aReturn buffered data received on stdout and stderr channels. This is useful for non-interactive call where a set of command passed to the API call and their result is needed after the call is concluded. Should be called after run_forever() or update() TODO: Maybe we can process this and return a more meaningful map with channels mapped for each input. r)rr)r-outs r2read_allzWSClient.read_alls  ))CDIDN Jr4c|jS)z&True if the connection is still alive.)r)r-s r2rCzWSClient.is_opens ??r4c0|jt|y)z)The same as write_channel with channel=0.N)rL STDIN_CHANNEL)r-rFs r2 write_stdinzWSClient.write_stdins}d+r4c|jsy|jjsd|_yt j|jjfdd|\}}}|r |jj d\}}|t jk(rd|_y|t jk(s|t jk(r|j}tjr|jdd}t|dkDrot|d}|dd}|rY|t t"fvr|xj$|z c_||j&vr||j&|<y|j&|xx|z cc<yyyyy) z@Update channel buffers with at most one complete frame of input.NFTzutf-8replacer r)rCr+ connectedrselectrecv_data_framer OPCODE_CLOSE OPCODE_BINARY OPCODE_TEXTrFsixPY3decodelenordrNrXrr)r-r7r_op_codeframerFr9s r2r8zWSClient.updates> <<> 99  do mmTYY^^-r2w?GAq!yy006ngu D%% % d(( (Gt7G7G,Gzz 77Wi0$ t9q=QL'ab$ >>::ii4idnn,(,dnnW%nnW%-%  -H  r4c|rtj}|jrrtj|z |krW|j|tjz |z|jrtj|z |krUyyyy|jr$|jd|jr#yy)zpWait till connection is closed or timeout reached. Buffer any input received during this time. r6N)rBrCr8)r-r7rEs r2 run_foreverzWSClient.run_forevers iike LLNtyy{U2W< Wtyy{2U: < LLNtyy{U2W>B>>B ,.< " r4r WSResponserFct|}t|}|jdk(rd|d<t|S|jdk(rd|d<t|S)Nhttpwsrhttpswss)r listschemer )r/ parsed_urlpartss r2get_websocket_urlrs]}* z %& E!H E G#E!H E r4cn|d}|jdd}|jdd}|jd}g}|jdiD]H\}} |dk(r+t| tr| D]} |j|| f6|j|| fJ|r|d t |zz } t |t ||} |s| S| j| td d j| jzS#tttf$r} td t| d} ~ wwxYw)zAn internal function to be called in api-client when a websocket connection is required. args and kwargs are the parameters of apiClient.request method. r _request_timeout<_preload_contentTr0 query_paramscommand?r6z%srr)statusreasonN)get isinstancerrrrrrxrjoinr` ExceptionKeyboardInterrupt SystemExitrstr) r.argsr{r/rrr0rkeyvaluerclientes r2websocket_callrs5 Q#ZZ 2B7ZZ 2D9 JJy !',JJ~r2jc5 iJud3'S'N+3,' 33<( ((C0 m%6s%;W EF  m /0 dRWWV__%677 88 & 30 aA //0s.Drsp"0  22 NN  z z z$[ # #L6( ; 0r4