o jn1@stdZddlZddlZddlmZddlmZddlZddlm Z ddl m Z m Z m Z mZmZmZmZmZmZddlmZGd d d eejZGd d d eejZd dhZeeejdddfZdedBdedBdedBfddZeddZej dgddej!ej"ddej#dfde deeej$ddd dfd!eeej$d"d#d$dfd%e d&e d'ed(eddfd)d*Z%ej d+d,d-gdej#dfde d.ed'ed(eddf d/d0Z&ej d1gd2dddd3ej#dfde d4eeej$d5d6dfdeedBej$d7d8dfdeedBej$d9d:dfd;ee'ej$ddfd'ed(eddfd?d@Z(ej dAdBdCgdddej#dfde d.edeedBej$d7dDdfdeedBej$d9dEdfd'ed(eddfdFdGZ)ej dHdIdJgddd3ej#dfde d.edAeedBej$dKdLdfdMee'ej$dNdOdPdfd'ed(eddfdQdRZ*ej dSdTdUgddd3ej#dfde d.edAeedBej$dKdVdfdMee'ej$dNdOdPdfd'ed(eddfdWdXZ+ej dYdZgdej#dfde d.ed[eeejd\dfd'ed(eddf d]d^Z,ej d_d`dagddd3ej#dfde d.edAeedBej$dKdbdfdMee'ej$dNdOdPdfd'ed(eddfdcddZ-ej dedfgdej#dfde d.ed'ed(eddf dgdhZ.dS)izYContains 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@s eZdZdZdZdZdZdZdS)DiscussionStatusopenclosedmergeddraftallN)__name__ __module__ __qualname__rrrrrrrR/home/kuhnn/.local/lib/python3.10/site-packages/huggingface_hub/cli/discussions.pyr's rc@seZdZdZdZdZdS)DiscussionKindr discussion pull_requestN)rrrrrrrrrrr/srrrz&The discussion or pull request number.)helpminbody body_filereturncCsH|dur |dur td|dur"t|dkrtjS|jddS|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!rrr _read_bodyEs    r,z0Manage discussions and pull requests on the Hub.)rz 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_typetokenc Cst|d}|tjkr d}n |tjkrd}nd}|tjkrd} n|j} g} |j||| ||jdD]} |jtvr=| j |jkr=q/| | t | |krJnq/dd| D} t j | gdd d d id dS) z-List discussions and pull requests on a repo.r5rrN)r/r2discussion_typediscussion_statusr4cSsg|]}t|qSr)r ).0drrr sz#discussion_list..)numtitleis_pull_requestr0r2 created_atr<right)headersid_key alignments)r rrrrrvalueget_repo_discussions_CLIENT_SIDE_STATUSESr0appendlenrtable) r/r0r1r2r3r4r5api api_statusapi_discussion_type discussionsr:itemsrrrdiscussion_listSs< !       rOinfoz'hf discussions info username/my-model 5z5hf discussions info username/my-model 5 --format jsonr<cCs*t|d}|j|||jd}t|dS)z,Get info about a discussion or pull request.r6r/discussion_numr4N)r get_discussion_detailsrDrdictr/r<r4r5rJdetailsrrrdiscussion_infos rWcreate)zrsrurenamez9hf discussions rename username/my-model 5 "Updated title" new_titlezThe new title.cCs@t|d}|j||||jdtjd|d||||ddS)z$Rename a discussion or pull request.r6)r/rRrwr4z Renamed #ra)r<rcr=N)r rename_discussionrDrr])r/r<rwr4r5rJrrrdiscussion_renamegs "rymergez(hf discussions merge username/my-model 5zEhf discussions merge username/my-model 5 --comment "Merging, thanks!"z)An optional comment to post when merging.cCsZtjd|d|d|dt|d}|j||||jdtjd|d|||d d S) zMerge a pull request.zMerge #rirjrkr6r`zMerged #rarbN)rror merge_pull_requestrDr]rqrrrdiscussion_merges  r|diffz'hf discussions diff username/my-model 5cCs@t|d}|j|||jd}|jrt|jdStddS)z Show the diff of a pull request.r6rQzNo diff available.N)r rSrDr}rtextrUrrrdiscussion_diffs r)/__doc__enumr(pathlibrtypingrr%huggingface_hubr _cli_utilsrrrr r r r r r_outputrr'EnumrrrFintArgumentDiscussionNumArgr,discussions_clicommandrrmodelOptionrOrWboolr^rerrruryr|rrrrrs    ,  "   ?    ! 1  "  "  "     !