ddlZddlZddlZddlZddlmZddlmZddlm Z m Z m Z m Z m Z mZmZmZmZmZddlmZmZddlmZGdde ZGd d e ZGd d ZeGd deZGddeZGddZdS)N)Counter) dataclass) ListUnionTupleOptional NamedTupleDictAnySequence GeneratorAnyStr) BeforeAnyAfterAny) current_timecPeZdZUeed<eed<defdZededdfdZ dS)StreamEntryKeytsseqreturncH|jd|jS)N-)rrencodeselfs /Users/kimhansen/Desktop/03 Workspace/ceo-agents/chl-effectiveness/mcp-servers/whoop/.venv/lib/python3.11/site-packages/fakeredis/model/_stream.pyrzStreamEntryKey.encodes&'&&DH&&--/// entry_keycPt|tr|n|}|d}t |dkrt |ddfn)t |dt |df\}}t ||S)Nrr) isinstancebytesdecodesplitlenintr)r entry_key_strparts timestampsequences r parse_strzStreamEntryKey.parse_strs3=i3O3O^Y--///U^ ##C((69%jjAooU1X 2 2CPUVWPXMM[^_def_g[h[hKiHi222rN) __name__ __module__ __qualname__r&__annotations__r"r staticmethodrr+rrrr sr GGG HHH000003V3(8333\333rrc2eZdZUdZeed<eed<eed<dS)PelEntryzFPending Entry List entry: tracks consumer ownership and delivery count consumer_name time_readtimes_deliveredN)r,r-r.__doc__r"r/r&r1rrr3r3s9PPNNNrr3czeZdZdZdeeeefdefdZ e de defdZ e d dededdfd Zd S) StreamRangeTestz0Argument converter for sorted set LEX endpoints.value exclusivec"||_||_dSNr:r;)rr:r;s r__init__zStreamRangeTest.__init__'s "rrrc^ t|dS#t$rYdSwxYw)NTF)rr+ ValueError)rs r valid_keyzStreamRangeTest.valid_key+sB   $ $Y / / /4   55 s  ,,Fc6|dkr|tdS|dkr|tdS|dddkr,|t|dddS|t||S)N-T+r ()rrrr+)clsr:r;s rr#zStreamRangeTest.decode3s D==3y{{D)) ) d]]3xzz4(( ( 2A2Y$  3~//abb ::DAA As>++E22I>>>rNF)r,r-r.r7rrrrboolr?r0rrB classmethodr"r#r1rrr9r9$s::#eNIx$GH#UY####V\??5?T?>O???[???rr9cxeZdZUeed<eed<eed<eed<deddfdZdedeee eefffd Z dS) StreamConsumerInfonamepending last_attempt last_successrNcZ||_d|_t}||_||_dSNr)rMrNrrOrP)rrM_times rr?zStreamConsumerInfo.__init__Es/  !!r curr_timecD|j|j||jz ||jz dS)N)rMrNidleinactive)rMrNrOrP)rrTs rinfozStreamConsumerInfo.infoLs1I| 11!D$55    r) r,r-r.r"r/r&r?r strrrXr1rrrLrL>s KKK LLL"U"t"""" c d3eSj0A+A&B      rrLcneZdZ d#dddededeefdZdedeed dfd Zd ed efd Z d ed efd Z d e e e eeefffdZd e eeffdZd edededed e e eee eefff dZd$dZdeed efdZdeedeedeedeedeed e e ef dZd e efdZdedeeeeefd edeed ed ee ee eff d!Zdededed e efd"ZdS)% StreamGroupNstreamXStreamrM start_key entries_readcv||_||_||_||_i|_||_||_i|_dSr=)r\rMr^r_ consumerslast_delivered_key last_ack_keypel)rr\rMr^r_s rr?zStreamGroup.__init__VsE  "(:<"+%46rlast_delivered_strrc(|j||_|j|j\}}|pd|_|jt ||pdzt|jdz |_dS)z#Set last_delivered_id for the grouprr N) r\ parse_ts_seqr^ find_indexr_ get_indexminr%rb)rrer_ start_index_s rset_idzStreamGroup.set_idis112DEE;11$.AAa(-A"&+"7"7Kz.StreamGroup.consumers_info..~s2OOO1q!&&|~~66OOOr)rars`rconsumers_infozStreamGroup.consumers_info}sOOOOOOOOrc*|j|j\}}|j|j\}}|j|j\}}||jpdzt |jkr"t |j|z |jpdz }nt |jdz |z }|jt |jt |j |j |j|d}|S)Nrr )snames consumersspendingslast-delivered-ids entries-readslag) r\rhr^rbrcr_r%rMrardr)rrkrllast_delivered_indexlast_ack_indexlagrrs r group_infozStreamGroup.group_infos//?? Q"&+"8"89P"Q"Qa K2243DEE $+0q 1C 4D4D D Ddk""[0D4E4JKCCdk""Q&)==CYdn--DH "&"9"@"@"B"B!.    rstart_idcountnoackct}|jvrt|j|<|j|_|dkrj}n-t t |j}j ||}|s|D]}t||dj |<t|dkr>t j|d_j pdt|z_ |j|_j|xjt|z c_fd|DS)N>r rcDg|]}j|Sr1)r\ format_recordrwxrs rryz*StreamGroup.group_read..s)??? ))!,,???r)rrarLrOrbmaxrr+r\ stream_readr3rdr%r_rPrN) rr4rrrrSr^ids_readrxs ` r group_readzStreamGroup.group_readsL  . .,>},M,MDN= )5:}%2 t  /IIN44X>>@WXXI;**9e<< @ @ @&}eQ?? x==1  &)$*A8B<&P&PD #!%!2!7a3x== HD 5:}%2 }%--X>--????h????rc dtj|jdD}|D]R}||jvrt ||j|<|||j|_|||j|_SdS)NcDi|]\}}|t|djS)c|jSr=)r5rs rzAStreamGroup._calc_consumer_last_time...sAKrkey)rjr5)rwrxvs r z8StreamGroup._calc_consumer_last_time..sB   1 s1//000:   rc|jSr=)r4rs rrz6StreamGroup._calc_consumer_last_time..s1?rr) itertoolsgroupbyrdvaluesrarLrOrP)rnew_last_success_mapconsumers r_calc_consumer_last_timez$StreamGroup._calc_consumer_last_times  !)$(//*;*;AZAZ[[[   - S SHt~--+=h+G+Gx(4H4RDN8 $ 14H4RDN8 $ 1 1  S Srargscd}|D]q} t|}n#t$rY*wxYw||jvr:|j|j}|j|xjdzc_|j|=|dz }r||Sro)rr+ Exceptionrdr4rarNr)rrrrrxparsedr4s rackzStreamGroup.acks  A '11!44    !! $ 0 > }-55:55HV$q %%''' s # 00rVstartendrc2ttj}fd|D}fd|D}fd|D}|t |d|}fd|DS)Nc@g|]}j|jk|Sr1rdr4)rwrxrrs rryz'StreamGroup.pending..s-]]]!tx{7PT\7\7\A7\7\7\rcFg|]}j|jzk|Sr1rdr5)rwrxrSrVrs rryz'StreamGroup.pending..s2\\\!tx{7Lt7SV[7[7[A7[7[7[rcg|]>}j|ksj|kr&jsj|ksj|k5j<|?Sr1r>)rwrxrrs rryz'StreamGroup.pending..s\kAoo5;!+;+;EO+;)a--SY!^^CM^ .<^^rcg|]M}|j|jj|jz j|jgNSr1)rrdr4r5r6)rwrxrSrs rryz'StreamGroup.pending..s^   XXZZ!2EDHQK.s"GGG48A;4GGGrrc$g|] }||g Sr1r1)rwicounters rryz/StreamGroup.pending_summary..s! . . .a_ . . .r)rrdr%rjrr)rdatars` @rpending_summaryzStreamGroup.pending_summarysGGGGdhGGGHH MM&)$(mma&7&7CMM " " "T&)$(mma&7&7CMM " " "T . . . .g . . .    r min_idle_msmsgsrSforcecLt}||}||j|t|_gg}}|D],} t | t rt| n| } n#t$rYBwxYw| |j vrX|rUt||d|j | <| |j vr| | n| | |j | =||j | jz |kr|j | j} t||| dz|j | <| |j vr| | | | |j | =.|t#|t#|fSNr )rragetrLrOr!r"rr+rrdr3r\appendr5r6rr) rrrr4rSrrT claimed_msgs deleted_msgsmsgrold_times_delivereds rclaimzStreamGroup.claims!NN =E\e=*<]*K*KLLY%'l  " "C 7A#u7M7MVn..s333SV    $(""*$,]E1$E$EDHSMdk))$++C0000$++C000 HSM48C=22[@@"&(3-"? $]E;NQR;RSSDHSMdk!!##C((((##C(((HSMM %%'''l##VL%9%999s 1A<< B B ct|ttfdjD}t |t |}|d|S)NcRg|]#}j|jz k|k!|$Sr1r)rwrxrTrrr^s rryz-StreamGroup.read_pel_msgs..s>tttQy48A;;P/PT_/_/_efjsesesqesesesr)rr+rrrdrjr%)rrrrrrTr^s`` @@r read_pel_msgszStreamGroup.read_pel_msgssr",,U33  NN ttttttt$(tttuuE3t99%%FUF|rr=rN)r,r-r.r"rrr&r?rmrprsrr rYrrzr rrIrrrrr9rNrr rrr1rrr[r[Us '+ 6666" 6 sm 6666&vvhsmvPTvvvv%C %CPT#uUCZ/@*@%A BPPPPD,$@"@.3@': ': } ':  ': tN#T.%99 :':':':':RU34P^K_rr[c eZdZdZd2dZdedeefdZdeded ee ddfd Z dede fd Z de e eeffd Zd ede efdZde ede fdZd3deeee fdededeffdZdZde fdZdee eee efedffdZdedeee effdZde defdZdedefdZd4dedede e effdZ!d ede e effd!Z"e#d"edefd#Z$ d5d$ee d%eed&ee de fd'Z%d6d)e&d*e&d+ede efd,Z'defd-Z(d.ed/ee dfde efd0Z)dede eee eefffd1Z*dS)7r]aClass representing stream. The stream contains entries with keys (timestamp, sequence) and field->value pairs. This implementation has them as a sorted list of tuples, the first value in the tuple is the key (timestamp, sequence). The structure of _values list is: [ ((timestamp, sequence), [field1, value1, field2, value2, ...]), ((timestamp, sequence), [field1, value1, field2, value2, ...]), ] rNchg|_i|_i|_tdd|_d|_dSrR)_ids _values_dict_groupsr_max_deleted_id_entries_addedrs rr?zXStream.__init__/s8*, ?A13 -a33r group_namec8|j|dSr=)rrrrs r group_getzXStream.group_get6s| D111rrM start_key_strr_c|dkr6t|jdkr |jdntdd}nt|}t |||||j|<dS)zAdd a group listening to stream :param name: Group name :param start_key_str: start_key in `timestamp-sequence` format, or $ listen from last. :param entries_read: Number of entries read. $rrN)r%rrr+r[r)rrMrr_r^s r group_addzXStream.group_add9sq D ),TY!);); " PQSTAUAUII&00??I(tY MM Trc,||jvr |j|=dSdS)Nr r)rrs r group_deletezXStream.group_deleteFs#  % % Z(1qrcg}|jD]+}|}||,|Sr=)rrrr)rrrgroup group_ress r groups_infozXStream.groups_infoLsO&(\((** " "E((**I JJy ! ! ! ! rfullc tjtjtjdkr jdndtjdkr jdndjjtjdkrjdndd}|r<fdjD|d<djD|d<ttj | S) Nrrs0-0)slengthgroupss first-entrys last-entrysmax-deleted-entry-ids entries-addedsrecorded-first-entry-idc:g|]}|Sr1)r)rwrrs rryz'XStream.stream_info..^s'HHHt11!44HHHrsentriesc6g|]}|Sr1)r)rwgs rryz'XStream.stream_info.._s LLLallnnLLLrr) r%rrrrrrrrrchainitems)rrrrs` r stream_infozXStream.stream_infoSs949~~4<((@CDIQR@R@RD..ty|<<4;O'P'P$%din5IcNq r*fieldsrc^t|dzdksJt|tr|}||dkrt dt jzd}}t|jdkrS|jdj|kr=|jdj|kr'|jdj}|jdjdz}t||}n|ddkr| d}t|dkrdSt |d|d}}t|jdkr,||jdjkr|jdjdz}nd}t||}nt |}t|jdkr|jd|krdS|j |t||j|<|xjdz c_|S) a+Add entry to a stream. If the entry_key cannot be added (because its timestamp is before the last entry, etc.), nothing is added. :param fields: List of fields to add, must [key1, value1, key2, value2, ... ] :param entry_key: Key for the entry, formatted as 'timestamp-sequence' If entry_key is '*', the timestamp will be calculated as current time and the sequence based on the last entry key of the stream. If entry_key is 'ts-*', and the timestamp is greater or equal than the last entry timestamp, then the sequence will be calculated accordingly. :returns: The key of the added entry. None if nothing was added. :raises AssertionError: If len(fields) is not even. rNrirr r)r%r!r"r#r&timerrrrr$r+rrrrr)rrrrrts_seqr$s raddz XStream.addrs$6{{Q!#### i ' ' +!((**I   S 0 0$,--qB49~~!!dim&6"&<&<2ARVYAYAYYr]%im'!+#B,,FF r]c ! !OOC((E5zzQt%(mmU1XB49~~!!bDIbM,<&<&<im'!+#B,,FF#--i88F ty>>A  $)B-&"8"84    $(LL&! q }}rcdS)NTr1rs r__bool__zXStream.__bool__strc*t|jSr=)r%rrs r__len__zXStream.__len__s49~~rcdttttttftdfffd }|S)Nrc3NKjD]}|VdSr=)rr)rxrs rgenzXStream.__iter__..gens?Y , ,((++++++ , ,r)r rrr"r )rrs` r__iter__zXStream.__iter__sT ,YtE%e*<$=>TIJ , , , , , ,suu rrcL|jt|Sr=)rrr+rrs r __getitem__zXStream.__getitem__s !9!9#!>!>??rrc|j|Sr=)r)rrs rrizXStream.get_indexsy~rc||jvSr=)rrs r __contains__zXStream.__contains__sd'''rT from_leftct|jdkrdS|rtj|j|}|}ntj|j|}|dz }||t|jko|j||kfS)a^Find the closest index to entry_key_str in the stream :param entry_key: Key for the entry. :param from_left: If not found exact match, return index of last smaller element :returns: A tuple (index of entry with the closest (from the left) key to entry_key_str, whether the entry key is equal) r)rFr )r%rbisect bisect_left bisect_right)rrrr check_idxs rrhzXStream.find_indexs ty>>Q  8  $TY ::CII%di;;CaIYTY/UDIi4HI4UVVrr'c|dkr'tt|jdz ddfSt|}||S)a,Find the closest index to entry_key_str in the stream :param entry_key_str: key for the entry, formatted as 'timestamp-sequence.' :returns: A tuple (index of entry with the closest (from the left) key to entry_key_str, whether the entry key is equal) rr rT)rr%rrr+rh)rr'rs rrzXStream.find_index_key_as_strsV D s49~~)1--t3 3))-88v&&&r ts_seq_strcb|dkrtddSt|S)Nrr)rr+)rs rrgzXStream.parse_ts_seqs1   !!Q'' ''' 333r max_lengthstart_entry_keylimitcV||tdd}|t|j|z }n|||\}}|}t t |pdd|p t j}|jd||j|dc}|_|D] } |j| = |S)aTrim a stream :param max_length: Max length of the resulting stream after trimming (number of last values to keep) :param start_entry_key: Min entry-key to keep, cannot be given together with max_length. :param limit: Number of entries to keep from minid. :returns: The resulting stream after trimming. :raises ValueError: When both max_length and start_entry_key are passed. Nz/Can not use both max_length and start_entry_keyr) rAr%rrrjrsysmaxsizer) rr r r  start_indrexactrr remove_keysrxs rtrimz XStream.trims  !o&ANOO O#'  !DI3II  (33ODDJCIs9>1--u/C DD!%4C4$)CDD/ TY % %A!!$$ rFrstopreversecd dtdtdtffd }||}||d}fdt||D}|rt t |S|S) zReturns a range of the stream values from start to stop. :param start: Start key :param stop: Stop key :param reverse: Should the range be in reverse order? :returns: The range between start and stop Telemrrct|jtrdSt|jtrt jS|j|\}}|r|jr ||rdndz }|S)Nrr r)r!r:rrr%rrhr;)rrrrrs r _find_indexz#XStream.irange.._find_indexs$*i00 q$*h// &49~~%Y??JC . .I-qq2-JrF)rcPg|]"}j|#Sr1)rrrs rryz"XStream.irange..s-XXX4%%dil33XXXrT)r9rIr&rangerreversed)rrrrrrstop_indmatchess` rirangezXStream.iranges  o $ #       K&& ;tu555XXXXU9h=W=WXXX  +))** *rct|jdkr|jdndS)Nrrz0-0)r%rrrs r last_item_keyzXStream.last_item_keys9),TY!););ty}##%%%Orr^rc||\}}|r|dz }|t|krgS|||zt|krt|n||z}|j||Sr)rhr%r)rr^rrrend_inds rrzXStream.stream_reads??955 5   NI D ! !I$} E0ASYY0N0N#d)))T]`eTey7*++rcH|j|}||gSr=)rr)rrresultss rrzXStream.format_records"#C( g&&rr)rr)NNNrH)+r,r-r.r7r?r"rr[rr&rrrr r rrIrrrr rrYrrrr rrrrirrrhrr0rgrr9r r"rrr1rrr]r]!s      2E2h{.C2222 Ne NE NRU N[_ N N N Nu T$ucz"23 3 3c 3 3 3 3$v,3 ..(5#45.#.PUVZ\aVaPb....`)Dud5k/A)B$CS$$NO@u@~tE{/J)K@@@@S^((4((((WWNWtWuUXZ^U^O_WWWW$ '6 'eCI>N ' ' ' '44N444\4%))-# SM"#}  :O?T^bcf^g4PuPPPP,^,E#t)r-s  !!!!!!cccccccccccccccccccccccc33333333++++++ 3 3 3 3 3Z 3 3 3z????????4          ,IIIII&IIIXA'A'A'A'A'A'A'A'A'A'r