o ~riA@sPdZddlZddlZddlZddlmZddlmZmZddl Z ddl m Z ddl m Z mZddlmZdd lmZmZmZmZmZmZmZmZmZmZmZmZmZmZGd d d e ej!Z"Gd d d e ej!Z#Gddde ej!Z$ddhZ%ee&e j'dddfZ(de de fddZ)dee deedee fddZ*d|dede+ddfdd Z,ed!d"Z-e-j.d#gd$d%e"j/e#j0dd&ej1ej2ddfd'edee"e j3d(d)d*d"fd+ee#e j3d,d-d.d"fd/ed0ed1ed2ed3ed4eddfd5d6Z4e-j.d7gd8d%dddej1e$j5dfd'ed9e(d:ee+e j3d;dd?d"fd@ee+e j3dAdBd"fd1ed2ee$e j3dCd"fd4eddfdDdEZ6e-j.dFgdGd%dddej1dfd'edHee e j3dIdJd"fdeee e j3dKdLd"fdeeee j3dMdNd"fdOee+e j3dPdQdRd"fd1ed4eddfdSdTZ7e-j.dUdVdWgd%ddej1dfd'ed9e(deee e j3dKdXd"fdeeee j3dMdYd"fd1ed4eddfdZd[Z8e-j.d\d]d^gd%ddej1dfd'ed9e(dUeee e j3d_d`d"fdaee+e j3dbdcddd"fd1ed4eddfdedfZ9e-j.dgdhdigd%ddej1dfd'ed9e(dUeee e j3d_djd"fdaee+e j3dbdcddd"fd1ed4eddfdkdlZ:e-j.dmdngd%ej1dfd'ed9e(doee e j'dpd"fd1ed4eddf dqdrZ;e-j.dsdtdugd%ddej1dfd'ed9e(dUeee e j3d_dvd"fdaee+e j3dbdcddd"fd1ed4eddfdwdxZsz#discussion_list..rN)rNrLrIr-rOrPc Ss`d|dt|dddd|drdndtt|d dt|d dt|d dgS) NrDrNrLrG2)max_lenrIPRr-rOrP)rgetr3r:)itemr!r!r"s z!discussion_list..right)rhriid_keyheadersrow_fn alignments) rrrrr$rvalueget_repo_discussions_CLIENT_SIDE_STATUSESr-rSr[r)rer-r]rOrfrgrhrirjapi api_statusapi_discussion_type discussionsroitemsr!r!r"discussion_listsB #       rinfo)z'hf discussions info username/my-model 5z2hf discussions info username/my-model 5 --commentsz.hf discussions info username/my-model 5 --diffz5hf discussions info username/my-model 5 --format jsonrNr_z --commentszShow all comments.diffz--diffz"Show the diff (for pull requests).no_colorz --no-colorzDisable colored output.zOutput format (text or json).c Csddl}|r d|jd<t|d} | j|||jd} |tjkr5t| } |s*| ddt tj | dd dSt | |d |rQ| j rSt t t d t | j dSdSdS) z,Get info about a discussion or pull request.rN1NO_COLORrkrediscussion_numrgr)indent)rArE)osenvironrget_discussion_detailsr|r'r)rpoprJdumpsrbrrrM) rerNr_rrrgrhrjrrr@resultr!r!r"discussion_infos*+      rcreate)zr*enumr)r;pathlibrtypingrrr8huggingface_hubrhuggingface_hub.communityrrhuggingface_hub.utilsr _cli_utilsr r r r rrrrrrrrrrr:Enumrr$r'r~intArgumentDiscussionNumArgr3r?boolrbdiscussions_clicommandrrmodeltableOptionrr(rrrrrrrrr!r!r!r"s     @  "  +  L  =  ! 3  "  &  &     %