jn1pdZddlZddlZddlmZddlmZddlZddlm Z ddl m Z m Z m Z mZmZmZmZmZmZddlmZGd d eejZGd d eejZd dhZeeejddfZdedzdedzdedzfdZedZe dgdej!ej"ddej#dfde deeej$dddfd eeej$d!d"d#fd$e d%e d&ed'eddfd(Z%e d)d*d+gej#dfde d,ed&ed'eddf d-Z&e d.gd/ddd0ej#dfde d1eeej$d2d3fdeedzej$d4d5fdeedzej$d6d7fd8ee'ej$d9d:d;fd&ed'eddfd<Z(e d=d>d?gddej#dfde d,edeedzej$d4d@fdeedzej$d6dAfd&ed'eddfdBZ)e dCdDdEgdd0ej#dfde d,ed=eedzej$dFdGfdHee'ej$dIdJdKfd&ed'eddfdLZ*e dMdNdOgdd0ej#dfde d,ed=eedzej$dFdPfdHee'ej$dIdJdKfd&ed'eddfdQZ+e dRdSgej#dfde d,edTeeejdUfd&ed'eddf dVZ,e dWdXdYgdd0ej#dfde d,ed=eedzej$dFdZfdHee'ej$dIdJdKfd&ed'eddfd[Z-e d\d]gej#dfde d,ed&ed'eddf d^Z.dS)_zYContains commands to interact with discussions and pull requests on the Hugging Face Hub.N)Path) Annotated) constants) AuthorOptLimitOpt RepoIdArgRepoType RepoTypeOptTokenOptapi_object_to_dict get_hf_api typer_factory)outc"eZdZdZdZdZdZdZdS)DiscussionStatusopenclosedmergeddraftallN)__name__ __module__ __qualname__rrrrrb/var/www/html/Carbon-Document/venv/lib/python3.11/site-packages/huggingface_hub/cli/discussions.pyrr's' D F F E CCCrrceZdZdZdZdZdS)DiscussionKindr discussion pull_requestN)rrrrr r!rrrrr/s CJ!LLLrrrrz&The discussion or pull request number.)helpminbody body_filereturnc||tjd|Gt|dkrtjS|dS|S)zFResolve body text from --body or --body-file (supports '-' for stdin).Nz'Cannot use both --body and --body-file.-zutf-8)encoding)typer BadParameterstrsysstdinread read_text)r$r%s r _read_bodyr1Esd I1 !JKKK y>>S 9>>## #""G"444 Krz0Manage 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_idstatusz-sz--statusz4Filter by status (open, closed, merged, draft, all).kindz-kz--kindz/Filter by kind (discussion, pull_request, all).authorlimit repo_typetokenct|}|tjkrd}n|tjkrd}nd}|tjkrd} n|j} g} |||| ||jD]K} |jtvr| j |jkr!| | t| |krnLd| D} tj | gdddd i dS) z-List discussions and pull requests on a repo.r:rrN)r4r7discussion_typediscussion_statusr9c,g|]}t|Sr)r ).0ds r z#discussion_list..s! 8 8 8q  " " 8 8 8r)numtitleis_pull_requestr5r7 created_atrCright)headersid_key alignments)rrrrrrvalueget_repo_discussions_CLIENT_SIDE_STATUSESr5appendlenrtable) r4r5r6r7r8r9r:api api_statusapi_discussion_type discussionsrAitemss rdiscussion_listrVSsDB 5 ! ! !C!&&& #* * *   ~!!!""jK  % %+$/ &   <0 0 0QX5M5M 1 {  u $ $ E % 9 8K 8 8 8EI UUU7# rinfoz'hf discussions info username/my-model 5z5hf discussions info username/my-model 5 --format jsonrCct|}||||j}tj|dS)z,Get info about a discussion or pull request.r<r4discussion_numr9N)rget_discussion_detailsrKrdictr4rCr9r:rQdetailss rdiscussion_infor_sQ 5 ! ! !C((/)G HWrcreate)z>|D)5 ? %Z^ % % %4CJ?$??*.??g??Z^YcYgmpqqqqqqrcommentzIhf discussions comment username/my-model 5 --body "Thanks for reporting!"z9hf discussions comment username/my-model 5 --body "LGTM!"z%The comment text (supports Markdown).z0Read the comment from a file. Use '-' for stdin.ct||}|tjdt|}|||||jt jd|d|||dS)z(Comment on a discussion or pull request.Nz)Either --body or --body-file is required.r<r4rZrhr9zCommented on # in rCrepo)r1r*r+rcomment_discussionrKrrf)r4rCr$r%r9r:rhrQs rdiscussion_commentros8y))G !LMMM 5 ! ! !C/  J22222'JJJJJJrclosez(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.ctjd|d|d|t|}|||d||jtjd|d ||| d S) z#Close a discussion or pull request.zClose # on ''?rqr<rr4rZ new_statusrhr9zClosed #rkrlNrconfirmrchange_discussion_statusrKrfr4rCrhrqr9r:rQs rdiscussion_closer|s:K/#//G///S9999 5 ! ! !C  / !J,#,,7,,#GDDDDDDrreopenz)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.ctjd|d|d|t|}|||d||jtjd|d ||| d S) z+Reopen a closed discussion or pull request.zReopen #rsrtrur<rrvz Reopened #rkrlNrxr{s rdiscussion_reopenr>s:K0300W000c:::: 5 ! ! !C  / !J.C..W..CgFFFFFFrrenamez9hf discussions rename username/my-model 5 "Updated title" new_titlezThe new title.ct|}|||||jtjd|d||||dS)z$Rename a discussion or pull request.r<)r4rZrr9z Renamed #rk)rCrmrDN)rrename_discussionrKrrf)r4rCrr9r:rQs rdiscussion_renamergsq& 5 ! ! !C/  J-3--G--3WIVVVVVVrmergez(hf discussions merge username/my-model 5zEhf discussions merge username/my-model 5 --comment "Merging, thanks!"z)An optional comment to post when merging.ctjd|d|d|t|}|||||jtjd|d||| d S) zMerge a pull request.zMerge #rsrtrur<rjzMerged #rkrlN)rryrmerge_pull_requestrKrfr{s rdiscussion_mergers:K/#//G///S9999 5 ! ! !C/  J,#,,7,,#GDDDDDDrdiffz'hf discussions diff username/my-model 5ct|}||||j}|jrt j|jdSt jddS)z Show the diff of a pull request.r<rYzNo diff available.N)rr[rKrrtextr]s rdiscussion_diffrst 5 ! ! !C((/)G |'  %&&&&&r)/__doc__enumr-pathlibrtypingrr*huggingface_hubr _cli_utilsrrr r r r r rr_outputrr,EnumrrrMintArgumentDiscussionNumArgr1discussions_clicommandrrmodelOptionrVr_boolrgror|rrrrrrrrs( `_  %%%%%%                      sDI"""""S$)""""7+EN 5 S4ZD4KC$J -%WXXX$  %^+<< <   G     <   B     <$ %<& '<()<* +<, -<<<<~ 1?&^                0   %^A.r.r .r   ?     .r  d  7     .r  t  G    !.r.   A    /.r>?.r@ A.rB C.r.r.r.rb SC  %^%KK K K  d  8     K t  C    K"#K$ %K& 'KKKKD 2S  %^'EE E E d  <    E    ,     E$%E& 'E( )EEEED 3d  %^'GG G G d  >    G    ,     G$%G& 'G( )GGGGD C&^WW W W !    WW W WWW  W. 2O  %^'EE E E d  <    E    ,     E$%E& 'E( )EEEEB 1&^ '' ' ''  '  '''  '''r