qiA dZddlZddlZddlZddlmZddlmZmZddl Z ddl m Z ddl m Z mZddlmZdd lmZmZmZmZmZmZmZmZmZmZmZmZmZmZGd d e ejBZ"Gd d e ejBZ#Gdde ejBZ$ddhZ%ee&e jNddfZ(de de fdZ)dee deedee fdZ*dpdede+ddfdZ,edZ-e-j]d gd!"e"j^e#j`dd#ejbejdddfd$edee"e jfd%d&d'fd(ee#e jfd)d*d+fd,ed-ed.ed/ed0ed1eddfd2Z4e-j]d3gd4"dddejbe$jjdfd$ed5e(d6ee+e jfd7d8fd9ee+e jfd:d;fdfd.ed/ee$e jfd?fd1eddfd@Z6e-j]dAgdB"dddejbdfd$edCee e jfdDdEfdeee e jfdFdGfdeeee jfdHdIfdJee+e jfdKdLdMfd.ed1eddfdNZ7e-j]dOdPdQg"ddejbdfd$ed5e(deee e jfdFdRfdeeee jfdHdSfd.ed1eddfdTZ8e-j]dUdVdWg"ddejbdfd$ed5e(dOeee e jfdXdYfdZee+e jfd[d\d]fd.ed1eddfd^Z9e-j]d_d`dag"ddejbdfd$ed5e(dOeee e jfdXdbfdZee+e jfd[d\d]fd.ed1eddfdcZ:e-j]dddeg"ejbdfd$ed5e(dfee e jNdgfd.ed1eddf dhZ;e-j]didjdkg"ddejbdfd$ed5e(dOeee e jfdXdlfdZee+e jfd[d\d]fd.ed1eddfdmZt|dk(rtjj S|j dS|S)zFResolve body text from --body or --body-file (supports '-' for stdin).z'Cannot use both --body and --body-file.-zutf-8)encoding)typer BadParameterstrsysstdinread read_text)r9r:s r% _read_bodyrE_s[ I1  !JKK y>S 99>># #""G"44 Kr$Fdetails show_commentsc p|jrdnd}ttj|jdtj d|j t|j|j|jjdg}|jr=|jr1|jdtj|jt|ddj||jrs|jrg|jd urttj d n:ttj d d j|jd}g}|j"D]6}t%|t&s|j(r!||}&|j|8|rG|j*j-r-tt|j*j-|r|rtttj d |D]u}tttj|jd|jjdt|j*j-wnI|rGtttj t/|dt/|dk7rdnddttdtj0|j2y)Nz Pull Request Discussion #z%Y-%m-%d %H:%Mzinto z: u · TzHas conflicting fileszConflicting files: z, ────────────────────────────────────────────────────────────z commentr sz (use --comments to show) View on Hub: )is_pull_requestprintr boldtitlegraynumr8r1author created_atstrftime target_branchappendjoinconflicting_filesr7events isinstancerhiddencontentstriplenr6url)rFrGkindpartsr9commentseventcomments r%_print_discussion_inforijse$44>,D TYYw}} % &a Agkk]2C(D'E FG GNN +W^^W=O=O=X=XYi=j kE7#8#8 uTYYw'<'<=>?@ TF"V[['( )*7#<#<  $ $ , $++56 7 $++ 3DIIg>W>W4X3YZ[ \ DH' e. / |& '  ""$  dll  "#  dii #$ +G G TYYw~~./tG4F4F4O4OP`4a3bc d '//'') * +   dii3x=/X!9K#QS1TTmnop G M$))GKK01 23r$z0Manage discussions and pull requests on the Hub.)r/z list | ls)z%hf discussions list username/my-modelzIhf discussions list username/my-model --kind pull_request --status mergedzFhf discussions list username/my-dataset --type dataset --status closedzBhf discussions list username/my-model --author alice --format json)examplesrepo_idz-sz--statusz4Filter by status (open, closed, merged, draft, all).rdz-kz--kindz/Filter by kind (discussion, pull_request, all).rVlimit repo_typeformatquiettokenc  t|} |tjk(rd} n|tjk(rd} nd} |tj k(rd} n |j } g} | j||| | |j D]O} |j tvr| j|j k7r/| j| t| |k\sOn| D cgc] } t| }} t|||dgdddd i ycc} w) z-List discussions and pull requests on a repo.rqrrN)rlrVdiscussion_typediscussion_statusrnrU)rUrSrPr1rVrWc "d|dt|jddd|jdrdndtt|jd dt|jd dt|jd dgS) NrKrUrSrN2)max_lenrPPRr1rVrW)rgetr8r@)items r%z!discussion_list..s{U }  '2. ;HH./DR 3txx"56 7 2& ' ,3 4  r$right)rorpid_keyheadersrow_fn alignments)rrrrr'rvalueget_repo_discussions_CLIENT_SIDE_STATUSESr1rZrbrr)rlr1rdrVrmrnrorprqapi api_statusapi_discussion_type discussionsditemss r%discussion_listrsF 5 !C!&&& #** *   ~!!!""jjK  % %+$// &  <<0 0QXX5M 1 { u $  -8 8q  " 8E 8 U 7# 9sDinfo)z'hf discussions info username/my-model 5z2hf discussions info username/my-model 5 --commentsz.hf discussions info username/my-model 5 --diffz5hf discussions info username/my-model 5 --format jsonrUrfz --commentszShow all comments.diffz--diffz"Show the diff (for pull requests).no_colorz --no-colorzDisable colored output.zOutput format (text or json).cddl}|rd|jd<t|} | j|||j} |t j k(r@t| } |s| jddtt j| d yt| | |rK| jr>tttjd t| jyyy) z,Get info about a discussion or pull request.rN1NO_COLORrsrldiscussion_numrnr)indent)rGrL)osenvironrget_discussion_detailsrr+r-rpoprQdumpsrirr rT) rlrUrfrrrnrorqrrrFresults r%discussion_inforsV!$ : 5 !C((//)G  #G, JJvt $ djj*+7(;    dii #$ glltr$create)z|D HTF!DII*..)9&:;r?rcomment_discussionrrQr rR)rlrUr9r:rnrqrhrs r%discussion_commentrosu8y)G  !LMM 5 !C//   N3%tDIIg$6#7 89r$closez(hf discussions close username/my-model 5zIhf discussions close username/my-model 5 --comment "Closing as resolved."z --commentz)An optional comment to post when closing.yesz--yesz-yzSkip confirmation prompt.c"|s=tjd|d|d}|stdtjt |}|j ||d||j td|d tj|y ) z#Close a discussion or pull request.zClose # on ''?Aborted.rsrrlr new_statusrhrnzClosed #rN r>confirmrQExitrchange_discussion_statusrr rRrlrUrhrrnrqrrs r%discussion_closers: --'#eG9B ?@ * **,  5 !C  // ! HSEdii01 23r$reopenz)hf discussions reopen username/my-model 5zZhf discussions reopen username/my-model 5 --comment "Reopening for further investigation."z+An optional comment to post when reopening.c"|s=tjd|d|d}|stdtjt |}|j ||d||j td|d tj|y ) z+Reopen a closed discussion or pull request.zReopen #rrrrsrrz Reopened #rNrrs r%discussion_reopenrs: --(3%uWIR @A * **,  5 !C  // ! Jse4 ' 23 45r$renamez9hf discussions rename username/my-model 5 "Updated title" new_titlezThe new title.c t|}|j||||jtd|dt j |dt j |y)z$Rename a discussion or pull request.rs)rlrrrnz Renamed #z to rN)rrename_discussionrrQr rR)rlrUrrnrqrs r%discussion_renamersb& 5 !C//   IcU$tyy34D79K8L MNr$mergez(hf discussions merge username/my-model 5zEhf discussions merge username/my-model 5 --comment "Merging, thanks!"z)An optional comment to post when merging.c |s=tjd|d|d}|stdtjt |}|j ||||j td|dtj|y ) zMerge a pull request.zMerge #rrrrsrzMerged #rN) r>rrQrrmerge_pull_requestrr rRrs r%discussion_mergers: --'#eG9B ?@ * **,  5 !C//   HSEdii01 23r$z'hf discussions diff username/my-model 5ct|}|j|||j}|jrt |jyt dy)z Show the diff of a pull request.rsrzNo diff available.N)rrrrrQ)rlrUrnrqrrFs r%discussion_diffr;sN 5 !C((//)G || gll "#r$)F)>r.enumr-rApathlibrtypingrrr>huggingface_hubrhuggingface_hub.communityrrhuggingface_hub.utilsr _cli_utilsr r r rrrrrrrrrrrr@Enumrr'r+rintArgumentDiscussionNumArgr8rEboolridiscussions_clicommandrrmodeltableOptionrr,rrrrrrrrr#r$r%rs` & %N&$sDII"S$))" dii"7+ENN 5  3 3 Xc]x~(3-(4$9(4$(4[_(4V %WX$  %^^$**/I I   G   I   B   I$ %I& 'I()I* +I, -I. /I0 1IIX $   %^^ ?: : :  %  :   5   :"  *  #:01:2  0   3:> ?:@ A::z 0   %^^A07 07   ?   07   7   07  G  !07.   A  /07>?07@ A07B C0707f SC  %^^%: : :   8   : C  :"#:$ %:& '::D 2S  %^^'#4 #4 #4  <  #4    ,   #4$%#4& '#4( )#4#4L 3d  %^^'#6 #6 #6  >  #6    ,   #6$%#6& '#6( )#6#6L C&^^O O O !  OO O O  O. 2O  %^^'"4 "4 "4  <  "4    ,   "4$%"4& '"4( )"4"4J 1&^^ $ $ $$  $  $  $r$