Skip to content

Commit 92dc142

Browse files
committed
update typings for customreward PR
1 parent be1eafd commit 92dc142

File tree

1 file changed

+13
-14
lines changed

1 file changed

+13
-14
lines changed

twitchio/ext/eventsub/models.py

Lines changed: 13 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -4,8 +4,7 @@
44
import hashlib
55
import logging
66
from enum import Enum
7-
from typing import Dict, TYPE_CHECKING, Optional, Type, Union
8-
from typing_extensions import Literal
7+
from typing import Dict, TYPE_CHECKING, Optional, Type, Union, Literal, Tuple
98

109
from aiohttp import web
1110

@@ -456,33 +455,33 @@ def __init__(self, data, broadcaster):
456455
self.cost: int = data["cost"]
457456
self.prompt: str = data["prompt"]
458457

459-
self.enabled: bool = data.get("is_enabled", None)
460-
self.paused: bool = data.get("is_paused", None)
461-
self.in_stock: bool = data.get("is_in_stock", None)
458+
self.enabled: Optional[bool] = data.get("is_enabled", None)
459+
self.paused: Optional[bool] = data.get("is_paused", None)
460+
self.in_stock: Optional[bool] = data.get("is_in_stock", None)
462461

463-
self.cooldown_until: datetime.datetime = (
462+
self.cooldown_until: Optional[datetime.datetime] = (
464463
_parse_datetime(data["cooldown_expires_at"]) if data.get("cooldown_expires_at", None) else None
465464
)
466465

467-
self.input_required: bool = data.get("is_user_input_required", None)
468-
self.redemptions_skip_queue: bool = data.get("should_redemptions_skip_request_queue", None)
469-
self.redemptions_current_stream: bool = data.get("redemptions_redeemed_current_stream", None)
466+
self.input_required: Optional[bool] = data.get("is_user_input_required", None)
467+
self.redemptions_skip_queue: Optional[bool] = data.get("should_redemptions_skip_request_queue", None)
468+
self.redemptions_current_stream: Optional[bool] = data.get("redemptions_redeemed_current_stream", None)
470469

471-
self.max_per_stream: tuple[bool, int] = (
470+
self.max_per_stream: Tuple[Optional[bool], Optional[int]] = (
472471
data.get("max_per_stream", {}).get("is_enabled"),
473472
data.get("max_per_stream", {}).get("value"),
474473
)
475-
self.max_per_user_stream: tuple[bool, int] = (
474+
self.max_per_user_stream: Tuple[Optional[bool], Optional[int]] = (
476475
data.get("max_per_user_per_stream", {}).get("is_enabled"),
477476
data.get("max_per_user_per_stream", {}).get("value"),
478477
)
479-
self.cooldown: tuple[bool, int] = (
478+
self.cooldown: Tuple[Optional[bool], Optional[int]] = (
480479
data.get("global_cooldown", {}).get("is_enabled"),
481480
data.get("global_cooldown", {}).get("seconds"),
482481
)
483482

484-
self.background_color: str = data.get("background_color", None)
485-
self.image: str = data.get("image", data.get("default_image", {})).get("url_1x", None)
483+
self.background_color: Optional[str] = data.get("background_color", None)
484+
self.image: Optional[str] = data.get("image", data.get("default_image", {})).get("url_1x", None)
486485

487486

488487
class CustomRewardAddUpdateRemoveData(EventData):

0 commit comments

Comments
 (0)