2i\HddlZddlZddlZddlZddlZddlmZmZmZm Z m Z m Z m Z m Z mZmZmZddlZddlmZmZmZmZmZddlmZddlmZddlmZmZm Z m!Z!m"Z"m#Z#ddl$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,dd l-m.Z.ej^d Z0d ed efd Z1dee2d eeeeffdZ3de4de4d e4fdZ5da6d e4fdZ7GddZ8y)N) ListAnyTupleOptionalCallableUnionMatchAnyStr GeneratorSequenceType)XStreamZSetHashExpiringMembersSet ClientInfo)_msgs) extract_args)IntFloatSUPPORTED_COMMANDSCOMMANDS_WITH_SUB Signature CommandItem) SimpleErrorvalid_response_type SimpleString NoResponse casematchcompile_patternQUEUEDdecode_command_bytes)ResponseErrorType fakeredisvalreturnct|tr|jSt|trt j|dSt|t rBt tj|j}|Dcgc] }t|c}St|t tfr|Dcgc] }t|}}|S|Scc}wcc}w)NF) humanfriendly) isinstancestrencodefloatrdictlist itertoolschainitems_convert_to_resp2tuple)r&resultitemress c/mnt/e/genesis-system/.venvs/voice-bridge/lib/python3.12/site-packages/fakeredis/_basefakesocket.pyr3r3s#szz|#u||Cu55#tioosyy{344:;D!$';;#e}%3674 &77 J <7s C 4Cfieldsct|d}|tvr+t|dk\r|dt|dzz }|dd}||fS|dd}||fS)aExtracts the command and command arguments from a list of `bytes` fields. :param fields: A list of `bytes` fields containing the command and command arguments. :return: A tuple of the command and command arguments. Example: ``` fields = [b'GET', b'key1'] result = _extract_command(fields) print(result) # ('GET', ['key1']) ``` r rN)r#rlen)r9cmd cmd_argumentss r8_extract_commandr@+so vay )C CK1$4 s)&)444qr   qr  x bits_countcXd}t|D]}||z dzs |d|dz |z zz}|S)Nrr)range)rBrCr5is r8 bin_reverserGAsE F : 0 Fa< aJNQ./ /F0 MrAc$t}tdz a|S)Nr) _next_file_no)r5s r8_get_next_file_norKLs FQM MrAc eZdZUegdfed<hdZejZddde de de d e d df fd Z e d efd Ze d ee d ffdZe d efdZde d dfdZd7dZd7dZde d dfdZd e fdZde d dfdZd7dZeded eeeffdZd ede dffdZdeed dfdZ de!ee ge fde"dee d e#d e f d!Z$d"e%d e&fd#Z'd$e d e fd%Z(d&e!e)e*e fdee#ge fd e fd'Z+d(ed ee!ee ge fe"ffd)Z,d*e-d dfd+Z.d,e/ed-e ded ee)eeeffd.Z0d/e1d0e*d e fd1Z2d2e*d3e#d efd4Z3d2e d efd5Z4ed/e1d e5fd6Z6xZ7S)8BaseFakeSocketN_clear_watches>ping subscribe psubscribe ssubscribe unsubscribe punsubscribe sunsubscribeserver FakeServerdb client_classargskwargsr'ctt| |i|ddlm}||_t |_||_|j|j|_ ||_ tj|_d|_|j!|_|j"j%d|||||j'di}t)di||_|j j,j/|y)Nr)rWF client_info)superrM__init__r%rW_serverrK_fileno_db_numdbs_db _client_classqueueQueue responses_paused_parse_commands_parsersendpopr _client_infosocketsappend) selfrVrXrYrZr[rWinfo __class__s r8r`zBaseFakeSocket.__init__`s nd,d=f=(#) (*  ::dll+)7<{{} ++-  $    zz-,&.. ##D)rAc.|jjSN)rouserrrs r8 current_userzBaseFakeSocket.current_users  %%%rA.c.|jjSrv)raversionrxs r8r{zBaseFakeSocket.versions||###rAc.|jjSrv)ra server_typerxs r8r}zBaseFakeSocket.server_types||'''rAmsgcD|j}|r|j|yy)zgPut a response message into the queue of responses. :param msg: The response message. N)riput)rrr~ris r8 put_responsezBaseFakeSocket.put_responses!NN  MM#  rAcd|_y)NT)rjrxs r8pausezBaseFakeSocket.pauses  rAcHd|_|jjdy)NFrA)rjrlrmrxs r8resumezBaseFakeSocket.resumes  #rA_c8|jjyrv)rlclose)rrrs r8shutdownzBaseFakeSocket.shutdowns rAc|jSrv)rbrxs r8filenozBaseFakeSocket.filenos ||rAc|jjD]}|j||jjD]}|j||j y)zRemove all the references to `self` from `server`. This is called with the server lock held, but it may be some time after self.close. N) subscribersvaluesdiscard psubscribersrN)rrrVsubss r8_cleanupzBaseFakeSocket._cleanupsd &&--/ D LL  ''..0 D LL   rAc|jjj||jjj t j |d|_d|_d|_yrv) rarpremoveclosed_socketsrqweakrefrefrerirxs r8rzBaseFakeSocket.closesR ##D) ##**7;;t+<= rAbufc6|jddz}|dkrDttjj |j j |d|}||d}|jds|j j jdd}dj|j j jddd}ttjj |d|dz||fS)N rrs r<'z' ) findrmsgsUNKNOWN_COMMAND_MSGformatdecodestripendswithsplitjoin)rposlinecommandrZs r8 _extract_linezBaseFakeSocket._extract_lineshhuo! !8d66==cjjl>P>P>RST T4Cy#$i}}W%kkm))+11#6q9G88DKKM//177&>&E&EcjjlFXFXFZ&[\\T!BZ#h!+5LC#h!+ c'6l+&1*,' (  ! !& )'s'FFB F=BF1Fr9cV|sytjd|jjdd|t |\}} |j |\}}|j jj|jj|jj||j j5 |j jj}|}|r|j|j  K#t$rYnwxYwt!j }|j j"j%D] } || _ |j'||j(|j*At,j.|j0vr%|j*j3|||ft4} n|j7|||d} dddn #1swYnxYwn#t8$rt} |j*d|_|dk(rP| j<j?dr5d| j<d dz| _d|_d|_|jA| } Yd} ~ nd} ~ wwxYw|jC } tE| tFs|jI| yy) Nz>>> raddrz: TFexeczERR z,EXECABORT Transaction discarded because of: )%LOGGERdebugrogetr@ _name_to_funcraaclvalidate_commandrwas_byteslockrrnr IndexErrortimerdr check_arityr{ _transactionrFLAG_TRANSACTIONflagsrqr" _run_commandr_transaction_failedvalue startswithrN_decode_resultr*rr) rrr9r>r?funcsig weak_socksocknowrXr5excs r8rzBaseFakeSocket._process_commandsS  tD--11':;2fXFG-f5]" **3/ID# LL   - -d.?.?.D.DdFWFWF`F`Fbdj k"" P8$(LL$?$?$C$C$E  ){ MM$,,7& iik,,**113"B!BG" t||<$$0T5J5JRUR[R[5[%%,,dC-GH#F!..tS-OF' P P P(   ,,0(f}!5!5f!=JSYYWXWY]Z $(!+0(##%F $$V,&*-   f %.sWA=G7G*$D+&G* DG*DCG*! G7*G3/G77 I4A*I//I4rr from_scriptc0g} |j||j|j}|r5tj|j vrt tj|jr5|jtjvrt tjt|dk(r|d}n|\}}||}|jjdk(r'tj |j vr t#|}tj |j vr.t%||jjst'd||D].} | j)tj*|j v0|S#t $r }|}Yd}~Ed}~wwxYw)Nrrr;zInvalid response type for )remove_empty_val)applyrer{rFLAG_NO_SCRIPTrrCOMMAND_IN_SCRIPT_MSG_pubsubnamerMACCEPTED_COMMANDS_WHILE_PUBSUBBAD_COMMAND_IN_PUBSUB_MSGr=roprotocol_versionFLAG_SKIP_CONVERT_TO_RESP2r3rAssertionError writebackFLAG_LEAVE_EMPTY_VAL) rrrrrZr command_itemsretr5r command_items r8rzBaseFakeSocket._run_commandsb,.  ))D$((DLL9Ct22cii?!$"<"<==||0]0] ]!$"@"@AA3x1}Q&)#mt$$55:t?^?^fifofo?o.v6F22#))CL_D-->>M)+EfX)NOO* `L  " "D4M4MUXU^U^4^ " _ `  F sEF F FFerrorc|jjjdr(ddlm}|dj |j Sddlm}|dj |j S)Nvalkeyr) DefaultParseri)socket_read_size)rf __module__rvalkey.connectionr parse_errorrredis.connection)rrrValkeyDefaultParserRedisDefaultParsers r8 _decode_errorzBaseFakeSocket._decode_error-sV    ( ( 3 3H = N&>JJ5;;W W L%u=II%++V VrAr5ct|tr|Dcgc]}|j|c}St|tr |jSt|t r|j |S|Scc}w)z1Convert SimpleString and SimpleError, recursively)r*r/rrrrr)rrr5rs r8rzBaseFakeSocket._decode_result7sa fd #4:;qD''*; ;  -<<   ,%%f- -M n||||z} | t|k\rd } t!t | |j#| gS) aThis is the basis of most of the ``scan`` methods. This implementation is KNOWN to be un-performant, as it requires grabbing the full set of keys over which we are investigating subsets. The SCAN command, and the other commands in the SCAN family, are able to provide to the user a set of guarantees associated with full iterations. - A full iteration always retrieves all the elements that were present in the collection from the start to the end of a full iteration. This means that if a given element is inside the collection when an iteration is started and is still there when an iteration terminates, then at some point the SCAN command returned it to the user. - A full iteration never returns any element that was NOT present in the collection from the start to the end of a full iteration. So if an element was removed before the start of an iteration and is never added back to the collection for all the time an iteration lasts, the SCAN command ensures that this element will never be returned. However, because the SCAN command has very little state associated (just the cursor), it has the following drawbacks: - A given element may be returned multiple times. It is up to the application to handle the case of duplicated elements, for example, only using the returned elements to perform operations that are safe when re-applied multiple times. - Elements that were not constantly present in the collection during a full iteration may be returned or not: it is undefined. )z*matchz*typez+countN r0keyr'cpt|tr|jd}j|SdS)Nzutf-8T)r*r+r,match)r regexs r8 match_keyz'BaseFakeSocket._scan..match_keys4#s#jj)','85;;s# Bd BrAc|duxs6ttjj|jSrv)r rM_key_value_typerer)r _typerrs r8 match_typez(BaseFakeSocket._scan..match_types4D=iIn.L.LTXXVY].[.a.ach$i irAr)rrsortedr= bit_lengthrGr!bytesrboolr r0islicer*r4rqr+r,)rrrrrZpatterncountrrbits_len result_cursor result_datarrr& compare_valrr s` @@r8_scanzBaseFakeSocket._scanpsq:V%1$8U%V"%md|IM--/VX. SY ":   ,3,?(T C5 CU4ut+C%D C  jE jd j  %"3 ''ffunE ,(23(>c!fC [)j.E&&s+ , v7K CI %MK x89@@BKPPrAr scalec|sy|jytt|j|jjz |zS)Nr)expireatrroundrer)rrr rs r8_ttlzBaseFakeSocket._ttls; \\ !ucllTXX]]:eCDE ErArr)cV|jdk\rd|z}tj||SN)r)r{rr,)rrrr)s r8 _encodefloatzBaseFakeSocket._encodefloats( <<4 IE||E=11rAcT|jdk\rd|z}tj|Sr&)r{rr,)rrrs r8 _encodeintzBaseFakeSocket._encodeints& <<4 IEzz%  rAc|j tdSt|jtr tdSt|jtr tdSt|jt r tdSt|jt r tdSt|jtr tdSt|jtr tdSJ)Nsnonesstringslistssetszsetshashsstream) rrr*rr/rrrr)r s r8rzBaseFakeSocket._key_value_types 99 ( (  5 ) * *  4 (( (  #5 6' '  4 (( (  4 (( (  7 + * * 5rA)r'N)8__name__r __qualname__r__annotations__rredisConnectionErrorrrr rr`propertyrryrr{r+r}rrrrrrr staticmethodrr rkrrrrrrrr$rrrr-rrr rr rrr$r(r*rr __classcell__)rts@r8rMrMSsRX&&&"$33** * *  *  * *@&e&&$sCx$$(S((   #$ s t   5 U5%<%8  *4d?!;*8+&tE{+&t+&ZXseSj129BJNs)bf 6W;W3DW S S %s *;!<HdVUX[DY^a2 c eHXseSj=Q4RT]4].^  F t <Q(5/<Q3<Qu<QeTY[_`e[fTfNgIh<Q|F FEFcF2%222 !!! [\rArM)9r0loggingrgrrtypingrrrrrrr r r r r r/fakeredis.modelrrrrrrr_command_args_parsingr _commandsrrrrrr_helpersrrrrr r!r"r#_typingr$ getLoggerrr3rr@rrGrJrKrMr^rAr8r=s hhhh OO/``   '   ; ' 3 3 T%[U3S >-B,3CC 3~~rA