diff --git a/src/momento/internal/aio/_add_header_client_interceptor.py b/src/momento/internal/aio/_add_header_client_interceptor.py index 0430ec56..771c1aed 100644 --- a/src/momento/internal/aio/_add_header_client_interceptor.py +++ b/src/momento/internal/aio/_add_header_client_interceptor.py @@ -16,9 +16,8 @@ def __init__(self, name: str, value: str): class AddHeaderStreamingClientInterceptor(grpc.aio.UnaryStreamClientInterceptor): - are_only_once_headers_sent = False - def __init__(self, headers: list[Header]): + self.are_only_once_headers_sent = False self._headers_to_add_once: list[Header] = list( filter(lambda header: header.name in header.once_only_headers, headers) ) @@ -40,18 +39,17 @@ async def intercept_unary_stream( for header in self.headers_to_add_every_time: new_client_call_details.metadata.add(header.name, header.value) - if not AddHeaderStreamingClientInterceptor.are_only_once_headers_sent: + if not self.are_only_once_headers_sent: for header in self._headers_to_add_once: new_client_call_details.metadata.add(header.name, header.value) - AddHeaderStreamingClientInterceptor.are_only_once_headers_sent = True + self.are_only_once_headers_sent = True return await continuation(new_client_call_details, request) class AddHeaderClientInterceptor(grpc.aio.UnaryUnaryClientInterceptor): - are_only_once_headers_sent = False - def __init__(self, headers: list[Header]): + self.are_only_once_headers_sent = False self._headers_to_add_once: list[Header] = list( filter(lambda header: header.name in header.once_only_headers, headers) ) @@ -73,9 +71,9 @@ async def intercept_unary_unary( for header in self.headers_to_add_every_time: new_client_call_details.metadata.add(header.name, header.value) - if not AddHeaderClientInterceptor.are_only_once_headers_sent: + if not self.are_only_once_headers_sent: for header in self._headers_to_add_once: new_client_call_details.metadata.add(header.name, header.value) - AddHeaderClientInterceptor.are_only_once_headers_sent = True + self.are_only_once_headers_sent = True return await continuation(new_client_call_details, request) diff --git a/src/momento/internal/synchronous/_add_header_client_interceptor.py b/src/momento/internal/synchronous/_add_header_client_interceptor.py index cafd7437..7ed11327 100644 --- a/src/momento/internal/synchronous/_add_header_client_interceptor.py +++ b/src/momento/internal/synchronous/_add_header_client_interceptor.py @@ -19,9 +19,8 @@ def __init__(self, name: str, value: str): class AddHeaderStreamingClientInterceptor(grpc.UnaryStreamClientInterceptor): - are_only_once_headers_sent = False - def __init__(self, headers: list[Header]): + self.are_only_once_headers_sent = False self._headers_to_add_once: list[Header] = list( filter(lambda header: header.name in header.once_only_headers, headers) ) @@ -43,17 +42,15 @@ def intercept_unary_stream( for header in self.headers_to_add_every_time: new_client_call_details.metadata.append((header.name, header.value)) - if not AddHeaderStreamingClientInterceptor.are_only_once_headers_sent: + if not self.are_only_once_headers_sent: for header in self._headers_to_add_once: new_client_call_details.metadata.append((header.name, header.value)) - AddHeaderStreamingClientInterceptor.are_only_once_headers_sent = True + self.are_only_once_headers_sent = True return continuation(new_client_call_details, request) class AddHeaderClientInterceptor(grpc.UnaryUnaryClientInterceptor): - are_only_once_headers_sent = False - @staticmethod def is_only_once_header(header: Header) -> bool: return header.name in header.once_only_headers @@ -63,6 +60,7 @@ def is_not_only_once_header(header: Header) -> bool: return header.name not in header.once_only_headers def __init__(self, headers: list[Header]): + self.are_only_once_headers_sent = False self._headers_to_add_once: list[Header] = list(filter(AddHeaderClientInterceptor.is_only_once_header, headers)) self.headers_to_add_every_time = list(filter(AddHeaderClientInterceptor.is_not_only_once_header, headers)) @@ -77,9 +75,9 @@ def intercept_unary_unary( for header in self.headers_to_add_every_time: new_client_call_details.metadata.append((header.name, header.value)) - if not AddHeaderClientInterceptor.are_only_once_headers_sent: + if not self.are_only_once_headers_sent: for header in self._headers_to_add_once: new_client_call_details.metadata.append((header.name, header.value)) - AddHeaderClientInterceptor.are_only_once_headers_sent = True + self.are_only_once_headers_sent = True return continuation(new_client_call_details, request)