Skip to content

Commit 76d25ba

Browse files
committed
Add multi_subscribe overloads
1 parent de64e48 commit 76d25ba

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

twitchio/client.py

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@
2929
import math
3030
from collections import defaultdict
3131
from types import MappingProxyType
32-
from typing import TYPE_CHECKING, Any, Literal, NamedTuple, Self, Unpack
32+
from typing import TYPE_CHECKING, Any, Literal, NamedTuple, Self, Unpack, overload
3333

3434
from .authentication import ManagedHTTPClient, Scopes, UserTokenPayload
3535
from .eventsub.enums import SubscriptionType
@@ -3533,6 +3533,16 @@ async def _multi_sub(self, subscriptions: list[SubscriptionPayload], *, stop_on_
35333533

35343534
return MultiSubscribePayload(success=success, errors=errors)
35353535

3536+
@overload
3537+
async def multi_subscribe(
3538+
self, subscriptions: list[SubscriptionPayload], *, wait: Literal[True] = True, stop_on_error: bool = False
3539+
) -> MultiSubscribePayload: ...
3540+
3541+
@overload
3542+
async def multi_subscribe(
3543+
self, subscriptions: list[SubscriptionPayload], *, wait: Literal[False] = False, stop_on_error: bool = False
3544+
) -> asyncio.Task[MultiSubscribePayload]: ...
3545+
35363546
async def multi_subscribe(
35373547
self, subscriptions: list[SubscriptionPayload], *, wait: bool = True, stop_on_error: bool = False
35383548
) -> MultiSubscribePayload | asyncio.Task[MultiSubscribePayload]:

0 commit comments

Comments
 (0)