o 2jU3@sdZddlZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z m Z ddlZddlmZddlmZmZmZmZmZGdd d ee ZGd d d Zd ed efddZde d eee ffddZedZded efddZ ded efddZ!ded efddZ"de#ede#e#e d eeeffddZ$d e d efd!d"Z%de#ede#e#ed#e&d e&fd$d%Z'd e d efd&d'Z(eZ)dS)(z"Output framework for the `hf` CLI.N)Sequence)Enum)Anycast)ConfirmationError)ANSI StatusLinedisable_progress_barsis_agenttabulatec@s$eZdZdZdZdZdZdZdZdS) OutputFormatzGOutput format for CLI commands with auto detection of agent/human mode.agentautohumanjsonquietN) __name__ __module__ __qualname____doc__r rrrrrrV/var/www/html/whisper/venv/lib/python3.10/site-packages/huggingface_hub/cli/_output.pyr sr c @seZdZUdZeed<eed<d2ddZejfdeddfdd Z deddfd d Z defd d Z d3dddde dBde dBde dBddfddZ dddddeee efdee dBde dBdee e fdBddf ddZdddede dBddfddZde deddfd d!Zd"d"d#d$de d%ed&ed'e ddf d(d)Zd3de dBdefd*d+Zde ddfd,d-Zde ddfd.d/Zde ddfd0d1ZdS)4OutputzOutput 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_truncatereturnNcCsd|_|dS)NF)rset_modeselfrrr__init__4s zOutput.__init__cCs:|tjkrtr tjntj}||_|tjkrtdSdS)zPOverride the output mode (called once at startup and again per '--format' flag).N)r rr r rrr )rrrrrr8s   zOutput.set_modecCs ||_dS)z2Toggle off cell truncation for human table output.Nr)rrrrrset_no_truncate@s zOutput.set_no_truncatecCs |jtjkSN)rr rrrrris_quietD zOutput.is_quiet)rr msgrr cCsx|dur|dus |durtd|}t|}|jtjkr*|dur(t|dSdStjkr:|dur8t|dSdSdS)z)Print a free-form text message to stdout.Nz&Cannot mix 'msg' with 'human'/'agent'.) ValueError _strip_ansirr rprintr )rr%rr rrrtextGs   z Output.text)headersid_key alignmentsitemsr*r+r,css'|jtjkr ntjkrnntddStjkr%tddSdSdurrsz*Output.table...)anyr1)r-r3r r&z Output.table..csg|] fddDqS)csg|]}|qSrr.r1hr2rrr8ss+Output.table...rr7)r*r<rr8sscSg|]}t|qSr _to_headerr:rrrr8wcSsg|] }dd|DqS)cSr>r_format_table_value_humanr1vrrrr8xrAr=rr1rowrrrr8xr cSsi|] \}}t||qSrr?r1krErrr }rHz Output.table..zlist[list[str | int]])r*r,z>Use `--no-truncate` or `--format json` to display full values. cs|]}t|VqdSr")_format_table_cell_agentrDrrrr5zOutput.table..default)rr r rr(rlistkeys_truncate_columnsr_infer_alignmentsr-r rhintjoindumpsstrrr/)rr-r*r+r, all_columnsrowsscreaming_headersformatted_rows is_truncatedinferredscreaming_alignmentsrG quiet_keyr2r)r*r-rtableXsT"    z Output.table)r+datacCspt|rt|tst|}|jtjkr"|dur"t| |ddS|jtj kr*dnd}tt j ||t ddS)zPrint structured data as JSON in all modes (indented for human, compact otherwise). Accepts a dict or a dataclass. NrR)indentrQ) dataclasses is_dataclass isinstancetype_dataclass_to_dictrr rr(r/rrrYrZ)rrdr+rfrrrdictsz Output.dictmessagecKs|jtjkr2td|g}|D]\}}|dur(|d|d|qtd|dStj krQdd|D}t|rLd|dS|dStj krit|rdt j |t d dSd dStj krt|}|r}t|d dSdSdS) z"Print a success summary to stdout.u✓ Nz z:  cSs&g|]\}}|dur|d|qS)N=rrIrrrr8r9z!Output.result.. rPrRr)rr rrgreenr-appendr(rXr rrYrZrrSvalues)rrmrdpartsrJrErsrrrresults&    $ z Output.resultFz--yes)rQyes confirm_paramrQrvrwcCs<|rdS|jtjkrt|d|dtj||dddS)zV Ask for confirmation. Raises `ConfirmationError` in non-human modes. Nz Use z to skip confirmation.T)rQabort)rr rrtyperconfirm)rrmrQrvrwrrrrzs  zOutput.confirmcCs(t|jtjkd}|dur|||S)zEReturn a status line that emits only in human mode (no-op otherwise).)enabledN)rrr rupdate)rrmstatusrrrr}s z Output.statuscCB|jtjkrttd|tjddStd|tjddS)z0Print a non-fatal warning to stderr (all modes).z Warning: fileN)rr rr(ryellowsysstderrrrmrrrwarning zOutput.warningcCr~)z%Print an error to stderr (all modes).zError: rN)rr rr(rredrrrrrrerrorrz Output.errorcCsR|jtjkrdS|jtjkrttd|tjddStd|tjddS)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) rr rrr(rgrayrrrrrrrWs  z Output.hint)rNr")rrrrr __annotations__boolrrrr!r#rZr)rrlrrSrcrurzrr}rrrWrrrrr*s<  0  : ( rrErcCsLt|tjr |St|trdd|DSt|tr$dd|DS|S)z4Recursively serialize a value to be JSON-compatible.cS"i|] \}}|dur|t|qSr"_serialize_value)r1keyvalrrrrK"z$_serialize_value..cSr>rrr0rrrr8rAz$_serialize_value..)ridatetime isoformatrlr-rS)rErrrrs   rinfocCsddt|DS)z0Convert a dataclass to a json-serializable dict.cSrr"rrIrrrrKrz&_dataclass_to_dict..)rgasdictr-)rrrrrksrkz\033\[[0-9;]*mr)cCs td|S)NrR)_ANSI_REsubr)rrrr'r$r'cCsd|S)Nrp)rXsplitrrrr _single_linesrnamecCstdd|}|S)zAConvert a camelCase or PascalCase string to SCREAMING_SNAKE_CASE.z([a-z])([A-Z])z\1_\2)rerupper)rsrrrr@sr@r*r\cs8i}t|D]\}tfdd|Drd||<q|S)zNReturn ``{"col": "right"}`` for columns where every non-None value is numeric.c3s>|]}|dupt|ttfot|t VqdSr")riintfloatrrFcrrr5s<z$_infer_alignments..right) enumerateall)r*r\rur;rrrrVs rVvaluecCs|durdSt|tr|rdSdSt|tjr|dSt|tr-td|r-|ddSt|tr6t|St|trEd dd |DSt|t r]d |vrVtt|d Stt |Stt|S) z/Convert a value to string for terminal display.NrRu✔z%Y-%m-%dz^\d{4}-\d{2}-\d{2}T z, csrMr"rBrDrrrr5rOz,_format_table_value_human..r) rirrstrftimerZrmatchrrSrXrlrrYrrrrrC s"         rCrc s|ssdSt}fddt|D}tjtd|d}t||kr*dS|ddDt|krbtfddtDfd d d d }|dkrTn|d8<t|ks;d}D]#}t|D]\} } t| | krd }| d| dd|| <qlqf|S)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)`. Fcs<g|]ttgfddttDRqS)c3s |] }t|VqdSr")len)r1r)rr\rrr53sz/_truncate_columns...)maxrranger7)r*r\rrr83s<z%_truncate_columns..rcSsg|] }tt|dqS))rrr:rrrr8>rHc3s$|] \}}||kr|VqdSr"r)r1iw) min_widthsrrr5As"z$_truncate_columns..cs|Sr"r)r)capsrrBsz#_truncate_columns..)rrQTNz...) rrshutilget_terminal_sizecolumnsrsumcopyr) r*r\rnnaturalbudgetwidest truncatedrGrcellr)rr*rr\rrU"s8      rUcCs t|tjr |Stt|S)zHFormat a cell value for agent TSV output (ISO timestamps, tabs escaped).)rirrrrZrrrrrNRs  rN)*rrgrrrrrcollections.abcrenumrtypingrrryhuggingface_hub.errorsrhuggingface_hub.utilsrrr r r rZr robjectrrlrkcompilerr'rr@rSrVrCrrUrNoutrrrrsF     : *   0