Skip to content

Commit 750a9fb

Browse files
committed
Parametrise the value to yield instead of None when there is none
1 parent 5ddbe85 commit 750a9fb

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

click_async_plugins/itc.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,7 @@ async def updates(
4040
yield_immediately: bool = True,
4141
timeout: float | None = None,
4242
at_most_every: float | None = None,
43+
yield_for_no_value: Any | None = None,
4344
) -> AsyncGenerator[Any | None]:
4445
at_most_every = 0 if at_most_every is None else at_most_every
4546

@@ -55,7 +56,7 @@ async def updates(
5556
self._events[key].append(event)
5657

5758
if yield_immediately:
58-
yield self._objects.get(key)
59+
yield self._objects.get(key, yield_for_no_value)
5960

6061
try:
6162
timestamp = 0.0
@@ -71,7 +72,7 @@ async def updates(
7172
logger.debug(f"Too early, sleeping for {waitremain:.02f}s")
7273
await asyncio.sleep(waitremain)
7374

74-
yield self._objects.get(key)
75+
yield self._objects.get(key, yield_for_no_value)
7576
event.clear()
7677
timestamp = time.monotonic()
7778

0 commit comments

Comments
 (0)