o iU @sdZddlmZddlZddlZddlZddlZeeZ GdddeZ Gddde dgde Z d d Z d d Zd dZdS)z%Utilities for working with callables.)ABCNc@s(eZdZdZejGdddejZdS)OutcomeaA sum type describing the outcome of some call. Attributes: kind: One of Kind.RETURNED or Kind.RAISED respectively indicating that the call returned a value or raised an exception. return_value: The value returned by the call. Must be present if kind is Kind.RETURNED. exception: The exception raised by the call. Must be present if kind is Kind.RAISED. c@s eZdZdZeZeZdS)z Outcome.Kindz8Identifies the general kind of the outcome of some call.N)__name__ __module__ __qualname____doc__enumautoRETURNEDRAISEDr r d/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/grpc/framework/foundation/callable_util.pyKind%s rN)rrrrruniqueEnumrr r r r rs rc@seZdZdZdS) _EasyOutcomez$A trivial implementation of Outcome.N)rrrrr r r r r-sr)kind return_value exceptionc Os\zttjj||i|dWSty-}zt|ttjjd|WYd}~Sd}~wwN)rrrr Exception_LOGGERrr )behaviormessageargskwargser r r _call_logging_exceptions6s rcstfdd}|S)aWraps a callable in a try-except that logs any exceptions it raises. Args: behavior: Any callable. message: A string to log if the behavior raises an exception. Returns: A callable that when executed invokes the given behavior. The returned callable takes the same arguments as the given behavior but returns a future.Outcome describing whether the given behavior returned a value or raised an exception. cstg|Ri|Srr)rrrrr r wrapped_behaviorNsz0with_exceptions_logged..wrapped_behavior) functoolswraps)rrr r rr with_exceptions_logged@sr#cOst||g|Ri|S)aCalls a behavior in a try-except that logs any exceptions it raises. Args: behavior: Any callable. message: A string to log if the behavior raises an exception. *args: Positional arguments to pass to the given behavior. **kwargs: Keyword arguments to pass to the given behavior. Returns: An Outcome describing whether the given behavior returned a value or raised an exception. r)rrrrr r r call_logging_exceptionsUs r$)rabcr collectionsrr!logging getLoggerrrr namedtuplerrr#r$r r r r s"