o i@s~dZddlmZmZddlmZmZmZmZddl m Z m Z m Z m Z ddlmZddlmZGdd d eZGd d d Zd S) a Generic handler resolution logic for container standards. This module provides a reusable template for resolving handlers with a consistent priority order across different container types (SageMaker, etc.). ## Handler Resolution Priority Order 1. **Environment variable** specified function 2. **Registry** @decorated function 3. **Customer script** function 4. **Default** handler (if any) ## Usage Examples ```python from model_hosting_container_standards.handler_resolver import GenericHandlerResolver # Define handler-specific configuration class MyHandlerConfig: def get_env_handler(self, handler_type: str): # Return handler from environment variable pass def get_customer_script_handler(self, handler_type: str): # Return handler from customer script pass # Create resolver with configuration resolver = GenericHandlerResolver(MyHandlerConfig()) # Resolve specific handlers ping_handler = resolver.resolve_handler("ping") invoke_handler = resolver.resolve_handler("invoke") ``` )ABCabstractmethod)AnyCallableOptionalUnion)HandlerFileNotFoundErrorHandlerNotFoundErrorHandlerResolutionErrorInvalidHandlerSpecError)logger)handler_registryc@sZeZdZdZededeedefedffddZ edede edeffdd Z dS) HandlerConfigzAbstract base class for handler configuration. Subclasses must implement methods to load handlers from different sources. handler_typereturn.NcCdS)aGet handler from environment variable. Args: handler_type: Type of handler (e.g., "ping", "invoke") Returns: Callable: Function handler str: Router path None: No handler found Raises: HandlerResolutionError: If env var is set but invalid InvalidHandlerSpecError: If handler spec is malformed Nselfrrrm/home/ubuntu/.local/lib/python3.10/site-packages/model_hosting_container_standards/common/handler/resolver.pyget_env_handler9szHandlerConfig.get_env_handlercCr)zGet handler from customer script. Args: handler_type: Type of handler (e.g., "ping", "invoke") Returns: Callable function or None if not found Nrrrrrget_customer_script_handlerMs z)HandlerConfig.get_customer_script_handler) __name__ __module__ __qualname____doc__rstrrrrrrrrrrrr3s rc@s|eZdZdZddeddfddZdedeefdd Z dedeefd d Z dedeefd d Z dedeefddZ dS)GenericHandlerResolverzGeneric handler resolver that implements consistent priority order. The resolver uses a clean separation of concerns with individual methods for each resolution step, making the code more maintainable and testable. NconfigrcCs||_|pt|_dS)zInitialize the handler resolver. Args: config: Handler configuration for loading from different sources registry: Handler registry (defaults to global registry) N)rrregistry)rrr rrr__init__cszGenericHandlerResolver.__init__rc Csz/|j|}t|r!t|dt|}td|d||WS|r.td|d|Wntt fyBt d|dwtd|d d S) a[Try to resolve handler from environment variable. Args: handler_type: Type of handler to resolve Returns: Handler function if found, None otherwise Raises: HandlerResolutionError: If env var is set but invalid InvalidHandlerSpecError: If handler spec is malformed rz Found env handler: zEnv z handler is router path: zEnvironment variable z handler configuration errorzNo env  handler foundN) rrcallablegetattrrr infodebugr r error)rr env_handler handler_namerrr_try_env_handlerms&  z'GenericHandlerResolver._try_env_handlercCsP|j|}|rt|dt|}td|d||Std|ddS)zTry to resolve handler from registry decorators. Args: handler_type: Type of handler to resolve Returns: Handler function if found, None otherwise rzFound decorator r"z No decorator r#N)r get_decorator_handlerr%rr r')rrdecorated_handlerr*rrr_try_decorator_handlers  z-GenericHandlerResolver._try_decorator_handlerc Csz |j|}|rt|dt|}td|d||WSWn2ttfyB}ztd|dt|j WYd}~nd}~wt yRt d|dwtd |d dS) zTry to resolve handler from customer script. Args: handler_type: Type of handler to resolve Returns: Handler function if found, None otherwise rzFound customer module z: zNo customer script z function found: NzCustomer script z function failed to loadzNo customer module z found) rrr%rr r'rr typer Exceptionr()rrcustomer_handlerr*errr_try_customer_script_handlers(   z3GenericHandlerResolver._try_customer_script_handlercCsRtd|d|j|j|jfD] }||}|r|Sqtd|ddS)a Resolve handler with priority order: 1. Environment variable specified function 2. Registry @decorated function 3. Customer script function 4. Default handler registered by the framework Args: handler_type: Type of handler to resolve (e.g., "ping", "invoke") Returns: Resolved handler function or None if not found resolve_z_handler calledzNo z handler found anywhereN)r r'r+r.r3)rrresolver_methodhandlerrrrresolve_handlers z&GenericHandlerResolver.resolve_handler)N) rrrrrr!rrrr+r.r3r7rrrrr\s $rN)rabcrrtypingrrrr,model_hosting_container_standards.exceptionsrr r r 0model_hosting_container_standards.logging_configr r rrrrrrrs$  )