-:jE1@dZddlZddlZddlmZddlmZddlZddlm Z ddl m Z m Z m Z mZmZmZmZmZddlmZmZGd d eej.ZGd d eej.Zd dhZeeej8ddfZdedzdedzdedzfdZedZejAdgdejBejDddejFdfde deeejHdddfd eeejHd!d"d#fd$e d%e d&ed'eddfd(Z%ejAd)d*d+gejFdfde d,ed&ed'eddf d-Z&ejAd.gd/ddd0ejFdfde d1eeejHd2d3fdeedzejHd4d5fdeedzejHd6d7fd8ee'ejHd9d:d;fd&ed'eddfd<Z(ejAd=d>d?gddejFdfde d,edeedzejHd4d@fdeedzejHd6dAfd&ed'eddfdBZ)ejAdCdDdEgdd0ejFdfde d,ed=eedzejHdFdGfdHee'ejHdIdJdKfd&ed'eddfdLZ*ejAdMdNdOgdd0ejFdfde d,ed=eedzejHdFdPfdHee'ejHdIdJdKfd&ed'eddfdQZ+ejAdRdSgejFdfde d,edTeeej8dUfd&ed'eddf dVZ,ejAdWdXdYgdd0ejFdfde d,ed=eedzejHdFdZfdHee'ejHdIdJdKfd&ed'eddfd[Z-ejAd\d]gejFdfde d,ed&ed'eddf d^Z.y)_zYContains commands to interact with discussions and pull requests on the Hugging Face Hub.N)Path) Annotated) constants) AuthorOptLimitOpt RepoIdArgRepoType RepoTypeOptTokenOpt get_hf_api typer_factory)_dataclass_to_dictoutc eZdZdZdZdZdZdZy)DiscussionStatusopenclosedmergeddraftallN)__name__ __module__ __qualname__rrrrra/var/www/html/tokenscope/api/venv/lib/python3.12/site-packages/huggingface_hub/cli/discussions.pyrr&s D F F E CrrceZdZdZdZdZy)DiscussionKindr discussion pull_requestN)rrrrr r!rrrrr.s CJ!Lrrrrz&The discussion or pull request number.)helpminbody body_filereturnc||tjd|>t|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)r$r%s r _read_bodyr1Ds[ I1  !JKK y>S 99>># #""G"44 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|}|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| } } tj| gddycc} w) z-List discussions and pull requests on a repo.r:rrN)r4r7discussion_typediscussion_statusr9)numtitleis_pull_requestr5r7 created_atr?)headersid_key)r rrrrrvalueget_repo_discussions_CLIENT_SIDE_STATUSESr5appendlenrrtable) r4r5r6r7r8r9r:api api_statusapi_discussion_type discussionsditemss rdiscussion_listrQRs B 5 !C!&&& #** *   ~!!!""jjK  % %+$// &  <<0 0QXX5M 1 { u $  -8 8q  " 8E 8II U 9sDinfoz'hf discussions info username/my-model 5z5hf discussions info username/my-model 5 --format jsonr?ct|}|j|||j}tj|y)z,Get info about a discussion or pull request.r<r4discussion_numr9N)r get_discussion_detailsrErdictr4r?r9r:rKdetailss rdiscussion_inforZs> 5 !C((//)G HHWrcreate)z|D)5HZ^^$ %4CJJ$r*..!1gY?Z^^YcYgYgmpqrcommentzIhf 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|}|j ||||j t jd|d|||y)z(Comment on a discussion or pull request.Nz)Either --body or --body-file is required.r<r4rUrcr9zCommented on # in r?repo)r1r*r+r comment_discussionrErra)r4r?r$r%r9r:rcrKs rdiscussion_commentrjss8y)G  !LMM 5 !C//  JJuD 2'Jrclosez(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|}|j||d||jtj d|d ||| y ) z#Close a discussion or pull request.zClose # on ''?rlr<rr4rU new_statusrcr9zClosed #rfrgNrconfirmr change_discussion_statusrErar4r?rcrlr9r:rKs rdiscussion_closerwsp:KK'#eG9B/S9 5 !C  // !JJ#d7),#GDrreopenz)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|}|j||d||jtj d|d ||| y ) z+Reopen a closed discussion or pull request.zReopen #rnrorpr<rrqz Reopened #rfrgNrsrvs rdiscussion_reopenrz<sp:KK(3%uWIR0c: 5 !C  // !JJC5WI.CgFrrenamez9hf discussions rename username/my-model 5 "Updated title" new_titlezThe new title.ct|}|j||||jtjd|d||||y)z$Rename a discussion or pull request.r<)r4rUr|r9z Renamed #rf)r?rhr@N)r rename_discussionrErra)r4r?r|r9r:rKs rdiscussion_renameresR& 5 !C//  JJ3%tG9-3WIVrmergez(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|}|j||||jtj d|d||| y ) zMerge a pull request.zMerge #rnrorpr<rezMerged #rfrgN)rrtr merge_pull_requestrErarvs rdiscussion_mergersm:KK'#eG9B/S9 5 !C//  JJ#d7),#GDrdiffz'hf discussions diff username/my-model 5ct|}|j|||j}|jr t j |jyt j dy)z Show the diff of a pull request.r<rTzNo diff available.N)r rVrErrtextrXs rdiscussion_diffrsV 5 !C((//)G ||  %&r)/__doc__enumr-pathlibrtypingrr*huggingface_hubr _cli_utilsrrr r r r r r_outputrrr,EnumrrrGintArgumentDiscussionNumArgr1discussions_clicommandrrmodelOptionrQrZboolrbrjrwrzrrrrrrrs`  %   -sDII"S$))""7+ENN 5 S4ZD4KC$J %WX$  %^^+; ;   G   ;   B   ;$ %;& ';();* +;, -;;| 1?&^^             0   %^^A.r .r   ?   .r  d  7   .r  t  G  !.r.   A  /.r>?.r@ A.rB C.r.rb SC  %^^%K K K  d  8   K t  C  K"#K$ %K& 'KKD 2S  %^^'E E E d  <  E    ,   E$%E& 'E( )EED 3d  %^^'G G G d  >  G    ,   G$%G& 'G( )GGD C&^^W W W !  WW W W  W. 2O  %^^'E E E d  <  E    ,   E$%E& 'E( )EEB 1&^^ ' ' ''  '  '  'r