o Y۷i@sUdZddlmZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl mZmZmZmZddlmZddlmZddlmZddlmZmZmZmZd d lmZmZmZm Z m!Z!m"Z"erudd lm#Z#m$Z$ed Z%d Z&de'd<e(ddej)e*e ddedde+e"ddeddej,e*e j-ddedde.e/e 0dj1ddej2e*ej3e*ej4e*ej5e*iZ6de'd<d>d?d%d&Z7Gd'd(d(e j8Z9   )d@dAd2d3Z:dBdCd8d9Z;dDd;d<Zr#cCs tt|SN)strdictrr!r!r"r# s cCsd|jdS)N cbor_simple:d)valuerr!r!r"r#!r$cCsdS)Nz cbor:undefr!rr!r!r"r#"scCs|Sr%) isoformatrr!r!r"r##scC|jSr%)urnrr!r!r"r#%cCsd|jd|jiS)NzCBORTag:r))tagr*rr!r!r"r#&scCr,r%)patternrr!r!r"r#(r.z dict[type, Callable[[Any], Any]]default_encodersr!decoderrr/r ignore_tagsCollection[int]returnobjectcCsD|j|vr|jS|jdkr||jS|jr d|jd|jS|S)NzCBORtag::)r/r*decode_from_bytes immutable)r3r/r4r!r!r"tag_hook0s   r<c@seZdZdddZdS)DefaultEncodervr r6cCs*|j}t|}|r||Stj||Sr%) __class__r2getjson JSONEncoderdefault)selfr>obj_typeencoderr!r!r"rC=s  zDefaultEncoder.defaultN)r>r r6r )__name__ __module__ __qualname__rCr!r!r!r"r=<sr=strictfr ,Callable[[CBORDecoder, CBORTag], Any] | None object_hook3Callable[[CBORDecoder, dict[Any, Any]], Any] | None str_errors%Literal['strict', 'error', 'replace'] Iterator[Any]ccs:t||||d} z|VWn tyYdSwq )N)r<rMrO)rrEOFError)rKr<rMrOr3r!r!r" iterdecodeFs rSr)dict_idsset[int] | None$str | list[Any] | dict[str, Any] | TcsZtpgi}t|tsFt|trd|jdSt|tttfrDt|vr+td t|fdd|D} t||S|St|vrPtd t|| D]O\}}t|t rl|j ddd}nt|tryd|jd}n t|tttfrt|}t|trt|||<q[t|tttfrfd d|D||<q[|||<q[|S) Nr(r)z,Cannot convert self-referential data to JSONcg|]}t|qSr! key_to_str.0r rTr!r" azkey_to_str..rrrcrWr!rXrZr\r!r"r]wr^)set isinstancer'rr*tuplelistid ValueErroraddremoveitemsbytesrr frozensetr&rY)r)rTrvalr>kr!r\r"rYTs:         rYNonec Csd}d}tj||d}|jddtddd|jd d dgd d |jd ddddd|jdddddd|jdddddd|jdddddd|jddtddd |}|jdkr_d!}d}n|j}d"}|jd#}d$d%|D}tt |d&}t |d'd(d)|d*} |j D]u} t h} | dkrt jj} n | t | d+d,} |jrtt| } z.|jrt| |d-} nt| |d-f} | D]}tjt|| |jd.|jtdd/| d0qWnt!t"fy}zt#|d}~wwWdn1swYqWddS1swYdS)1Nzpython -m cbor2.toolz[A simple command line interface for cbor2 module to validate and pretty-print CBOR objects.)prog descriptionz-oz --outfilez output file-)typehelprCinfiles*z2Collection of CBOR files to process or - for stdin)nargsrCrqz-kz --sort-keys store_trueFz5sort the output of dictionaries alphabetically by key)actionrCrqz-pz--prettyzindent the output to look goodz-sz --sequencez+Parse a sequence of concatenated CBOR itemsz-dz--decodez-CBOR data is base64 encoded (handy for stdin)z-iz --tag-ignorer0z@Comma separated list of tags to ignore and only return the value)rprCrqrT,cSs(h|]}t|r|drt|qS)r)lenisdigitint)r[nr!r!r" s(zmain..)r4wrr)moderrclosefdrb)r~)r<)N) sort_keysindentcls ensure_ascii )$argparseArgumentParser add_argumentr& parse_argsoutfile tag_ignoresplitr r<openrrrsysstdinbuffer enter_contextrioBytesIObase64 b64decodereadsequencerSrrAdumprYrprettyr=writerdrR SystemExit)rmrnparseroptionsrrignore_sdroptagsmy_hookoutfppathstackinfileobjsobjer!r!r"main~s        $r__main__)r!)r3rr/rr4r5r6r7)NNrJ) rKr r<rLrMrNrOrPr6rQr%)r)rrTrUr6rV)r6rl)>__doc__ __future__rrrdecimal fractionsr ipaddressrAreruuidcollections.abcrrrr contextlibrr functoolsr typingr r r r r0rrrrrrrrrr__annotations__rhDecimalr&rpFractionUUIDr_rbcompiler? IPv4Address IPv6Address IPv4Network IPv6Networkr2r<rBr=rSrYrrGr!r!r!r"sb             *_