j/fTdZddlZddlZddlZddlZddlmZmZddlm Z m Z m Z dej fde de ee ffdZedejd d ZGd d eZGd deZGddZGddZeZdZde defdZdZdZdS)a8JSONPath implementation for Python. This module provides a lightweight JSONPath implementation with support for: - Standard JSONPath operators ($, @, ., .., *, []) - Filter expressions with comparison, membership, and regex operators - Sorter expressions for ordering results - Field extractor expressions - Value updates via JSONPath Example: >>> from jsonpath import JSONPath, search >>> data = {"store": {"book": [{"price": 10}, {"price": 20}]}} >>> JSONPath("$..price").parse(data) [10, 20] >>> search("$.store.book[0].price", data) [10] N) OrderedDict defaultdict)AnyCallableUnionnamelevelcBtj|}|jr|Stjd|dd}tj}|||||||||S)z,Get or create a logger used for local debug.z%(asctime)s-%(levelname)s-[z ] %(message)sz[%Y-%m-%d %H:%M:%S])datefmt)logging getLoggerhandlers Formatter StreamHandlersetLevel setFormatter addHandler)rr logger formatterhandlers [/lsinfo/ai/hellotax_ai/base_platform/venv/lib/python3.11/site-packages/jsonpath/jsonpath.py create_loggerrs  t $ $F !"S"S"S"S]rsssI#%%G U ### OOE g Mjsonpath PYLOGLEVELINFOceZdZdZdS)ExprSyntaxErrorzRaised when a JSONPath expression has invalid syntax. Examples of invalid syntax: - Using sorter on non-collection types - Using field-extractor on non-dict types N__name__ __module__ __qualname____doc__rrrr2rrceZdZdZdS)JSONPathTypeErrorzRaised when type-related errors occur during JSONPath operations. Examples: - Comparing incompatible types during sorting (e.g., str vs int) - Sorting with missing keys that result in None comparisons Nrr$rrr'r';r%rr'c eZdZdZdddZeZdZdZe j dZ e j dZ e j d Z e j d Ze j d Ze j d Ze j d Ze j dZe j dZe j dZe j dZe j dZe j dZe j dZe j dZe j dZe j dZe j dZehejej ej!ej"ej#ej$ej%ej&ej'ej(ej)ej*ej+ej,ej-ej.ej/ej0ej1ej2ej3ej4ej5ej6ej7ej8ej9ej:ej;ej<ej=ej>ej?ej@ejAejBejCddDzZDehdZEeddhZFdeGfdZHdId"ZIdJd#ZJd$ZKdKd&eGd'eGd(eGd)eGfd*ZLdKd&eGd+eGd(eGd)eGfd,ZMd-ZNd.ZOd/ZPd0ZQd1ZRd2ZSd3ZTd4ZUeVd5ZWeVd6eGd)eGfd7ZXeVd8eYfd9ZZeVd:e[d6eGfd;Z\eVde]d6eGfd?Z^eVd@Z_eVdAZ`eVdBZaeVdCZbd:e[d6eGdDeGfdEZcd:e[fdFZdd>eeefeYfdGeee]ege]ge]ffd)e]fdHZhd!S)LJSONPathaJSONPath expression parser and evaluator. A JSONPath expression is used to navigate and extract data from JSON objects. This implementation supports extended syntax including filters, sorters, and field extractors. Attributes: RESULT_TYPE: Supported result types ('VALUE' or 'PATH'). Example: >>> jp = JSONPath("$.store.book[?(@.price < 10)].title") >>> jp.parse({"store": {"book": [{"title": "A", "price": 5}]}}) ['A'] zA list of specific values.zAll path of specific values.)VALUEPATH;z;..;z\.\.z(?=|==|!=|>|<| in| not| is|\s|\)|$)|len\(@([.\[].*?)\)zB(?:\.|^)(?P\w+)|\[['\"](?P.*?)['\"]\]|\[(?P\d+)\]z =~\s*/(.*?)/z\.(\w+|'[^']*'|\"[^\"]*\")z\.(\.#B)z(?zJSONPath.s2 h h hqX_`cefXgXg h73?? h h hr)IndexNumStrBytes NameConstant>len__obj RegexPatternr9r;exprctt|_g|_d|_g|_d|_d|_||}d| tj D|_t|j|_ttjr$td|jdSdS)zInitialize JSONPath with an expression. Args: expr: JSONPath expression string (e.g., "$.store.book[*].price") rr*Ncg|]}||Sr$r$)r1ss r z%JSONPath.__init__..sBBBqBBBBrz segments : )rlistsubxsegmentslpathresult result_type_custom_eval_func _parse_exprsplitr)SEPr9r isEnabledForr DEBUGdebugselfr<s r__init__zJSONPath.__init__s %%    "!%%%BBDJJx|$<$<BBB ''   w} - - 9 LL7 77 8 8 8 8 8 9 9rr*NcPt|ttfstd|tjvr;t dttj||_ ||_ g|_ | |dd|j S)aParse JSON object using the JSONPath expression. Args: obj: JSON object (dict or list) to parse result_type: Type of result to return - 'VALUE': Return matched values (default) - 'PATH': Return JSONPath strings of matched locations eval_func: Custom eval function for filter expressions. If None (default), uses a safe expression evaluator that prevents code injection. Pass a custom function only if you trust the JSONPath expressions being evaluated. Returns: List of matched values or paths depending on result_type Raises: TypeError: If obj is not a dict or list ValueError: If result_type is invalid zobj must be a list or a dict.zresult_type must be one of r$) isinstancerAdict TypeErrorr) RESULT_TYPE ValueErrortuplekeysrFrGrE_trace)rOobjrF eval_funcs rparsezJSONPath.parses(#d|,, =;<< < h2 2 2_5AUAZAZA\A\;];]__`` `&!*  CC   {rc.|||S)zDAlias for parse(). Search JSON object using the JSONPath expression.)r])rOr[rFs rsearchzJSONPath.searchszz#{+++rcttjrtd|t j|j|}t j |j |}t j |j |}t j |j|}t jd|}t jt j|}t jt j|}t j|j|}t j|j|}t j|j|}t j|j|}|dkrd}n|dr |dd}ttjrtd||S) zParse and normalize JSONPath expression into segments. Handles special patterns (quotes, brackets, parentheses) by temporarily replacing them with placeholders, then splits by dots and restores. zbefore expr : z\1rRz$;Nzafter expr : )rrKr rLrMr) REP_GET_QUOTEsub _get_quoteREP_GET_BACKQUOTE_get_backquote REP_GET_PAREN _get_parenREP_GET_BRACKET _get_bracketREP_DOTDOT_BRACKET REP_DOUBLEDOT SEP_DOUBLEDOTREP_DOTrJREP_PUT_BRACKET _put_bracket REP_PUT_PAREN _put_parenREP_PUT_BACKQUOTE_put_backquote REP_PUT_QUOTE _put_quote startswithrNs rrHzJSONPath._parse_exprs   w} - - 2 LL0$00 1 1 1%))$/4@@)--d.A4HH%))$/4@@'++D,=tDD*..ud;;%))(*@$GG##HL$77'++D,=tDD%))$/4@@)--d.A4HH%))$/4@@ 3;;DD __T " " 8D   w} - - 2 LL0$00 1 1 1 rra pattern_typecontentwrapperreturnct|j|}|j|||r|||S||S)a2Save pattern content and return placeholder. Args: pattern_type: Pattern identifier (e.g., '#Q', '#BQ', '#B', '#P') content: Content to save wrapper: Optional wrapper format string (e.g., "'{}'", "`{}`") Returns: Placeholder string )r9rBappendformat)rOryrzr{r2s r _save_patternzJSONPath._save_patternsi  ,' ( ( ,&&w///  8>>\"61"6"677 7####rindexct|j|t|}|r||S|S)aARestore pattern content from placeholder. Args: pattern_type: Pattern identifier (e.g., '#Q', '#BQ', '#B', '#P') index: Index as string wrapper: Optional wrapper format string (e.g., "'{}'", "`{}`") Returns: Original content with optional wrapper )rBintr)rOryrr{rzs r_restore_patternzJSONPath._restore_patterns:)L)#e**5  +>>'** *rcT|d|dS)N#QrgrouprOms rrezJSONPath._get_quote(s"!!$ 333rcV|d|ddS)Nrrz'{}'rrrs rrwzJSONPath._put_quote+s$$$T1771::v>>>rcV|d|ddS)N#BQrz`{}`rrs rrgzJSONPath._get_backquote.s$!!%V<<.replEs; Atz!!1xxx::: rr:zlen(r)rr) REP_ATTR_PATHrd)ris_lenrzrrEs r_gen_objzJSONPath._gen_obj@s4'''!***     (,,T7;;7"  &%F%%%F rpathct|tr|d|dS|s*|r/|ddr|d|S|d|dS)zBuild JSON path string for a given key. Args: path: Current path string key: Key (string) or index (int) Returns: Formatted path string rr_arrr)rSr isidentifierreplaceisalnum)rkeys r _build_pathzJSONPath._build_pathQs c3   $##S### #      ## ##++c3*?*?*G*G*I*I #??S?? "!!#!!!!rrc|dr|dS|dr|dS|drt|dSdS)zExtract key from regex match group dictionary. Args: group: Match group dictionary with 'dot', 'quote', or 'int' keys Returns: Key as string or int dotquoterN)rrs r_extract_key_from_groupz JSONPath._extract_key_from_groupcsS < <  > "> ! < %uU|$$ $tric Pt|tr;t|D])\}}|||t||g|R*dSt|t r>|D]+\}}|||t||g|R*dSdS)aCTraverse object children and apply function to each. Args: f: Function to apply to each child element obj: Object to traverse (list or dict) i: Current segment index path: Current JSONPath string *args: Additional arguments to pass to function f N)rSrA enumerater)rrTitems)fr[rrargsidxvks r _traversezJSONPath._traverseus c4  >#C.. @ @Q!Q,,T377?$????? @ @ T " " >  > >1!Q,,T155====== > > > >rFconvert_number_strr[cd|vr.t|tr ||vr ||}nctjS|}|dD]?}t|tr||vr ||}$tjcStjcS|rXt|t rC |rt|St|S#t$rYnwxYw|S)aZGet attribute value from object by dot-notation path. Args: obj: Source object (dict) path: Dot-separated path string (e.g., "author.name") convert_number_str: If True, convert numeric strings to int/float Returns: The value at the path, or _MISSING sentinel if not found r) rSrTr)_MISSINGrIstrisdigitrfloatrW)r[rrrrs r_getattrzJSONPath._getattrs d??#t$$ )I((AZZ__ - -a&&-AvvaD'0000#,,,,  *Q"4"4  99;;"q66MQxx    s""CC C! C!c^d |ddddD]c}|}|dr||ffd dH||ffd ddS#t$r}t d ||d}~wwxYw) z2Sort objects by multiple fields using stable sort.clt|d|d}|tjur|ndS)NrTr)r)rr)trrs rkey_funcz"JSONPath._sorter..key_funcs7!!!A$d!CCA!22211 z"JSONPath._sorter..sAabbE0B0BrT)rreversec||SNr$rs rrz"JSONPath._sorter..sXXa^^r)rz2not possible to compare str and int when sorting: )rIstriprxsortrUr')r[sortbyssortbyers @r_sorterzJSONPath._sorters  = = = e!--,,TTrT2 E E$$S))EHH(.BBBBB $ HHV!C!C!C!C!CHDDDD E E e e e#$\YZ$\$\]]cd d esBB B,B''B,c tj|d}n%#t$r}td||d}~wwxYwtj|D]}t |}|t jvrtd|j|tj ur*|j t j vrtd|j |tj ur1|j drtd|j |tjurFt!|jtj r|jj t jvstd dS) aValidate that a filter expression only contains safe AST constructs. Raises ValueError if the expression contains potentially dangerous constructs like function calls (except len/RegexPattern), attribute access to dunder names, or disallowed node types. eval)modez"Invalid filter expression syntax: Nz!Disallowed expression construct: z&Disallowed name in filter expression: rzDisallowed attribute access: zEOnly len() and RegexPattern() calls are allowed in filter expressions)r/r] SyntaxErrorrWwalktyper)_ALLOWED_AST_NODESr Nameid_ALLOWED_NAMES AttributeattrrxCallrSfunc_ALLOWED_CALLS)r<treernode node_types r_validate_filter_exprzJSONPath._validate_filter_exprsi N9T///DD N N NE!EEFFA M NHTNN n nDT I ;;; !YYEW!Y!YZZZCH$$8O)O)O !S$'!S!STTTCM))di.B.B3.G.G) !L!L!LMMMCH$$"49ch77nDILHLc2.#|dg0h0hiiirc8|d}d}t|dkr||dnd}t|dkr||dnd}t|dkr||dnd}t|||S)zParse a slice expression string into a slice object. Args: s: Slice string like '1:3', '::2', '-1:', etc. Returns: A slice object :cP|}|rt|ndSr)rr)rs rto_intz%JSONPath._parse_slice..to_ints$ A(3q666D (rrNrrb)rIr9slice)r?partsrstartstopsteps r _parse_slicezJSONPath._parse_slices  ) ) )%(JJNNuQx   #&u::>>vveAht#&u::>>vveAhtUD$'''rrcd} |j ||d|td}n|||}n#t$rYnwxYw|r||||dSdS)a"Evaluate filter expression and continue trace if condition is true. Args: obj: Current object to evaluate against filter i: Next segment index to trace path: Current JSONPath string step: Python expression string to evaluate FN)r:r;)rGr;r ExceptionrZ)rOr[rrrrs r_filterzJSONPath._filters  %1**4Vb7c7cdd**455    D   & KKQ % % % % % & &s=A AAc r ||jkr|jdkr|j|n%|jdkr|j|tt jr td|d|dS|j |}|dkr"| |j ||dz|dS|dkr9| ||dz|| |j |||dSt|tr^|rJt|}|t!|kr&| |||dz|d |d dSt!|d kr"|d d kr|dd kr |ddn|}t|t"r:||vr6| |||dz|||dSt|trt&j|ratt-|}|||}|D]%\} } | | |dz|d | d &dSt|t"rt&j|rf|dD]N} | } | |vr4| || |dz||| OdS|r4|d dvr)|dr|dr|d d}t&j|j|}t&j d|}d|vr t&j!d|}t|t"r|"||dz||| |j"||dz||dS|dr"t|trqtt-|}|#||d d|D]3\} } | | |dz||| 4nt|t"rvt|$}|#||d d|D]3\} } | | |dz||| 4ntKddS|r|d dkr|drt|t"rzi} |dddD]?} | } |&|| } | t&j'ur| | | <@| | |dz|ntKddSdSdSdS)aRecursively traverse object following JSONPath segments. This is the core evaluation method that processes each segment of the parsed JSONPath expression and navigates through the object accordingly. Args: obj: Current object being traversed i: Index of current segment in self.segments path: JSONPath string representing current location r*r+zpath: z | value: N*rz..rrrbrrrrz?/rz?(r:z=~z@ RegexPattern(r'\1')z/(z"sorter must acting on list or dictrz#field-extractor must acting on dict)(rDrFrEr~rrKr rLrMrCrrZrSrArrr9rTrr)REP_SLICE_CONTENT fullmatchrrREP_SELECT_CONTENTrIrendswithrxREP_FILTER_CONTENTrdr REP_BARE_ATREP_REGEX_PATTERNrrrrrr) rOr[rrrikeystep_keyindexedvalsrrrobj_s rrZzJSONPath._tracesN  ??7** ""3''''!V++ ""4(((""7=11 = ;d;;c;;<<< F}Q 3;; NN4;QUD 9 9 9 F 4<< KKQUD ) ) ) NN4;Q 5 5 5 F c4  T\\^^ t99Dc#hh CIq1u.?.?.?.?.?@@@ F#&d))q..T!W^^RTW4"::^b c4  X__ KKH q1ud.>.>tX.N.N O O O F c4  X%?%I%I$%O%O 9S>>**G4,,T223D 8 8Q Aq1u&6&6&6&6&67777 F c4  X%@%J%J4%P%P ZZ__ J JGGII88KKAAt/?/?a/H/HIII F  DGtOO c(:(:Ot$$ AbDz266t}dKK +//>>4<<#599:RTXYYDc4((9LLa!eT4888t|S!a%tDDDt$$ c4(( Py~~..CLLd1R4j111"%KKQ Aq1ud.>.>tS.I.IJJJJKT**Psyy{{++CLLd1R4j111 #II1 Aq1ud.>.>tQ.G.GHHHHI**NOOO  DGsNNt}}S'9'9N#t$$ Mad))#..$$A A c1--A 111"#Q D!a%....%&KLLL F  NNNNr value_or_funcc||d}t|}t|dkr|ddkr|r ||n|S|D]}ttj|}|s1|}|ddD]1}||} || }2||d} |r||| n||| <|S)aIUpdate values in JSON object using JSONPath expression. Args: obj: JSON object (dict or list) to update value_or_func: Static value or callable that transforms the current value Returns: Updated object (modified in-place for nested paths, returns new value for root) r+)rFrrrRNr) r]callabler9rAr)REP_PATH_SEGMENTfinditerr groupdict) rOr[r pathsis_funcrmatchestargetmatchrs rupdatezJSONPath.updates( 3F 33=)) u::??uQx3)0C==%%%m C S SD84==dCCDDG F " % %225??3D3DEE..wr{/D/D/F/FGGC8?R--s 444]F3KK r)r*N)r*)ra)ir r!r"r#rVobjectrrJrnrecompilermrorcrvrfrtrjrprhrrrrrrrrrlr frozensetr/ ExpressionBoolOpAndOrBinOpAddSubMultDivFloorDivModMatMultUnaryOpNotUAddUSubCompareEqNotEqLtLtEGtGtEIsIsNotInNotInConstantr SubscriptSliceListTupleDictrrLoadrrrrrPr]r_rHrrrerwrgrurkrqrirs staticmethodrrrTrrrrrrrrrrrZrrArrr$rrr)r)Ds   ..K vxxH CMBJw''Mbj+,,GBJ~..MBJz**M" >22" ;// bj!122O bj,,OBJ/00MBJz**M# #BCC#$?@@#$mnn!rz"ghh" ?33BJ<==M#K00"*344K#0 N0  J0  G 0 F 0  I0  G0  G0  H0  G0  L0  G0  K0 K!0 " G#0 $ H%0 & H'0 * K+0 , F-0 . I/0 0 F10 2 G30 4 F50 6 G70 8 F90 : I;0 < F=0 > I?0 B LC0 D HE0 H MI0 J IK0 N HO0 P IQ0 R HS0 V MW0 Z H[0 ^ H_0 d i h$T h h he2 i44jY???@@NY~677N9S9999(    D,,,,>$$#$$c$SV$$$$"SsTW 444???===888:::777@@@777\ "#"s"""\""t\">S>>>>\>">> pattern = RegexPattern(r"^test") >>> "testing" @ pattern True cF||_tj||_dS)z'Initialize with a regex pattern string.N)patternrr _compiled)rOrBs rrPzRegexPattern.__init__s G,,rc~t|tr't|j|SdS)z@Right matmul operator (@) - checks if other matches the pattern.F)rSrboolrCr_)rOothers r __rmatmul__zRegexPattern.__rmatmul__s7 eS ! ! 6--e4455 5urN)r r!r"r#rPrGr$rrr;r;s<  --- rr;r<r|c|tvrt|nOtttkrtdt |t|<t|S)zGet or create a cached JSONPath instance. Args: expr: JSONPath expression string Returns: Cached or newly created JSONPath instance F)last)_jsonpath_cache move_to_endr9_CACHE_MAX_SIZEpopitemr)r<s r_get_cached_jsonpathrPsm ##D))))   ? 2 2  # # # / / / ( 4  rc t|S)aCompile a JSONPath expression for reuse. Returns a cached JSONPath instance when available, avoiding redundant parsing. Args: expr: JSONPath expression string Returns: JSONPath object that can be used to parse multiple JSON objects Example: >>> jp = compile("$.store.book[*].price") >>> jp.parse(data1) >>> jp.parse(data2) )rPrOs rrrs  % %%rcFt||S)zSearch JSON data using JSONPath expression with caching. Args: expr: JSONPath expression string data: JSON data (dict or list) Returns: List of matched values )rPr])r<datas rr_r_s   % % + +D 1 11r)r#r/r osr collectionsrrtypingrrrrrrrgetenvrrrr'r)r;rKrMrPrr_r$rrrXs$  00000000''''''''''#W\5c?( z929\6#B#B C Ci b b b b b b b b J2+--!s!x!!!!(&&&& 2 2 2 2 2r