This commit is contained in:
Matthias Bisping 2022-05-17 12:12:43 +02:00
parent fb325ce43d
commit 47f1d77c03

View File

@ -1,13 +1,14 @@
from pyinfra.server.dispatcher.dispatcher import Dispatcher
from pyinfra.server.stream.queued_stream_function import QueuedStreamFunction
class QueuedStreamFunctionDispatcher(Dispatcher):
def __init__(self, queue):
self.queue = queue
def __init__(self, queued_stream_function: QueuedStreamFunction):
self.queued_stream_function = queued_stream_function
def patch(self, package):
self.queue.push(package)
return self.queue.pop()
self.queued_stream_function.push(package)
return self.queued_stream_function.pop()
def post(self, package):
return self.patch(package)