Marmelator/pyinfra
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 36 Wiki Activity

renaming

Browse Source
This commit is contained in:
Matthias Bisping 2022-07-04 11:46:53 +02:00
parent 3bf00d11cd
commit 36d8d3bc85
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
pyinfra/server/stream/queued_stream_function.py
View File

@ -4,13 +4,18 @@ from pyinfra.server.buffering.queue import stream_queue, Queue
class QueuedStreamFunction: class QueuedStreamFunction:
def __init__(self, fn): def __init__(self, stream_function):
"""Combines a stream function with a queue.
Args:
stream_function: Needs to operate on iterables.
"""
self.queue = Queue() self.queue = Queue()
self.fn = fn self.stream_function = stream_function
def push(self, item): def push(self, item):
self.queue.append(item) self.queue.append(item)
def pop(self): def pop(self):
items = stream_queue(self.queue) items = stream_queue(self.queue)
return first(self.fn(items)) return first(self.stream_function(items))