ci~dZddlmZddlmZmZmZmZmZm Z m Z m Z m Z ddl Z ddl mZddlmZddlmZmZGdd eZGd d eZy) zMExceptionToStatusInterceptor catches GrpcException and sets the gRPC context.)contextmanager) AnyAsyncGenerator AsyncIterableCallable GeneratorIterableIteratorNoReturnOptionalN)aio) GrpcException)AsyncServerInterceptorServerInterceptorc eZdZdZddeej fdZdedejde de d e eddff d Z ededejde d edfd Zd ededejde d ef d Zdededejde d ef dZy)ExceptionToStatusInterceptora;An interceptor that catches exceptions and sets the RPC status and details. ExceptionToStatusInterceptor will catch any subclass of GrpcException and set the status code and details on the gRPC context. You can also extend this and override the handle_exception method to catch other types of exceptions, and handle them in different ways. E.g., you can catch and handle exceptions that don't derive from GrpcException. Or you can set rich error statuses with context.abort_with_status(). Args: status_on_unknown_exception: Specify what to do if an exception which is not a subclass of GrpcException is raised. If None, do nothing (by default, grpc will set the status to UNKNOWN). If not None, then the status code will be set to this value if `context.abort` hasn't been called earlier. It must not be OK. The details will be set to the value of repr(e), where e is the exception. In any case, the exception will be propagated. Raises: ValueError: If status_code is OK. Nstatus_on_unknown_exceptioncb|tjjk(r td||_yNz3The status code for unknown exceptions cannot be OKgrpc StatusCodeOK ValueError_status_on_unknown_exceptionselfrs K/tmp/pip-target-z3e9_cxr/lib/python/grpc_interceptor/exception_to_status.py__init__z%ExceptionToStatusInterceptor.__init__-( &$//*<*< <RS S,G)request_or_iteratorcontext method_nameresponse_iteratorreturnc#tK|j|||5|Ed{dddy7 #1swYyxYwwzs8 Q  Q  ! !"&97K P P Qs 3 3 0+303r0ct|tr(|j|j|j||j s1|j %|j|j t||)Override this if extending ExceptionToStatusInterceptor. This will get called when an exception is raised while handling the RPC. Args: ex: The exception that was raised. request_or_iterator: The RPC request, as a protobuf message if it is a unary request, or an iterator of protobuf messages if it is a streaming request. context: The servicer context. You probably want to call context.abort(...) method_name: The name of the RPC being called. Raises: This method must raise and cannot return, as in general there's no meaningful RPC response to return if an exception has occurred. You can raise the original exception, ex, or something else.  isinstancerabort status_codedetailscoderreprrr0r"r#r$s rr/z-ExceptionToStatusInterceptor.handle_exceptionGs_0 b- ( MM".."** 500< d??bJr!methodc|j|||5|||}dddttr|j||||S|S#1swY/xYw)FDo not call this directly; use the interceptor kwarg on grpc.server().N)r)r4r r*)rr;r"r#r$response_or_iterators r interceptz&ExceptionToStatusInterceptor.interceptfsp # #$7+ N H#)*=w#G  H *H 5++#Wk;O  ( ' H Hs A  Ar,)__name__ __module__ __qualname____doc__r rrrrServicerContextstrr rr*rr r)r.r r/rr?r!rrrs(HHT__4MH )  )%% ) ) $ ) 3d? # )Q#&Q151E1EQTWQ $QQ !%%     >((!(%% (  ( (r!rc eZdZdZddeej fdZdede jde de d e edff d Zd edede jde d ef d Zd edede jde d ef dZy)!AsyncExceptionToStatusInterceptorzAn interceptor that catches exceptions and sets the RPC status and details. This is the async analogy to ExceptionToStatusInterceptor. Please see that class' documentation for more information. Nrcb|tjjk(r td||_yrrrs rrz*AsyncExceptionToStatusInterceptor.__init__r r!r"r#r$r%r&cK |23d{}|7 6y#t$r'}|j||||d{7Yd}~yd}~wwxYwwr(r-)rr"r#r$r%rr0s rr*z5AsyncExceptionToStatusInterceptor._generate_responsessU W,  a , W'',?+V V V WsJA A  A A<AA A  A r0c(Kt|tr0|j|j|jd{||j s9|j -|j|j t|d{|7Q7w)r2Nr3r:s rr/z2AsyncExceptionToStatusInterceptor.handle_exceptions{0 b- (-- ; ; ;00<mmD$E$EtBxPPP <Qs":BBA BBBBr;cK |||}t|ds |d{S |j|||S7#t$r'}|j||||d{7Yd}~Ad}~wwxYww)r= __aiter__N)hasattrr.r/r*)rr;r"r#r$r>r0s rr?z+AsyncExceptionToStatusInterceptor.intercepts W#)*=w#G /=111> '' +7K  2 W'',?+V V V WsCA-:8:A-: A*A%AA% A-%A**A-r,)r@rArBrCr rrrrgrpc_aiorDrErrr*r.r r/rr?rFr!rrHrH{s HHT__4MH W  W)) W W ) W T " W !))     >  ! ))     r!rH)rC contextlibrtypingrrrrrr r r r rr rPgrpc_interceptor.exceptionsrgrpc_interceptor.serverrrrrHrFr!rrUsIS&    5M`(#4`(FK (>K r!