kKjdZddlmZddlZddlZddlZddlZddlm Z ddl m Z ddl m Z mZddlmZmZmZddlmZmZGd d ZGd d e Zy) z7Replacements for sys.displayhook that publish over ZMQ.) annotationsN) ContextVar) DisplayHook)Sessionextract_header)AnyInstancedefault) encode_images json_cleanc<eZdZdZdZdZdZdZedZ dZ y) ZMQDisplayHookzOA simple displayhook that publishes the object's repr over a ZeroMQ socket.sexecute_resultc||_||_td|_|jj ii|_y)zInitialize the hook. parent_headerN)session pub_socketr_parent_headerset_parent_header_global)selfrrs f/Users/ahmed/devFolder/Ultron/claude-voice/.venv/lib/python3.12/site-packages/ipykernel/displayhook.py__init__zZMQDisplayHook.__init__s9 $:D_:U #%'"cy)z$This method is replaced in kernelapprrs rget_execution_countz"ZMQDisplayHook.get_execution_count#srcR|y|t_tjj tj j |j dt|iid}|jj|jd||j|jy)zHandle a hook call.Nz text/plain)execution_countdatametadataexecute_result)parentident) builtins_sysstdoutflushstderrrreprrsendrrtopic)robjcontentss r__call__zZMQDisplayHook.__call__'s ;    #779!49-   OO  %%**  rcn |jjS#t$r|jcYSwxYwNrget LookupErrorrrs rrzZMQDisplayHook.parent_header<7 .&&**, , .-- - . 44c^t|}|jj|||_yzSet the parent header.Nrrrrrr#rs r set_parentzZMQDisplayHook.set_parentC(&v.   .%2"rN) __name__ __module__ __qualname____doc__r-rrr0propertyrr<rrrrrs6 E( *.. 3rrceZdZUdZdZeedZedZ de d<eZ de d<fd Z e d d Zed Zd ZdZedZdZdZdZddZdZxZS)ZMQShellDisplayHookzA displayhook subclass that publishes data using ZeroMQ. This is intended to work with an InteractiveShell instance. It sends a dict of different representations of the object.NT) allow_nonezContextVar[dict[str, Any]]rzdict[str, t.Any] | Nonemsgczt||i|td|_|jj iy)Nr)superrrrr)rargskwargs __class__s rrzZMQShellDisplayHook.__init__Ws6 $)&)(9 #r _thread_localc*tjSr2) threadinglocalrs r_default_thread_localz)ZMQShellDisplayHook._default_thread_local\s  rc|t|jdsg|j_|jjS)Nhooks)hasattrrLrRrs r_hookszZMQShellDisplayHook._hooks`s2t))73')D   $!!'''rc:|jj|y)a(Register a transform hook on the execute_result message. Mirrors ``ZMQDisplayPublisher.register_hook``. Each hook receives the outbound message dict and must return either a (possibly mutated) message dict to continue the chain, or ``None`` to suppress the send. N)rTappendrhooks r register_hookz!ZMQShellDisplayHook.register_hookfs 4 rcZ |jj|y#t$rYywxYw)z=Remove a previously registered hook. Returns True on success.TF)rTremove ValueErrorrWs runregister_hookz#ZMQShellDisplayHook.unregister_hookos.  KK  t $  s  **cn |jjS#t$r|jcYSwxYwr2r3rs rrz!ZMQShellDisplayHook.parent_headerwr6r7c^t|}|jj|||_yr9r:r;s rr<zZMQShellDisplayHook.set_parent~r=rc~|jr1|jjdiid|j|_yy)zStart the display hook.r")r r!)r#N)rrFrrs rstart_displayhookz%ZMQShellDisplayHook.start_displayhooksB <<||''  ")) (DH rcV|jr|j|jdd<yy)zWrite the output prompt.contentrN)rF prompt_countrs rwrite_output_promptz'ZMQShellDisplayHook.write_output_prompts( 88595F5FDHHY  1 2 rc|jr7tt||jdd<||jdd<yy)z!Write format data to the message.rcr r!N)rFr r )r format_dictmd_dicts rwrite_format_dataz%ZMQShellDisplayHook.write_format_datas> 88*4];5O*PDHHY  '.5DHHY  + rctjjtjj|jr|jddrn|j rb|j}|j D]}||}|d|_y|j j|j||jd|_y)aFinish up all displayhook activities. Runs the registered hook chain before ``session.send``. Each hook either returns a message (to continue) or ``None`` (to suppress the send). This mirrors the transform pipeline on ``ZMQDisplayPublisher.publish`` so a single hook implementation can attach to both the ``display_data`` and ``execute_result`` paths. rcr N)r$) r'r(r)r*rFrrTr,rr-)rrFrXs rfinish_displayhookz&ZMQShellDisplayHook.finish_displayhooks   88+F3 ((C 3i;#DH $ LL  doos$**  Err2)r>r?r@rAr-r rrrr__annotations__rLrr rPrBrTrYr]rr<rarerirk __classcell__)rKs@rrDrDJs& Ew40G%J..EM  $  _!!(( !.. 3 G 6 rrD)rA __future__rr%r'rNtypingt contextvarsrIPython.core.displayhookrjupyter_client.sessionrr traitletsrr r ipykernel.jsonutilr r rrDrrrrvsD=# "0:,,83333ld+dr