"j3 dZddlZddlZddlmZddlmZddlZddlm Z ddl m Z m Z m Z mZmZmZmZmZmZmZddlmZmZGd d eejZGd d eejZd dhZeeejddfZdedzdedzdedzfdZ edZ!e!"dgdej#ej$ddej%ej&dfdedeeej'dddfd eeej'd!d"d#fd$e d%e d&ed'e d(eddfd)Z(e!"d*d+d,gej%ej&dfded-ed&ed'e d(eddf d.Z)e!"d/gd0ddd1ej%ej&dfded2eeej'd3d4fdeedzej'd5d6fdeedzej'd7d8fd9ee*ej'd:d;d<fd&ed'e d(eddfd=Z+e!"d>d?d@gddej%ej&dfded-edeedzej'd5dAfdeedzej'd7dBfd&ed'e d(eddfdCZ,e!"dDdEdFgdd1ej%ej&dfded-ed>eedzej'dGdHfdIee*ej'dJdKdLfd&ed'e d(eddfdMZ-e!"dNdOdPgdd1ej%ej&dfded-ed>eedzej'dGdQfdIee*ej'dJdKdLfd&ed'e d(eddfdRZ.e!"dSdTgej%ej&dfded-edUeeejdVfd&ed'e d(eddfdWZ/e!"dXdYdZgdd1ej%ej&dfded-ed>eedzej'dGd[fdIee*ej'dJdKdLfd&ed'e d(eddfd\Z0e!"d]d^gej%ej&dfded-ed&ed'e d(eddf d_Z1dS)`zYContains commands to interact with discussions and pull requests on the Hugging Face Hub.N)Path) Annotated) constants) AuthorOptFormatWithAutoOptLimitOpt RepoIdArgRepoType RepoTypeOptTokenOptapi_object_to_dict get_hf_api typer_factory)OutputFormatWithAutooutc"eZdZdZdZdZdZdZdS)DiscussionStatusopenclosedmergeddraftallN)__name__ __module__ __qualname__rrrrri/lsinfo/ai/hellotax_ai/base_platform/venv/lib/python3.11/site-packages/huggingface_hub/cli/discussions.pyrr(s' D F F E CCCrrceZdZdZdZdZdS)DiscussionKindr discussion pull_requestN)rrrrr"r#rrrr!r!0s CJ!LLLrr!rrz&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_bodyr3Fsd 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_typeformattokenct|}|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)r6r9discussion_typediscussion_statusr;c,g|]}t|Sr)r).0ds r z#discussion_list..s! 8 8 8q  " " 8 8 8r)numtitleis_pull_requestr7r9 created_atrFright)headersid_key alignments)rrrrr!rvalueget_repo_discussions_CLIENT_SIDE_STATUSESr7appendlenrtable)r6r7r8r9r:r;r<r=api api_statusapi_discussion_type discussionsrDitemss rdiscussion_listrYTsDD 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 jsonrFct|}||||j}tj|dS)z,Get info about a discussion or pull request.r?r6discussion_numr;N)rget_discussion_detailsrNrdictr6rFr;r<r=rTdetailss rdiscussion_inforbsQ 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?r6r]rkr;zCommented on # in rFrepo)r3r,r-rcomment_discussionrNrri) r6rFr&r'r;r<r=rkrTs rdiscussion_commentrrs:y))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 ''?rtr?rr6r] new_statusrkr;zClosed #rnroNrconfirmrchange_discussion_statusrNrir6rFrkrtr;r<r=rTs rdiscussion_closers<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 #rvrwrxr?rryz Reopened #rnroNr{r~s rdiscussion_reopenrDs<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?)r6r]rr;z Renamed #rn)rFrprGN)rrename_discussionrNrri)r6rFrr;r<r=rTs rdiscussion_renamernsq( 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 #rvrwrxr?rmzMerged #rnroN)rr|rmerge_pull_requestrNrir~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?r\zNo diff available.N)rr^rNrrtextr`s rdiscussion_diffrst 5 ! ! !C((/)G |'  %&&&&&r)2__doc__enumr/pathlibrtypingrr,huggingface_hubr _cli_utilsrrr r r r r rrr_outputrrr.Enumrr!rPintArgumentDiscussionNumArgr3discussions_clicommandrrmodelautoOptionrYrbboolrjrrrrrrrrrrrs `_  %%%%%%                        /.......sDI"""""S$)""""7+EN 5 S4ZD4KC$J -%WXXX$  %^ 4 9-== =   G     =   B     =$ %=& '=()=* +=, -=. /====@ 1?&^ 4 9         " 0   %^ 4 9C/r/r /r   ?     /r  d  7     /r  t  G    !/r.   A    //r>?/r@ A/rB C/rD E/r/r/r/rd SC  %^ 4 9' K K  K  K  d  8      K t  C     K"# K$ % K& ' K( ) K K K KF 2S  %^ 4 9) E E  E  E d  <     E    ,      E$% E& ' E( ) E* + E E E EF 3d  %^ 4 9) G G  G  G d  >     G    ,      G$% G& ' G( ) G* + G G G GF C&^ 4 9WW W W !    WW W W WWW  W0 2O  %^ 4 9)EE E E d  <    E    ,     E$%E& 'E( )E* +EEEED 1&^ 4 9 '' ' ''  '  '  '''  '''r