bgdZddlZddlZddlZddlZddlZddlZddlZddlmZddl m Z m Z ddl m ZddlmZddlmZmZmZddlmZmZejeZ Gd d eZGd d eeZGd deZdS)z ticker.py Websocket implementation for kite ticker :copyright: (c) 2021 by Zerodha Technology Pvt. Ltd. :license: see LICENSE for details. N)datetime)reactorssl)log)ReconnectingClientFactory)WebSocketClientProtocolWebSocketClientFactory connectWS) __version__ __title__cfeZdZdZdZdZdZdZdZdZ fdZ dZ dZ dZ d Zd Z d Zd ZxZS) KiteTickerClientProtocolz(Kite ticker autobahn WebSocket protocol.g@NcHtt|j|i|dS)z9Initialize protocol with all options passed from factory.N)superr__init__selfargskwargs __class__s ;e:\HARSH\Automation\J_Work\zerodha_ticker\zerodha_ticker.pyrz!KiteTickerClientProtocol.__init__(s,6&--6GGGGGGc||j_|jjr|j|||jdS)z7Called when WebSocket server connection was establishedN)factoryws on_connect resetDelayrresponses r onConnectz"KiteTickerClientProtocol.onConnect-sL  < " 4 L # #D( 3 3 3 !!!!!rc|||jjr|j|dSdS)zBCalled when the initial WebSocket opening handshake was completed.N) _loop_ping_loop_pong_checkron_openrs ronOpenzKiteTickerClientProtocol.onOpen8sY   <  ' L  & & & & & ' 'rcZ|jjr|j|||dSdS)z/Called when text or binary message is received.N)r on_message)rpayload is_binarys r onMessagez"KiteTickerClientProtocol.onMessageCs9 < " > L # #D'9 = = = = = > >rcJ|s(|jjr|j||||jjr|j|||d|_d|_|jr|j|jr|jdSdS)z!Called when connection is closed.N)ron_erroron_close_last_ping_time_last_pong_time _next_pingcancel_next_pong_check)r was_cleancodereasons ronClosez KiteTickerClientProtocol.onCloseIs :|$ : %%dD&999 <  6 L ! !$f 5 5 5 $# ? % O " " $ $ $   +  ! ( ( * * * * * + +rc^|jrR|jjrFtdt j|jz t j|_|jjr/td|dSdS)z%Called when pong message is received.zlast pong was {} seconds back.z pong => {}N)r1rdebugrformattimers ronPongzKiteTickerClientProtocol.onPong\s   cDL$6 c II6==dikkDL`>`aa b b b#y{{ <  5 IIl))(33 4 4 4 4 4 5 5rcF|jjrM|jrFtdt j|jz t j|_|jj|j|j |_ dS)z>Start a ping loop where it sends ping message every X seconds.zlast ping was {} seconds back.N) rr:r0rr;r<r callLater PING_INTERVALr#r2r&s rr#z#KiteTickerClientProtocol._loop_pingjs <  g# g :AA$)++PTPdBdeefff $y{{,.889KT_]]rcb|jrxtj|jz }|d|jzkrO|jjr-t d||d|jj |j|j |_ dS)z Timer sortof to check if connection is still there. Checks last pong message time and disconnects the existing connection to make sure it doesn't become a ghost connection. zBLast pong was {} seconds ago. So dropping connection to reconnect.T)abortN) r1r<r@rr:rr;dropConnectionrr?r$r4)rlast_pong_diffs rr$z)KiteTickerClientProtocol._loop_pong_checkvs   0!Y[[4+??NT%7!788<%)IIbii&(()))##$#///!% 4 > >t?QSWSh i ir)__name__ __module__ __qualname____doc__r@KEEPALIVE_INTERVALr2r4r1r0rr!r'r,r8r=r#r$ __classcell__rs@rrrs22MJOOHHHHH """'''>>> +++&555 ^ ^ ^jjjjjjjrrcJeZdZdZeZdZdZdZfdZ dZ dZ dZ d Z xZS) KiteTickerClientFactoryzQAutobahn WebSocket client factory to implement reconnection and custom callbacks.r Ncd|_d|_d|_d|_d|_d|_d|_d|_d|_tt|j |i|dS)z/Initialize with default callback method values.FN) r:rr%r.r/r)r on_reconnecton_noreconnectrrNrrs rrz KiteTickerClientFactory.__init__sm     "5%t,,5tFvFFFFFrc|js!|jrtdtj|_dS)z$On connecting start or reconnection.zStart WebSocket connection.N)_last_connection_timer:rr<)r connectors rstartedConnectingz)KiteTickerClientFactory.startedConnectings=) 5dj 5 II3 4 4 4%)Y[["""rc T|jdkrstd|jt t |j|jr||j||| dS)z2On connection failure (When connect request fails)rzNRetrying connection. Retry attempt count: {}. Next retry in around: {} secondsN) retriesrerrorr;introunddelayrQretrysend_noreconnectrrUr7s rclientConnectionFailedz.KiteTickerClientFactory.clientConnectionFaileds On connection lost (When ongoing connection got disconnected).rN)rXrQr]r^r_s rclientConnectionLostz,KiteTickerClientFactory.clientConnectionLosts\  > >  <  9)A 9!688O $,H\]]]] : 1  $SZ 0 0 0 $ $05,-(1(8 TX(Y(Y(Y%/3%,%++----- ##d##### $ $rcL|jr|jj|jjkrdSdS)z-Check if WebSocket connection is established.TF)rstate STATE_OPENr&s r is_connectedzKiteTicker.is_connecteds) 7 tw}(:::45rcN|jr|j||dSdSzClose the WebSocket connection.N)r sendCloserr6r7s r_closezKiteTicker._close s4 7 , G  dF + + + + + , ,rcZ||||dSr) stop_retryrrs rclosezKiteTicker.close%s,  D&!!!!!rc,tjdS)zStop the event loop. Should be used if main thread has to be closed in `on_close` method. Reconnection mechanism cannot happen past this method N)rrer&s rrezKiteTicker.stop*s  rcJ|jr|jdSdS)z'Stop auto retry when it is in progress.N)r stopTryingr&s rrzKiteTicker.stop_retry0s0 < & L # # % % % % % & &rc N |jtjt j|j|d|D]}|j|j|<dS#t$r<}| d t|d}~wwxYw)z Subscribe to a list of instrument_tokens. - `instrument_tokens` is list of instrument instrument_tokens to subscribe avTzError while subscribe: {}r7N) r sendMessagesixbjsondumps_message_subscribe MODE_QUOTEr Exceptionrr;strrinstrument_tokenstokenes rrzKiteTicker.subscribe5s  G  djt'>EV!W!WXXYY   + @ @04&u--4    KK:AA#a&&IIK J J J  sAA B$(7BB$c b |jtjt j|j|d|D]} |j|= #t$rYwxYwdS#t$r<}| d t|d}~wwxYw)z Unsubscribe the given list of instrument_tokens. - `instrument_tokens` is list of instrument_tokens to unsubscribe. rTzError while unsubscribe: {}rN) rrrrrr_message_unsubscriberKeyErrorrrr;rrs rrzKiteTicker.unsubscribeHs   G  djt'@GX!Y!YZZ[[   +  /66D4    KK<CCCFFKKK L L L  s<A A( AA( A# A("A##A(( B.27B))B.c H |jtjt j|j||gd|D] }||j|< dS#t$r<}| d t|d}~wwxYw)a' Set streaming mode for the given list of tokens. - `mode` is the mode to set. It can be one of the following class constants: MODE_LTP, MODE_QUOTE, or MODE_FULL. - `instrument_tokens` is list of instrument tokens on which the mode should be applied rTzError while setting mode: {}rN) rrrrrr_message_setmoderrrr;r)rrrrrs rset_modezKiteTicker.set_mode^s  G  djt'|r:: : ; ;rc>|jr||SdSr)rRr&s rrzKiteTicker._on_noreconnects+   -&&t,, , - -rctjs-t|tkr|d} t j|}n#t$rYdSwxYw|jrJ| ddkr1| dr|||d| ddkr,| |d| ddSdS)zParse text message.zutf-8NtypeorderdatarYr) rPY2rbytesdecoderloads ValueErrorrrr)rr*rs rrzKiteTicker._parse_text_messagesw .4==E11nnW--G :g&&DD    FF    5DHHV$4$4$?$?DHHVDTDT$?  tF| 4 4 4 88F  w & & NN4DHHV$4$4 5 5 5 5 5 ' &sA AAc ||}g}|D]V}||dd}|dz}||jdkrd}n||jdkrd}nd}||jd krd nd }t|d kr8|||j|||dd |z d t|dkst|dkrUt|dkr|jn|j} || |||dd |z ||d d|z ||dd|z ||dd|z ||dd|z dd} d| d<| dddkr,| d| ddz dz| ddz | d<t|dkrB tj ||dd} n#t$rd} YnwxYw| | d<|| %t|dkst|dkr t|dkr|jn|j} || |||dd |z ||d d||dd|z ||dd||dd||dd||dd|z ||dd|z ||dd |z ||d d|z d!d" } d| d<| dddkr,| d| ddz dz| ddz | d<t|dkr tj ||dd#} n#t$rd} YnwxYw tj ||d$d%} n#t$rd} YnwxYw| | d&<||d#d'| d(<||d'd)| d*<||d)d$| d+<| | d<ggd,} ttd%t|dD]\}}| |d-krd.nd/||||dz|||dz|d z|z |||d z|d0zd12d3| | d4<|| X|S)5z1Parse binary data to a (list of) ticks structure.rrlrsgcArvg@gY@ryFTro)tradablerinstrument_token last_price )highlowopenr)rrrrohlcchangerrrdNexchange_timestamp,$()rr r r) rrrrlast_traded_quantityaverage_traded_price volume_tradedtotal_buy_quantitytotal_sell_quantityr0r|@last_trade_time4oi8 oi_day_high oi_day_low)buysellrr%r$rOH byte_format)quantitypriceordersdepth) _split_packets _unpack_int EXCHANGE_MAPrrMODE_LTPr MODE_FULLr fromtimestampr enumeraterange)rbinpacketsrpacketrsegmentdivisorrrd timestamprr,ips rrzKiteTicker._parse_binarysm%%c**t t F#//1== &-G$+E222$D-e444!!(4+>AdG'+'7'7B'G'GAm$&*&6&6vr2&F&FAlO.7A*+ " "E !*%CKK*D*D E E1QffE:AA(,(8(8AE(J(J%)%5%5fa!eQU%K%Kg%U&*&6&6vq1ua"fZ]&6&^&^CC "'AgJ A s6')H H H )O?? P P)P<< Q  Q IcNtjd|z|||dS)z(Unpack binary data as unsgined interger.>r)structunpack)rr5rendr(s rr.zKiteTicker._unpack_intJs&}S;.E#I??BBrc"t|dkrgS||ddd}g}d}t|D]L}||||dzd}|||dz|dz|z|dz|z}M|S)z.Split the data to individual packets of ticks.rBrr&r')rr.r4r)rr5number_of_packetsr6jr< packet_lengths rr-zKiteTicker._split_packetsNs s88a<<I ,,S!QC,HH ()) & &A ,,S!QU,LLM NN3q1ua!em&;;< = = =A %AAr)FFN)NN)r>)+rFrGrHrIr/CONNECT_TIMEOUTRECONNECT_MAX_DELAYRECONNECT_MAX_TRIESrr1rr0r _message_coderrrrrrrrrrrrrerrrrrrrrrrrrrrr.r-rrrririsHXXv  L O&HIJHM$($% #& "5H^q!0?$?$?$?$B;;;*CCC*$*$*$*$X,,,, """"  &&& &,,---&,,, ...... . . . & & &;;;---666&{{{zCCCCrri)rIrrr<rrAloggingrrtwisted.internetrrtwisted.pythonrrtwisted.internet.protocolrautobahn.twisted.websocketrr r r r getLoggerrFrrNobjectrirrrrSs ))))))))------??????&&&&&&&&&&/.......g!!jjjjjjjjjj6jjjjjjZ?&?&?&?&?&46O?&?&?&DQ Q Q Q Q Q Q Q Q Q r