)`idZddlmZmZddlmZmZmZmZddl m Z m Z m Z m Z ddlmZddlmZGdd eZGd 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_registryceZdZdZededeedefedffdZ edede edeffdZ dS) HandlerConfigzAbstract base class for handler configuration. Subclasses must implement methods to load handlers from different sources. handler_typereturn.NcdS)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 Nselfrs /home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/model_hosting_container_standards/common/handler/resolver.pyget_env_handlerzHandlerConfig.get_env_handler9s $ cdS)zGet handler from customer script. Args: handler_type: Type of handler (e.g., "ping", "invoke") Returns: Callable function or None if not found Nrrs rget_customer_script_handlerz)HandlerConfig.get_customer_script_handlerMs  r) __name__ __module__ __qualname____doc__rstrrrrrrrrrrrr3s   xS!3, -   ^ &    (38$ %    ^    rrceZdZdZd deddfdZdedeefdZ dedeefdZ dedeefd Z dedeefd Z 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. Nconfigrc0||_|pt|_dS)zInitialize the handler resolver. Args: config: Handler configuration for loading from different sources registry: Handler registry (defaults to global registry) N)r$rregistry)rr$r&s r__init__zGenericHandlerResolver.__init__cs  4$4 rrc |j|}t|r:t|dt |}t jd|d||S|rt jd|d|n.#ttf$rt j d|dwxYwt jd|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) r$rcallablegetattrr!r infodebugr r error)rr env_handler handler_names r_try_env_handlerz'GenericHandlerResolver._try_env_handlerms  +55lCCK $$ &{JK@P@PQQ  OOOOOPPP""  O<OO+OO'(?@    LR RRR        ;|;;;<<<tsA"B%B+B-c|j|}|r:t|dt|}t jd|d||St jd|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,r!r r.)rrdecorated_handlerr1s r_try_decorator_handlerz-GenericHandlerResolver._try_decorator_handlers!M?? MM  %"!:s3D/E/EL LRLRRLRR S S S$ $ A\AAABBBtrc  |j|}|r:t|dt|}t jd|d||Snl#t tf$r6}t jd|dt|j Yd}~n*d}~wt$rt j d|dwxYwt jd |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) r$rr,r!r r.r r typer Exceptionr/)rrcustomer_handlerr1es r_try_customer_script_handlerz3GenericHandlerResolver._try_customer_script_handlers8 #{FF|TT  (&$j#6F2G2G    TlTTlTTUUU''  ( )*>?    LWlWWT!WWEUWW            LRLRRR S S S    ?<???@@@tsAAC*,B'Cctjd|d|j|j|jfD]}||}|r|cStjd|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.r2r6r<)rrresolver_methodhandlers rresolve_handlerz&GenericHandlerResolver.resolve_handlers  = ===>>>  !  '  -   O &ol33G    @<@@@AAAtr)N) rrrr rr'r!rrr2r6r<rArrrr#r#\s 55}55555"S"Xh5G""""H38H;M((ASrFs""H$#######111111111111 DCCCCC&&&&&&& & & & & C& & & RHHHHHHHHHHr