diff --git a/pyinfra/server/interpreter/interpreters/rest_callback.py b/pyinfra/server/interpreter/interpreters/rest_callback.py index e05f269..718fe97 100644 --- a/pyinfra/server/interpreter/interpreters/rest_callback.py +++ b/pyinfra/server/interpreter/interpreters/rest_callback.py @@ -18,6 +18,6 @@ def stream_response_payloads(endpoint): yield from response_stream -class PickupStreamer(Interpreter): +class RestPickupStreamer(Interpreter): def __call__(self, payloads: Iterable): yield from mapcat(stream_response_payloads, payloads) diff --git a/pyinfra/server/rest.py b/pyinfra/server/rest.py index 817dc89..42dff85 100644 --- a/pyinfra/server/rest.py +++ b/pyinfra/server/rest.py @@ -5,7 +5,7 @@ from funcy import rcompose from pyinfra.server.dispatcher.senders.rest import RestDispatcher from pyinfra.server.interpreter.interpreters.identity import IdentityInterpreter -from pyinfra.server.interpreter.interpreters.rest_callback import PickupStreamer +from pyinfra.server.interpreter.interpreters.rest_callback import RestPickupStreamer from pyinfra.server.packer.packers.rest import RestPacker from pyinfra.server.pipeline import Pipeline from pyinfra.server.receiver.receivers.rest import RestReceiver @@ -30,7 +30,7 @@ def process_lazily(url, data: Iterable[bytes], metadata: Iterable[dict]): - responses must have status code 206 for more responses coming and 204 for the last response already sent """ - pipeline = Pipeline(*pipeline_head(url), rcompose(PickupStreamer(), RestReceiver())) + pipeline = Pipeline(*pipeline_head(url), rcompose(RestPickupStreamer(), RestReceiver())) yield from pipeline(data, metadata)