o i<+@sddlZddlZddlZddlmZddlmZmZmZm Z m Z ej j dkr3ej j dkr3ddlmZnddlmZddlmZmZmZdd lmZdd lmZd d lmZmZmZmZd d lmZmZmZmZe Z!Gddde ddZ"GdddeeZ#GdddeeZ$dS)N) getLogger)CallableListOptional TypedDictUnion )timeout)ConnectionErrorInvalidResponse RedisError) EncodableT)HIREDIS_AVAILABLE)AsyncBaseParserAsyncPushNotificationsParser BaseParserPushNotificationsParser)#NONBLOCKING_EXCEPTION_ERROR_NUMBERSNONBLOCKING_EXCEPTIONSSENTINELSERVER_CLOSED_CONNECTION_ERRORc@sJeZdZUeegefed<eegefed<eeed<eeed<dS)_HiredisReaderArgs protocolError replyErrorencodingerrorsN)__name__ __module__ __qualname__rstr Exception__annotations__rr%r%J/home/ubuntu/.local/lib/python3.10/site-packages/redis/_parsers/hiredis.pyr"s  rF)totalc@sXeZdZdZddZddZddZdd Zd d Zd d Z e dfddZ dddZ dS)_HiredisParserz*Parser class for connections using HirediscCsFtstd||_t||_|j|_d|_d|_d|_ d|_ d|_ dS)NzHiredis is not installed) rrsocket_read_size bytearray_bufferhandle_pubsub_push_responsepubsub_push_handler_funcnode_moving_push_handler_funcmaintenance_push_handler_func#oss_cluster_maint_push_handler_funcinvalidation_push_handler_func_hiredis_PushNotificationTypeselfr)r%r%r&__init__,s  z_HiredisParser.__init__cCs$z|WdStyYdSwN) on_disconnectr#r4r%r%r&__del__8s  z_HiredisParser.__del__cCstd}|dt||SN push_responsezPush response: rdebugr"r4responseloggerr%r%r&r,>sz*_HiredisParser.handle_pubsub_push_responsecKsddl}|j|_|j|_t|j|jjtd}|jj r |jj |d<|j di||_ t|_ z|j|_WdSty@d|_YdSw)Nr)rrr notEnoughDatarr%)hiredis_socksocket_timeout_socket_timeoutr parse_errorencoderencoding_errorsNOT_ENOUGH_DATAdecode_responsesrReader_reader_next_responsePushNotificationr2AttributeError)r4 connectionkwargsrBr%r%r& on_connectCs"   z_HiredisParser.on_connectcCsd|_d|_t|_dSr6)rCrLrIrMr8r%r%r&r7Zs z_HiredisParser.on_disconnectcCs@|jstt|jtur|j|_|jtur|j|ddSdS)NF)r raise_on_timeoutT)rLr rrMrIgetsread_from_socket)r4r r%r%r&can_read_s   z_HiredisParser.can_readTc Cs|j}|tu}z}z-|r|||j|j}|dkrtt|j|jd|WW|r4||j dSdSt j yQ|rCt dYW|rO||j dSdSt y}z(t|jd}|sx|j|krxWYd}~W|rv||j dSdStd|jd}~ww|r||j ww)NrTzTimeout reading from socketFz!Error while reading from socket: )rCr settimeout recv_intor+r rrLfeedrEsocketr TimeoutErrorrrget __class__errnoargs)r4r rSsockcustom_timeoutbufflenexallowedr%r%r&rUisB    z_HiredisParser.read_from_socketFcCs|jstt|jtur/|j}t|_|jdur-t||jr-||}|r&|S|j||dS|S|r8|j d}n|j }|turW| |rN|j d}n|j }|tusAt|tr^||jduryt||jry||}|rr|S|j||dSt|t r|rt|dtr|d|S)Ndisable_decoding push_requestFr) rLr rrMrIr2 isinstancehandle_push_response read_responserTrUlistr4rgrhr?r%r%r&rksZ         z_HiredisParser.read_responseNFF) rr r!__doc__r5r9r,rRr7rVrrUrkr%r%r%r&r()s  r(c sxeZdZdZdZdeffdd ZddZdd Zd d Z d d Z ddZ dde de de eeeffddZZS)_AsyncHiredisParserzBAsync implementation of parser class for connections using Hiredis)rLr)cs8tstdtj|dd|_|j|_d|_d|_dS)NzHiredis is not available.)r)) rrsuperr5rLr,r-r1r2r3r^r%r&r5s z_AsyncHiredisParser.__init__cs td}|dt||Sr:r<r>r%r%r&r,sz/_AsyncHiredisParser.handle_pubsub_push_responsecCsddl}|j|_t|jtd}|jjr|jj|d<|jj |d<|j di||_d|_ z t |dd|_ WdStyBd|_ YdSw)Nr)rrrArrTrNr%)rBrL_streamr rFrIrGrJrrHrK _connectedgetattrr2rO)r4rPrBrQr%r%r&rRs$    z_AsyncHiredisParser.on_connectcCs d|_dS)NF)rtr8r%r%r&r7s z!_AsyncHiredisParser.on_disconnectc s|jstd|jturdSz)td4IdH|IdHWdIdHWS1IdHs3wYWdStjyEYdSw)NzBuffer is closed.TrF) rtOSErrorrLrTrI async_timeoutrUasyncior\r8r%r%r&can_read_destructives 4z(_AsyncHiredisParser.can_read_destructivecs>|j|jIdH}|rt|tsttd|j|dS)NT) rsread _read_sizeribytesr rrLrZ)r4bufferr%r%r&rUs   z$_AsyncHiredisParser.read_from_socketFrgrhreturncs|js ttd|r|jd}n|j}|tur4|IdH|r+|jd}n|j}|tust|tr;||jdur\t||jr\| |IdH}|sZ|j ||dIdHS|St|t rn|rnt|dtrn|d|S)NFrfr) rtr rrLrTrIrUrir2rjrkrlrmr%r%r&rks>       z!_AsyncHiredisParser.read_responsern)rr r!ro __slots__intr5r,rRr7ryrUboolrrrrk __classcell__r%r%rrr&rps"   rp)%rxr[sysloggingrtypingrrrrr version_infomajorminorr rw exceptionsr r rrutilsrbaserrrrrrrrobjectrIrr(rpr%r%r%r&s$