flytekit.clients.grpc_utils.auth_interceptor

Directory

Classes

flytekit.clients.grpc_utils.auth_interceptor.AuthUnaryInterceptor

This Interceptor can be used to automatically add Auth Metadata for every call - lazily in case authentication is needed.

class AuthUnaryInterceptor(
    get_authenticator: typing.Callable[[], flytekit.clients.auth.authenticator.Authenticator],
)

Methods

intercept_unary_stream()

def intercept_unary_stream(
    continuation,
    client_call_details,
    request,
)

Handles a stream call and adds authentication metadata if needed

intercept_unary_unary()

def intercept_unary_unary(
    continuation: typing.Callable,
    client_call_details: grpc.ClientCallDetails,
    request: typing.Any,
)

Intercepts unary calls and adds auth metadata if available. On Unauthenticated, resets the token and refreshes and then retries with the new token

Properties