o 0iA@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..rM)rMrKrHr-rNrOc Ss`d|dt|dddd|drdndtt|d dt|d dt|d dgS) NrCrMrKrF2)max_lenrHZPRr-rNrO)rgetr3r:)itemr!r!r"s z!discussion_list..right)rcrdZid_keyheadersZrow_fnZ alignments) rrrrr$rvalueZget_repo_discussions_CLIENT_SIDE_STATUSESr-rQrWr)r`r-rYrNrarbrcrdreapiZ api_statusZapi_discussion_typeZ discussionsrhitemsr!r!r"discussion_listsB #       ruinfo)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 jsonrMr[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_COLORrfr`discussion_numrbrw)indent)rArD)osenvironrget_discussion_detailsrqr'r)rpoprIdumpsr^rwrrL) r`rMr[rwrxrbrcrerrsr@resultr!r!r"discussion_infos*+      rcreate)zr*enumr)r;pathlibrtypingrrr8Zhuggingface_hubrZhuggingface_hub.communityrrZhuggingface_hub.utilsrZ _cli_utilsr r r r rrrrrrrrrrr:Enumrr$r'rrintZArgumentZDiscussionNumArgr3r?boolr^Zdiscussions_clicommandrrmodeltableOptionrur(rrrrrrrrr!r!r!r"s     @  "  +  L  =  ! 3  "  &  &     %