-:j4dZddlZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z m Z ddlZddlmZddlmZmZmZmZmZGdd ee Zd e d e d dfd ZGddZded efdZde d eee ffdZej>dZ ded efdZ!ded efdZ"ded efdZ#de$ede$e$e d eeeffdZ%de d efdZ&de$ede$e$ede'd e'fd Z(de d efd!Z)eZ*y)"z"Output framework for the `hf` CLI.N)Sequence)Enum)Anycast)ConfirmationError)ANSI StatusLinedisable_progress_barsis_agenttabulatec$eZdZdZdZdZdZdZdZy) OutputFormatzGOutput format for CLI commands with auto detection of agent/human mode.agentautohumanjsonquietN) __name__ __module__ __qualname____doc__rrrrr]/var/www/html/tokenscope/api/venv/lib/python3.12/site-packages/huggingface_hub/cli/_output.pyrr sQ E D E D Errvalueskwargsreturnc"t|i|ddiy)zLike `print`, but always flushed: some CLI flows block on user action right after printing (e.g. the device-code login), so output must not stay buffered.flushTN)print)rrs r _print_flushr!*s 6(V(4(rc eZdZUdZeed<eed<d%dZejfdeddfdZ deddfdZ defd Z d&ddd d e dzd e dzd e dzddfdZ dddddeee efdee dzde dzdee e fdzddf dZdddede dzddfdZde deddfdZddddde dedede ddf d Zd&de dzdefd!Zde ddfd"Zde ddfd#Zde ddfd$Zy)'OutputzOutput sink for the `hf` CLI. Mode is resolved once at init time based on `is_agent()` auto-detection and can be overridden per-command via `set_mode()`. mode no_truncaterNc2d|_|jy)NF)r%set_modeselfs r__init__zOutput.__init__:s  rc|tjk(r*trtjntj}||_|tjk7r t yy)zPOverride the output mode (called once at startup and again per '--format' flag).N)rrr rrr$r )r)r$s rr'zOutput.set_mode>sG <$$ $)1<%%9K9KD <%% % ! # &rc||_y)z2Toggle off cell truncation for human table output.Nr%)r)r%s rset_no_truncatezOutput.set_no_truncateFs &rc<|jtjk(SN)r$rrr(s ris_quietzOutput.is_quietJsyyL....r)rrmsgrrc||| td|}t|}|jxtjk(r| t |yytj k(r| t |yyy)z)Print a free-form text message to stdout.Nz&Cannot mix 'msg' with 'human'/'agent'.) ValueError _strip_ansir$rrr!r)r)r2rrs rtextz Output.textMss ? E$5 !IJJE$Eii###$ '%##$ '%$r)headersid_key alignmentsitemsr7r8r9c |sc|jxxtjk(rnxtjk(rnnt dytj k(r t dyy|At |dj}|Dcgc]tfd|Ds}}|Dcgc]!}|Dcgc]}|j|c}#} }}|jxtjk(r|Dcgc] }t|} }| D cgc]} | D cgc] } t| c} } } } t| | |j}it|| |xsi}|jD cic]\}} t|| }}} t t!t#d| | ||r|j%d yyxtjk(rDt d j'|| D]#} t d j'd | D%yxtj k(r/t t j(t |t* ytj,k(r,|xs|d}|D]}t |j|d yycc}wcc}wcc}}wcc}wcc} wcc} } wcc} }w)aPrint tabular data to stdout. Args: items: List of dicts. Headers are auto-detected from keys if not provided. headers: Explicit column names. If None, derived from dict keys (all-None columns filtered). id_key: Key to print in quiet mode. If None, uses the first header. alignments: Optional mapping of header name to "left" or "right". Defaults to "left". zNo results found.z[]Nrc3DK|]}|jduywr0)get).0itemcols r zOutput.table..xs8eW[#d9R8es r-zlist[list[str | int]])r7r9z>Use `--no-truncate` or `--format json` to display full values. c32K|]}t|ywr0)_format_table_cell_agentr>vs rrAzOutput.table..s*T1+CA+F*Tdefault)r$rrrr!rlistkeysanyr= _to_header_format_table_value_human_truncate_columnsr%_infer_alignmentsr:r rhintjoindumpsstrr)r)r:r7r8r9 all_columnsr@r?hrowsscreaming_headersrowrFformatted_rows is_truncatedinferredkscreaming_alignments quiet_keys ` rtablez Output.table^sV ))<'\''*<,*<*<< !45 "&& &  ?uQx}}/K&1fsS8e_d8e5esfGf;@A4g.!.AAii###U:CSQSUEM^^EU'VTQ 1 q(8'V$'V4nE 1#7 II^_ ###TYYw/0VC *TPS*T!TUV"""TZZU SAB##"0gaj !:D )R!89:$7g.A%E3^2o (WsH J$J/ J'8J" J';J- J7J2 /J76J="J'2J7)r8datacftj|rt|ts t |}|j t jk(r|t|j|dy|j t jk(rdnd}ttj||ty)zPrint structured data as JSON in all modes (indented for human, compact otherwise). Accepts a dict or a dataclass. NrJ)indentrI) dataclasses is_dataclass isinstancetype_dataclass_to_dictr$rrr!r=rrrTrU)r)rbr8res rdictz Output.dicts  # #D )*T42H%d+D 99 ** *v/A &"- . ii<#5#554TZZVSABrmessagec |jxtjk(rgtjd|g}|j D]\}}| |j d|d|!tdj|yxtjk(rO|j Dcgc]\}}| |d|}}}t|rdj|y|yxtjk(r/t|r tj|tyd ytjk(r+t|j}|rt|d yyycc}}w) z"Print a success summary to stdout.u✓ Nz z:  = rHrJr)r$rrrgreenr:appendr!rSrrrTrUrrKr)r)rlrbpartsr^rFrs rresultz Output.resultsii###d7)$456 JJL4DAq} r!Bqc]34TYYu-.###04 N1 A3asNNSXXe_C7C"""dTZZc:KK##dkkm, +$ Os - E(8 E(Fz--yes)rIyes confirm_paramrIrurvc|ry|jtjk7rt|d|dt j ||dy)zV Ask for confirmation. Raises `ConfirmationError` in non-human modes. Nz Use z to skip confirmation.T)rIabort)r$rrrtyperconfirm)r)rlrIrurvs rrzzOutput.confirmsE   99 ** *#wiu]OCY$Z[ [ gwd;rczt|jtjk(}||j ||S)zEReturn a status line that emits only in human mode (no-op otherwise).)enabled)r r$rrupdate)r)rlstatuss rr~z Output.statuss2DII1C1C$CD   MM' " rc|jtjk(r2tt j d|t jytd|t jy)z0Print a non-fatal warning to stderr (all modes).z Warning: fileN)r$rrr!ryellowsysstderrr)rls rwarningzOutput.warningsG 99 ** * y %:;#** M 9WI.SZZ @rc|jtjk(r2tt j d|t jytd|t jy)z%Print an error to stderr (all modes).zError: rN)r$rrr!rredrrrs rerrorz Output.errorsG 99 ** * GG9"56SZZ H 77),3:: >rc|jtjk(ry|jtjk(r2t t j d|tjyt d|tjy)a:Print a helpful hint to stderr (human: gray, json/agent: plain text). Suppressed in quiet mode. Kept in json mode (like agent) since agents commonly run with ``--format json`` and the next-command hints are useful there; hints go to stderr so they never pollute the parsed stdout. NzHint: r) r$rrrr!rgrayrrrs rrRz Output.hints^ 99 ** *  99 ** * VG9#56SZZ H 6'+#** =r)rNr0)rrrrr__annotations__boolr*rr'r.r1rUr6rrkrrKrartrzr r~rrrRrrrr#r#0s -9,=,=$\$$$'4'D'/$/($\`(d (S4Z(sUYz(ei(*%)!,0 8:S#X'8:cT! 8: d 8: cNT) 8: 8:t7; C Ct Ct C,c,3,4,&8=%fmC >D >rr#rFcBt|tjr|jSt|tr0|j Dcic]\}}| |t |c}}St|t r|Dcgc] }t |c}S|Scc}}wcc}w)z4Recursively serialize a value to be JSON-compatible.)rhdatetime isoformatrkr:_serialize_valuerK)rFkeyvalr?s rrrs!X&&'{{} At ;<779XxsC%c**XX At 3454 &55 HY5s BB?Binfoctj|jDcic]\}}| |t|c}}Scc}}w)z0Convert a dataclass to a json-serializable dict.)rfasdictr:r)rr^rFs rrjrjs?/:/A/A$/G/M/M/O atq!STS`A" " aa as AAz\033\[[0-9;]*mr6c.tjd|S)NrJ)_ANSI_REsubr6s rr5r5s <<D !!rc@dj|jS)Nrp)rSsplitrs r _single_liners 88DJJL !!rnamecPtjdd|}|jS)zAConvert a camelCase or PascalCase string to SCREAMING_SNAKE_CASE.z([a-z])([A-Z])z\1_\2)rerupper)rss rrNrNs!  (D1A 779rr7rXcfi}t|D]\}tfd|Dsd||<!|S)zNReturn ``{"col": "right"}`` for columns where every non-None value is numeric.c3K|]<}|duxs/t|ttfxrt|t >ywr0)rhintfloatr)r>rZcs rrAz$_infer_alignments..sHuils1v~d*SVc5\"B"c:VYZ[V\^bKcGcdusAAright) enumerateall)r7rXrtrWrs @rrQrQ s?F'" 1 uptu uF1I  Mrvaluec |yt|tr|rdSdSt|tjr|jdSt|trt j d|r|ddSt|tr t|St|trdjd|DSt|tr9d |vrtt |d Sttj|Stt |S) z/Convert a value to string for terminal display.NrJu✔z%Y-%m-%dz^\d{4}-\d{2}-\d{2}T z, c32K|]}t|ywr0)rOrEs rrAz,_format_table_value_human.. sE!215ErGr) rhrrstrftimerUrmatchrrKrSrkrrTrs rrOrOs }%u%2%%**+~~j))%"((+A5"ISbz%E""%yyEuEEE E4 U?E&M 23 3DJJu-.. E ##rr%c  |ssyt|}t|Dcgc]6tt|gfdttD8}}tjj td|dz z }t ||kry|j |Dcgc]}tt|dc} t |kDrEt fdt D fdd }|dkrn |xxdzcc<t |kDrEd} D]:} t| D]*\} t|  kDsd } | d d z d z| <,<| Scc}wcc}w)aTruncate cells in-place to fit the current terminal width. Returns `True` if any cell was truncated, so the caller can emit a hint. `shutil.get_terminal_size` is cross-platform: it honors `$COLUMNS`, then queries the OS-native API, then falls back to `(80, 24)`. Fc3@K|]}t|ywr0)len)r>rrrXs rrAz$_truncate_columns..9s%P!c$q'!*o%Psrc3:K|]\}}||kDs|ywr0r)r>iw min_widthss rrAz$_truncate_columns..Gs @41aa*Q-.?Q @sc|Sr0r)rcapss rz#_truncate_columns..Hs $q'r)rrITNz...) rrangemaxshutilget_terminal_sizecolumnssumcopyr)r7rXr%nrnaturalbudgetrWwidest truncatedrZcellrrs ` ` @@rrPrP(sn$ G A[`ab[cdVWs3wqz?Q%PuSY?O%PQdGd % % ' / /#aQ- ?F 7|v <<>D*12Q#c!fa.2J d)f  @9T? @!  A:  V   d)f I5 ~ 5GAt4y47" mQ! ,u4A 55 9e3s ;E&1E+c~t|tjr|jStt |S)zHFormat a cell value for agent TSV output (ISO timestamps, tabs escaped).)rhrrrrUrs rrDrDXs.%**+  E ##r)+rrfrrrrrcollections.abcrenumrtypingrrryhuggingface_hub.errorsrhuggingface_hub.utilsrr r r r rUrr!r#objectrrkrjcompilerr5rrNrKrQrOrrPrDoutrrrrsv) $ 4]]3)#))) s>s>r  6 bSbT#s(^b 2::' ("c"c""s"s"SS tCyT#YDcN$S$S$*- #Y- tCy/- -  -`$C$C$ hr