B U[@szdZddlZddlZddlmZddlmZddlmZddl m Z m Z m Z m Z mZddlmZGd d d eZd gZdS) zBase class to manage commsN)LoggingConfigurable) import_item) string_types)InstanceUnicodeDictAnydefault)Commc@sdeZdZdZedZeZeZddZ ddZ ddZ d d Z d d Z d dZddZddZdS) CommManagerzManager for Comms in the Kernelzipykernel.kernelbase.KernelcCs t|trt|}||j|<dS)aRegister a callable f for a given target name f will be called with two arguments when a comm_open message is received with `target`: - the Comm instance - the `comm_open` message itself. f can be a Python callable or an import string for one. N) isinstancerrtargets)self target_namefr5lib/python3.7/site-packages/ipykernel/comm/manager.pyregister_targets zCommManager.register_targetcCs |j|S)z5Unregister a callable registered with register_target)rpop)rrrrrrunregister_target*szCommManager.unregister_targetcCs|j}|j|_||j|<|S)zRegister a new comm)comm_idkernelcomms)rcommrrrr register_comm.s zCommManager.register_commcCs|j|j}dS)z,Unregister a comm, and close its counterpartN)rrr)rrrrrunregister_comm5szCommManager.unregister_commc CsXy |j|Stk rR|jd||jtjrN|jdt|j YnXdS)zGet a comm with a particular id Returns the comm if found, otherwise None. This will not raise an error, it will log messages if the comm cannot be found. zNo such comm: %szCurrent comms: %sN) rKeyErrorlogZwarningZ isEnabledForloggingDEBUGdebuglistkeys)rrrrrget_comm:s  zCommManager.get_commc Cs|d}|d}|d}|j|d}t|d|d}|||dkrV|jd|n6y|||dStk r|jjd|d d YnXy |Wn|jjd d d YnXdS) zHandler for comm_open messagescontentrrNF)rZprimaryrz"No such comm target registered: %sz&Exception opening comm with target: %sT)exc_infozqCould not close comm during `comm_open` failure clean-up. The comm may not have been opened yet.)rgetr rrerror Exceptionclose) rstreamidentmsgr%rrrrrrr comm_openKs(   zCommManager.comm_opencCs`|d}|d}||}|dkr&dSy||Wn&tk rZ|jjd|ddYnXdS)zHandler for comm_msg messagesr%rNzException in comm_msg for %sT)r&)r$Z handle_msgr)rr()rr+r,r-r%rrrrrcomm_msgfs zCommManager.comm_msgcCsh|d}|d}||}|dkr&dS|j|=y||Wn&tk rb|jjd|ddYnXdS)zHandler for comm_close messagesr%rNzException in comm_close for %sT)r&)r$rZ handle_closer)rr()rr+r,r-r%rrrrr comm_closess zCommManager.comm_closeN)__name__ __module__ __qualname____doc__rrrrrrrrrr$r.r/r0rrrrr s r )r4sysrZtraitlets.configrZipython_genutils.importstringrZipython_genutils.py3compatrZ traitletsrrrrr rr r __all__rrrrs    p