`gj4 dZddlZddlZddlZddlmZmZmZeje Z ejZ dZ dZdZd>dZdZd d d d d dd dddddddddddd ddddgddZdedefdZd Zd!d"d d d dd d#dd ddddd$ddd ddd%dd&gddZdedefd'Zd Zd(d)d d d dd d*dd d+dd dddd,gd-ddZdedefd.Zd/Zd0d1d d d dd d2dd dddd3dd4gddZdedefd5Zej>d d6eeegdd7d89 ej>d!d6eeegdd:d89 ej>d(d6eeegdd;d<9 ej>d0d6eeegdd=d<9 y)?aFeishu Drive Tools -- document comment operations via Feishu/Lark API. Provides tools for listing, replying to, and adding document comments. Uses the same lazy-import + BaseRequest pattern as feishu_comment.py. The lark client is injected per-thread by the comment event handler. N)registry tool_error tool_resultc|t_y)zFStore a lark client for the current thread (called by feishu_comment).N)_localclient)rs J/root/.hermes/venv/lib/python3.12/site-packages/tools/feishu_drive_tool.py set_clientr s FMc$ttddS)z7Return the lark client for the current thread, or None.rN)getattrrr r get_clientrs 68T **r cpddl} |jjdduS#ttf$rYywxYw)Nr lark_oapiF)importlib.utilutil find_spec ImportError ValueError) importlibs r _check_feishurs=~~'' 4D@@  $s #55c4ddlm}ddlm}ddlm}|dk(r |j n |j} |jj| j|j|jh} |r| j|} |r| j|} || j|} | j!} |j#| } t%| dd} t%| dd }i}t%| d d}|r>t'|d r2 t)j*|j,}|j/d i}|s9t%| d d}t5|t6r|}n|rt'|d r t9|}| ||fS#t(j0t2f$rY[wxYw)z?Build and execute a BaseRequest, return (code, msg, data_dict).r)AccessTokenType) HttpMethod) BaseRequestGETNcodemsgrawcontentdata__dict__)rrlark_oapi.core.enumr!lark_oapi.core.model.base_requestrrPOSTbuilder http_methoduri token_typesTENANTpathsqueriesbodybuildrequestr hasattrjsonloadsr"getJSONDecodeErrorAttributeError isinstancedictvars)rmethodr*r-r.r/rrrr)r(r1responserrr#r! body_json resp_datas r _do_requestr?(sr).=$*eO*..K  [ ! S o,,- .   --&//'* ,,t$mmoG~~g&H 8VT *D (E2 &C D (E4 (C wsI&  3;;/I==,D Hfd3 i &D 79j9 ?D d?$$n5   s 1E;;FFz./open-apis/drive/v1/files/:file_token/commentsfeishu_drive_list_commentsz[List comments on a Feishu document. Use is_whole=true to list whole-document comments only.objectstringzThe document file token.)type descriptionzFile type (default: docx).docx)rCrDdefaultbooleanz-If true, only return whole-document comments.Fintegerz&Number of comments per page (max 100).dzPagination token for next page.) file_token file_typeis_whole page_size page_tokenrJ)rC propertiesrequired)namerD parametersargsreturnc t}| tdS|jddj}|s tdS|jddxsd}|jdd}|jd d }|jd d}d|fd d t |fg}|r|j d |r|j d |ft |dtd|i|\} } } | dk7rtd| d| St| S)NFeishu client not availablerJr zfile_token is requiredrKrErLFrMrIrN user_id_typeopen_id)rLtruerr-r.rzList comments failed: code= msg=) rrr5stripstrappendr?_LIST_COMMENTS_URIr) rSkwargsrrJrKrLrMrNr.rrr#s r _handle_list_commentsrbs \F ~788,+113J 233f-7Ixx E*Hc*I,+J i # c)n%G +, j12!)Z(OD#t  qy7vU3%HII t r zB/open-apis/drive/v1/files/:file_token/comments/:comment_id/replies!feishu_drive_list_comment_repliesz:List all replies in a comment thread on a Feishu document.z#The comment ID to list replies for.z%Number of replies per page (max 100).)rJ comment_idrKrMrNrdc t}| tdS|jddj}|jddj}|r|s tdS|jddxsd}|jdd }|jd d}d|fd dt |fg}|r|j d |ft |d t||d |\} } } | dk7rtd| d| St| S)NrVrJr rdz&file_token and comment_id are requiredrKrErMrIrNrWrrJrdr[rzList replies failed: code=r\) rrr5r]r^r_r?_LIST_REPLIES_URIr) rSrarrJrdrKrMrNr.rrr#s r _handle_list_repliesrhs \F ~788,+113J,+113J ZBCCf-7Ic*I,+J i # c)n%G  j12!('zBOD#t  qy6tfE#GHH t r feishu_drive_reply_commentzReply to a local comment thread on a Feishu document. Use this for local (quoted-text) comments. For whole-document comments, use feishu_drive_add_comment instead.zThe comment ID to reply to.z6The reply text content (plain text only, no markdown).)rJrdr"rK)rJrdr"c t}| tdS|jddj}|jddj}|jddj}|r|r|s tdS|jddxsd}dd d d |id gii}t |d t ||dd|fg|\}} } |dk7rtd|d| St d| S)NrVrJr rdr"z0file_token, comment_id, and content are requiredrKrEelementstext_runtext)rCrlr'rf)r-r.r/rzReply comment failed: code=r\Tsuccessr#)rrr5r]r?_REPLY_COMMENT_URIr) rSrarrJrdr"rKr/rrr#s r _handle_reply_commentrqs \F ~788,+113J,+113Jhhy"%++-G ZwLMMf-7I  &!' 1  D"*'zBy)*  OD#t  qy7vU3%HII t$ //r z2/open-apis/drive/v1/files/:file_token/new_commentsfeishu_drive_add_commentzAdd a new whole-document comment on a Feishu document. Use this for whole-document comments or as a fallback when reply_comment fails with code 1069302.z8The comment text content (plain text only, no markdown).)rJr"rKr"c t}| tdS|jddj}|jddj}|r|s tdS|jddxsd}|d|d gd }t |d t d|i| \}}} |d k7rtd|d|St d| S)NrVrJr r"z#file_token and content are requiredrKrErm)rCrm)rKreply_elementsr')r-r/rzAdd comment failed: code=r\Trn)rrr5r]r?_ADD_COMMENT_URIr) rSrarrJr"rKr/rrr#s r _handle_add_commentrv_s \F ~788,+113Jhhy"%++-G W?@@f-7IW -  D"(Z( OD#t  qy5dV5FGG t$ //r feishu_drivezList document commentsu💬) rQtoolsetschemahandlercheck_fn requires_envis_asyncrDemojizList comment replieszReply to a document commentu✉️zAdd a whole-document comment)NNN) __doc__r3logging threadingtools.registryrrr getLogger__name__loggerlocalrr rrr?r`!FEISHU_DRIVE_LIST_COMMENTS_SCHEMAr9r^rbrg FEISHU_DRIVE_LIST_REPLIES_SCHEMArhrpFEISHU_DRIVE_REPLY_SCHEMArqruFEISHU_DRIVE_ADD_COMMENT_SCHEMArvregisterrr r rs <<   8 $   + .jF ) B!9 !;! "N  "G !@) 2"N7 #%!L  3 NY 0O!9 !D !;! "F !@' 0"<05$ Dt#JZ ) M !9 !< !W !;! &<+B!0!03!0PH ' 1 !9 !Y !;! "9-##:0d00D %  , !  (   ,  +  &   %  $ !  -   #  *   .  r