hoPdZddlZddlZddlZddlmZmZddlmZddl m Z ddl m Z m Z mZmZddlZddlmZmZddlmZdd lmZd d lmZej4eZeDcgc]}|j:d k7s|j: c}ZGd deZGddeZ GddeZ!de e ee"e#fde e"de"fdZ$GddeZ%GddeZ&GddeZ'GddeZ(ycc}w)aContains commands to interact with jobs on the Hugging Face Hub. Usage: # run a job hf jobs run # List running or completed jobs hf jobs ps [-a] [-f key=value] [--format TEMPLATE] # Stream logs from a job hf jobs logs # Inspect detailed information about a job hf jobs inspect # Cancel a running job hf jobs cancel N) Namespace_SubParsersAction)asdict)Path)DictListOptionalUnion)HfApi SpaceHardware)logging) load_dotenv)BaseHuggingfaceCLICommandz zero-a10gc"eZdZedefdZy) JobsCommandsparsercv|jddjd}jfdtj |t j |t j |tj |tj |tj |y)NjobszRun and manage Jobs on the Hub.helpz$huggingface.co jobs related commandsc$jSN) print_help)args jobs_parsers T/var/www/Befach/backend/env/lib/python3.12/site-packages/huggingface_hub/cli/jobs.pyz2JobsCommands.register_subcommand..>s;3I3I3Kfunc) add_parseradd_subparsers set_defaultsInspectCommandregister_subcommand LogsCommand PsCommand RunCommand CancelCommand UvCommand)rjobs_subparsersrs @rr&z JobsCommands.register_subcommand8s''5V'W %44:`4a   &K L **?;''8%%o6&&7))/:%%o6rN)__name__ __module__ __qualname__ staticmethodrr&rrrr7s 7$5 7 7rrc>eZdZededdfdZdeddfdZddZy) r)rreturnNcz|jdd}|jdtd|jddd d |jd d d d |jdtd|jdtd|jdtddjtd|jdtd|jdddd |jdtd|jdtd |jd!d"d#$|j t %y)&Nrunz Run a JobrimagezThe Docker image to use.typer-e--envappendzSet environment variables.actionr-s --secretsz!Set secret environment variables. --env-file(Read in a file of environment variables.--secrets-file/Read in a file of secret environment variables.--flavorTFlavor for the hardware, as in HF Spaces. Defaults to `cpu-basic`. Possible values: , . --timeoutzVMax duration: int/float with s (seconds, default), m (minutes), h (hours) or d (days).-d--detach store_truez3Run the Job in the background and print the Job ID. --namespaceVThe namespace where the Job will be created. Defaults to the current user's namespace.--tokenIA User Access Token generated from https://huggingface.co/settings/tokenscommand...zThe command to run.nargsrr )r" add_argumentstrjoinSUGGESTED_FLAVORSr$r)r run_parsers rr&zRunCommand.register_subcommandJsu&&u;&? c8RSghEabk(Ilm 3=gh 0sArs ghlhqhqsDiEhFFGH   i    F   i   \   =RSZ0rrc6|j|_|j|_i|_|jrE|jj t t |jj|jxsgD]&}|jj t |(i|_|jrE|jj t t |jj|jxsgD]&}|jj t |(|j|_ |j|_ |j|_ |j|_ |j|_yr)r6rPenvenv_fileupdaterr read_textsecrets secrets_fileflavortimeoutdetach namespacetokenselfr env_valuesecrets r__init__zRunCommand.__init__os** "&,, -/ == HHOOKT]](;(E(E(GH IRI HHOOK 2 3(13    LL   D1B1B,C,M,M,O P Qll(b(F LL   F 3 4)/3{{ &*ll  KK (,$(JJ rc t|j}|j|j|j|j |j |j|j|j}td|jtd|j|jry|j|jD] }t|y)Nre)r6rPr[r_rarbrdJob started with ID: View at: job_id)r rerun_jobr6rPr[r_rarbrdprintidurlrcfetch_job_logsrgapijoblogs rr5zRunCommand.runs$**%kk**LLLL;;LLnn  %cffX./ #''#$ ;; %%SVV%4C #J5rr3N r-r.r/r0rr&rrjr5r1rrr)r)Is<"1$5"1$"1"1H/Y/4/&rr)c>eZdZededdfdZdeddfdZddZy) r'rr3Nc|jdd}|jdtd|jdtd|jd td |jt y) NlogszFetch the logs of a JobrrpJob IDr7rLQThe namespace where the job is running. Defaults to the current user's namespace.rNrOr )r"rTrUr$r'rXs rr&zLogsCommand.register_subcommandsy&&v4M&N sB d   C&q   [1rrcj|j|_|j|_|j|_yrrprdrergrs rrjzLogsCommand.__init__s#;; (,$(JJ rct|j}|j|j|jD] }t |yNrlrprd)r rerurprdrr)rgrwrys rr5zLogsCommand.runs:$**%%%T[[DNN%SC #JTrrzr{r1rrr'r's; 2$5 2$ 2 2/Y/4/ rr'rowsheadersr3c tg||Dcgc]}td|D}}ttjjt |dz}t |t |z|kDr|jt|}||xxdzcc<t |t |z|kr#|t |z t |z ||z||<t |t |z|kDrdt |zj|}g}|j|j||j|j|Dcgc]}d|z c}|D]r} t| |D cgc];\} } t t| | kDrt| d| dz dzn t| =} } } |j|j| td j|Scc}wcc}wcc} } w) z Inspired by: - stackoverflow.com/a/8356620/593036 - stackoverflow.com/questions/9535954/printing-lists-as-tabular-data c3DK|]}tt|ywr)lenrU).0xs r z_tabulate..s/ac#a&k/s  z{{:{}}} -NrQ ) zipmaxosget_terminal_sizecolumnsrsumindexformatr;rUrV) rrcol col_widthsterminal_widthcol_to_minimize row_formatlineswrowr col_widthrow_format_argss r _tabulaters 9$**3z?;?#)# w<#j/ )^ ;*83z?*JSQX\*Y\fgv\w*wJ ' g,Z (> 9 4*s7|+33Z@J E LL"""G,- LL"""j$AS1W$ABC!$C 4 9033q6{Y/FCF?Y] #e +CPQF R    &Z&&89  99U #P%B sG* G/ 6AG4c^eZdZededdfdZdeddfdZd dZde e e fde fd Z d Z y) r(rr3Nc<|jdd}|jdddd|jd td |jd td |jdddgd|jdtd |jty)Npsz List Jobsrz-az--allrKz*Show all Jobs (default shows just running)r<rLzUThe namespace from where it lists the jobs. Defaults to the current user's namespace.r7rNrOz-fz--filterr;z>Filter output based on conditions provided (format: key=value))r=defaultrz--formatz%Format output using a custom templater )r"rTrUr$r(rXs rr&zPsCommand.register_subcommands&&t+&>   =   h   \    Q    8  Y/rrcH|j|_|j|_|j|_|j|_i|_|j D]H}d|vr3|j dd\}}||j|j<:td|dJy)N=rz)Warning: Ignoring invalid filter format 'z'. Use key=value format.) allrdrerfiltersfiltersplitlowerrr)rgrfkeyvalues rrjzPsCommand.__init__s(,$(JJ %)[[ ') AaxWWS!_ U,1 SYY[)A!D\]^ rc  t|j}|j|j}gd}g}|D]}|jr|jj nd}|j s|dvr8|j}|jxsd}|jxsg} | rdj| nd} |jxsd} |||j| d} |j| s|j||| | |g|sad } |jrDd d j|jj!Dcgc] \}}|d |c}}} t#d | y|j%||ycc}}w#t&j($r}t#d|Yd}~yd}~wt*t,t.f$r}t#d|Yd}~yd}~wt0$r.}t#dt3|j4d|Yd}~yd}~wwxYw)z Fetch and display job information for the current user. Uses Docker-style filtering with -f/--filter flag and key=value pairs. rl)rd)zJOB IDz IMAGE/SPACECOMMANDCREATEDSTATUSUNKNOWN)RUNNINGUPDATINGzN/A )rsr6statusrPz matching filters: rFrz No jobs foundNzError fetching jobs data: zError processing jobs data: zUnexpected error - z: )r re list_jobsrdrstagerrs docker_imagerPrV created_atr_matches_filtersr;ritemsrr _print_outputrequestsRequestExceptionKeyError ValueError TypeError Exceptionr8r-)rgrwr table_headersrrxrrpimage_or_spacerP command_strrjob_properties filters_msgkves rr5z PsCommand.runs A Adjj)C==4>>=:DVMD-0ZZ))YxxF2I$I"%!1!1!:U+++3:chhw/ !^^4u !+$lln* ",,^< V^[*fUVEJ <<$7 Z^ZfZfZlZlZnBoRVRSUVaS!:Bo8p7q"rK k]34   t] 3 Cp(( 4 .qc2 3 3*i0 6 04 5 5 A 'Q(8(8'9A3? @ @ AsHD;F=E: F'F:FH F&&H =G H $HH rcP|jjD]\}}||vryd|vsd|vrQ|jddjdd}tjd|d||tj rcy|j ||j vsyy) z+Check if job matches all specified filters.F*?z.*rG^$T)rrreplaceresearch IGNORECASEr)rgrrpattern regex_patterns rrzPsCommand._matches_filtersIs LL..0LC.(g~ 'T : B B3 L yy1]O1!5~c7JBMMZ s(;(A(A(CC1rc |jr_|j}|D]M}|}tgdD].\}}d|d}||vs|j|t||}0t |Oyt t ||y)z,Print output according to the chosen format.)rsr6rPcreatedrz{{.z}})rN)r enumeraterrUrrr) rgrrtemplaterlineifield placeholders rrzPsCommand._print_output\s ;;{{H )*Y ZHAu$)%"5K"d*#||KSVE![d  # rrz)r-r.r/r0rr&rrjr5rrUboolrrr1rrr(r(sb 0$5 0$ 0 0D _Y _4 _FAPtCH~$&rr(c>eZdZededdfdZdeddfdZddZy) r%rr3Nc|jdd}|jdtd|jdtd|jd d d |jt y)Ninspectz0Display detailed information on one or more JobsrrLrr7rNrOjob_idsrQzThe jobs to inspectrRr )r"rTrUr$r%rXs rr&z"InspectCommand.register_subcommandssz&&y7i&j  d   C&q    =RS^4rrcj|j|_|j|_|j|_yr)rdrerrs rrjzInspectCommand.__init__s#(,$(JJ "&,, rc $t|j}|jDcgc]}|j||j!}}t t j|Dcgc] }t|c}dtycc}wcc}w)Nrlr)indentr) r rer inspect_jobrdrrjsondumpsrrU)rgrwrprrxs rr5zInspectCommand.runsg$**%W[WcWcdVvHdd djj6#&+6q#NOe6s $B#B rzr{r1rrr%r%rs< 5$5 5$ 5 5/Y/4/ Prr%c>eZdZededdfdZdeddfdZddZy) r*rr3Nc|jdd}|jdtd|jdtd|jd td |jt y) Ncancelz Cancel a Jobrrprr7rLrrNrOr )r"rTrUr$r*rXs rr&z!CancelCommand.register_subcommandsx&&xn&E sB d   C&q   ]3rrcj|j|_|j|_|j|_yrrrs rrjzCancelCommand.__init__s#;; $(JJ rc~t|j}|j|j|jyr)r re cancel_jobrprd)rgrws rr5zCancelCommand.runs($**% dkkT^^Drrzr{r1rrr*r*s< 4$5 4$ 4 4/Y/4/ Err*c8eZdZdZedZdeddfdZddZy) r+z.Run UV scripts on Hugging Face infrastructure.cr|jdd}|jddd}|jdd }|jd d |jd d dg|jdtd|jdd|jdtddj t d|jdddd|jdddd |jd!td"|jd#td$|jd%td&|jd'd(d)d*|jd+td,|jd-td.|jd/dd0d12|jd3d4td5|j t6y7)8zRegister UV run subcommand.uvzERun UV scripts (Python with inline dependencies) on HF infrastructurer uv_commandz UV commandsT)destrrequiredr5z8Run a UV script (local file or URL) on HF infrastructurescriptz$UV script to run (local file or URL) script_argsrQzArguments for the script)rSrrz--imagez.Use a custom Docker image with `uv` installed.r7z--repozCRepository name for the script (creates ephemeral if not specified)rDrErFrGr9r:r;zEnvironment variablesr<r>r?zSecret environment variablesr@rArBrCrHz Max duration (e.g., 30s, 5m, 1h)rIrJrKzRun in backgroundrLrMrNzHF tokenz--withz%Run with the given packages installedwith_)r=rrz-pz--pythonz5The Python interpreter to use for the run environmentr N)r"r#rTrUrVrWr$r+)r uv_parser subparsersrYs rr&zUvCommand.register_subcommands %% X& --rs& 1..0)5'   H %,9WDTZZ;=VTZZW7,7$M*M`+4DeCHo./$s)6a)aHP.P4E-E2h)hc Xs #C/8C/