o 0jn1@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__rrrrrrr`/var/www/html/Deteccion_Ine/venv/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)Zexamplesrepo_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.r4rrN)r.r1Zdiscussion_typeZdiscussion_statusr3cSsg|]}t|qSr)r ).0drrr sz#discussion_list..)numtitleZis_pull_requestr/r1Z created_atr9right)headersZid_keyZ alignments)r rrrrrvalueZget_repo_discussions_CLIENT_SIDE_STATUSESr/appendlenrtable) r.r/r0r1r2r3r4apiZ api_statusZapi_discussion_typeZ discussionsr7itemsrrrdiscussion_listSs< !       rDinfoz'hf discussions info username/my-model 5z5hf discussions info username/my-model 5 --format jsonr9cCs*t|d}|j|||jd}t|dS)z,Get info about a discussion or pull request.r5r.discussion_numr3N)r get_discussion_detailsr=rdictr.r9r3r4rBdetailsrrrdiscussion_infos rLcreate)zrdrerenamez9hf 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.r5)r.rGrgr3z Renamed #rU)r9rWr:N)r Zrename_discussionr=rrQ)r.r9rgr3r4rBrrrdiscussion_renamegs "rhmergez(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 #r\r]r^r5rTzMerged #rUrVN)rrar Zmerge_pull_requestr=rQrbrrrdiscussion_merges  rjdiffz'hf discussions diff username/my-model 5cCs@t|d}|j|||jd}|jrt|jdStddS)z Show the diff of a pull request.r5rFzNo diff available.N)r rHr=rkrtextrJrrrdiscussion_diffs rm)/__doc__enumr(pathlibrtypingrr%Zhuggingface_hubrZ _cli_utilsrrrr r r r r r_outputrr'Enumrrr>intZArgumentZDiscussionNumArgr,Zdiscussions_clicommandrrmodelOptionrDrLboolrRrXrcrerhrjrmrrrrs    ,  "   ?    ! 1  "  "  "     !