o û°“i0ã@s8dZddlmZmZmZmZmZGdd„dƒZeƒZdS)z.Handler registry for managing Custom handlers.é)ÚAnyÚCallableÚDictÚListÚOptionalc@szeZdZdZd*dd„Zdededefddfd d „Zd ededefddfd d „Z d ededefddfdd„Z dede edeffdd„Z d ede edeffdd„Z d ede edeffdd„Zd edefdd„Zd edefdd„Zdedefdd„Zdeddfdd„Zd eddfdd„Zd eddfd d!„Zdeefd"d#„Zdeefd$d%„Zdeefd&d'„Zd*d(d)„ZdS)+ÚHandlerRegistryzDGeneral registry for managing handlers by name with proper priority.ÚreturnNcCsi|_i|_i|_dS)N)Ú _handlersÚ_framework_default_handlersÚ_decorator_handlers©Úself©rúm/home/ubuntu/.local/lib/python3.10/site-packages/model_hosting_container_standards/common/handler/registry.pyÚ__init__ s zHandlerRegistry.__init__ÚnameÚhandler.cCó||j|<dS)zSet a handler by name.N©r )r rrrrrÚ set_handlerózHandlerRegistry.set_handlerÚ handler_typecCr)zÊSet a framework default handler. Args: handler_type: The type of handler (e.g., 'ping', 'invoke') handler: The handler function to register as framework default N©r ©r rrrrrÚset_framework_defaultó z%HandlerRegistry.set_framework_defaultcCr)zÂSet a decorator handler. Args: handler_type: The type of handler (e.g., 'ping', 'invoke') handler: The handler function to register as decorator handler N©r rrrrÚset_decorator_handlerrz%HandlerRegistry.set_decorator_handlercCó |j |¡S)zGet a handler by name.)r Úget©r rrrrÚ get_handler(s zHandlerRegistry.get_handlercCr)zÚGet a framework default handler. Args: handler_type: The type of handler (e.g., 'ping', 'invoke') Returns: The framework default handler function, or None if not found )r r©r rrrrÚget_framework_default,ó z%HandlerRegistry.get_framework_defaultcCr)zÊGet a decorator handler. Args: handler_type: The type of handler (e.g., 'ping', 'invoke') Returns: The decorator handler function, or None if not found )r rr"rrrÚget_decorator_handler7r$z%HandlerRegistry.get_decorator_handlercCó ||jvS)zõCheck if a framework default handler is registered. Args: handler_type: The type of handler (e.g., 'ping', 'invoke') Returns: True if the framework default handler is registered, False otherwise rr"rrrÚhas_framework_defaultBó z%HandlerRegistry.has_framework_defaultcCr&)zåCheck if a decorator handler is registered. Args: handler_type: The type of handler (e.g., 'ping', 'invoke') Returns: True if the decorator handler is registered, False otherwise rr"rrrÚhas_decorator_handlerMr(z%HandlerRegistry.has_decorator_handlercCr&)z)Check if a handler is registered by name.rr rrrÚ has_handlerXs zHandlerRegistry.has_handlercCó|j |d¡dS)zRemove a handler by name.N)r Úpopr rrrÚremove_handler\ózHandlerRegistry.remove_handlercCr+)z+Remove a framework default handler by type.N)r r,r"rrrÚremove_framework_default`r.z(HandlerRegistry.remove_framework_defaultcCr+)z#Remove a decorator handler by type.N)r r,r"rrrÚremove_decorator_handlerdr.z(HandlerRegistry.remove_decorator_handlercCót|j ¡ƒS)z"List all registered handler names.)Úlistr Úkeysr rrrÚ list_handlershrzHandlerRegistry.list_handlerscCr1)z4List all registered framework default handler types.)r2r r3r rrrÚlist_framework_defaultslrz'HandlerRegistry.list_framework_defaultscCr1)z,List all registered decorator handler types.)r2r r3r rrrÚlist_decorator_handlersprz'HandlerRegistry.list_decorator_handlerscCs"|j ¡|j ¡|j ¡dS)zClear all registered handlers.N)r Úclearr r r rrrr7ts  zHandlerRegistry.clear)rN)Ú__name__Ú __module__Ú __qualname__Ú__doc__rÚstrrrrrrrr!r#r%Úboolr'r)r*r-r/r0rr4r5r6r7rrrrrs> ÿ ÿ þ ÿ ÿ þ     rN) r;ÚtypingrrrrrrÚhandler_registryrrrrÚs v