o 0 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*Zdecode_from_bytesZ 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"rA=s  zDefaultEncoder.defaultN)r<r r6r )__name__ __module__ __qualname__rAr!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:rKrM)rrEOFError)rIr:rKrMr3r!r!r" iterdecodeFs rQr)dict_idsset[int] | None$str | list[Any] | dict[str, Any] | TcsPtpgi}t|tsFt|trd|jdSt|tttfrDt|vr+td t|fdd|D} t||S|St|vrPtd t|| D]J\}}t|t rl|j ddd}nt|tryd|jd}n t|tttfrt|}t|trt|}nt|tttfrfd d|D}|||<q[|S) Nr(r)z,Cannot convert self-referential data to JSONcg|]}t|qSr! key_to_str.0r rRr!r" azkey_to_str..rrrcrUr!rVrXrZr!r"r[wr\)set isinstancer'rr*tuplelistid ValueErroraddremoveitemsbytesrr frozensetr&rW)r)rRrvalr<kr!rZr"rWTs:          rWNonec 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-)typehelprAinfiles*z2Collection of CBOR files to process or - for stdin)nargsrAroz-kz --sort-keys store_trueFz5sort the output of dictionaries alphabetically by key)actionrAroz-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)rnrArorT,cSs(h|]}t|r|drt|qS)r)lenisdigitint)rYnr!r!r" s(zmain..)r4wrr)moderrclosefdrb)r|)r:)N) sort_keysindentcls ensure_ascii )$argparseArgumentParser add_argumentr& parse_argsoutfileZ tag_ignoresplitr r:openrprsysstdinbuffer enter_contextrioBytesIObase64 b64decodereadsequencerQrr?dumprWrprettyr;writerbrP SystemExit)rkrlparseroptionsrr}Zignore_sZdroptagsZmy_hookoutfppathstackinfileZobjsobjer!r!r"main~s        $r__main__)r!)r3rr/rr4r5r6r7)NNrH) rIr r:rJrKrLrMrNr6rOr%)r)rrRrSr6rT)r6rj)>__doc__ __future__rrrdecimal fractionsr ipaddressr?reruuidcollections.abcrrrr contextlibrr functoolsr typingr r r r r0rrrrrrrrrr__annotations__rfDecimalr&rnFractionUUIDr]r`compiler= IPv4Address IPv6Address IPv4Network IPv6Networkr2r:r@r;rQrWrrEr!r!r!r"sb             *_