Skip to content

Commit c7f3865

Browse files
committed
Add BaseHypeTrain type
1 parent 0410318 commit c7f3865

File tree

2 files changed

+14
-28
lines changed

2 files changed

+14
-28
lines changed

twitchio/models/eventsub_.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4823,9 +4823,7 @@ class BaseHypeTrain(BaseEvent):
48234823
"type",
48244824
)
48254825

4826-
def __init__(
4827-
self, payload: HypeTrainBeginEvent | HypeTrainProgressEvent | HypeTrainEndEvent, *, http: HTTPClient
4828-
) -> None:
4826+
def __init__(self, payload: BaseHypeTrainEvent, *, http: HTTPClient) -> None:
48294827
self.broadcaster: PartialUser = PartialUser(
48304828
payload["broadcaster_user_id"], payload["broadcaster_user_login"], payload["broadcaster_user_name"], http=http
48314829
)

twitchio/types_/eventsub.py

Lines changed: 13 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,7 @@
4141
"AutomodTermsUpdateEvent",
4242
"BaseChannelPointsRewardData",
4343
"BaseEmoteData",
44+
"BaseHypeTrainEvent",
4445
"ChannelAdBreakBeginEvent",
4546
"ChannelBanEvent",
4647
"ChannelBitsUseEvent",
@@ -1232,47 +1233,34 @@ class HypeTrainSharedParticipants(TypedDict):
12321233
broadcaster_user_name: str
12331234

12341235

1235-
class HypeTrainBeginEvent(BaseBroadcasterEvent):
1236+
class BaseHypeTrainEvent(BaseBroadcasterEvent):
12361237
id: str
12371238
total: int
1239+
level: int
1240+
started_at: str
1241+
top_contributions: list[HypeTrainContributionData]
1242+
shared_train_participants: list[HypeTrainSharedParticipants]
1243+
type: Literal["treasure", "golden_kappa", "regular"]
1244+
is_shared_train: bool
1245+
1246+
1247+
class HypeTrainBeginEvent(BaseHypeTrainEvent):
12381248
progress: int
12391249
goal: int
1240-
top_contributions: list[HypeTrainContributionData]
1241-
level: int
12421250
all_time_high_level: int
12431251
all_time_high_total: int
1244-
shared_train_participants: list[HypeTrainSharedParticipants]
1245-
started_at: str
12461252
expires_at: str
1247-
type: Literal["treasure", "golden_kappa", "regular"]
1248-
is_shared_train: bool
12491253

12501254

1251-
class HypeTrainProgressEvent(BaseBroadcasterEvent):
1252-
id: str
1253-
total: int
1255+
class HypeTrainProgressEvent(BaseHypeTrainEvent):
12541256
progress: int
12551257
goal: int
1256-
top_contributions: list[HypeTrainContributionData]
1257-
level: int
1258-
shared_train_participants: list[HypeTrainSharedParticipants]
1259-
started_at: str
12601258
expires_at: str
1261-
type: Literal["treasure", "golden_kappa", "regular"]
1262-
is_shared_train: bool
12631259

12641260

1265-
class HypeTrainEndEvent(BaseBroadcasterEvent):
1266-
id: str
1267-
total: int
1268-
top_contributions: list[HypeTrainContributionData]
1269-
level: int
1270-
shared_train_participants: list[HypeTrainSharedParticipants]
1271-
started_at: str
1261+
class HypeTrainEndEvent(BaseHypeTrainEvent):
12721262
ended_at: str
12731263
cooldown_ends_at: str
1274-
type: Literal["treasure", "golden_kappa", "regular"]
1275-
is_shared_train: bool
12761264

12771265

12781266
class ShieldModeBeginEvent(BroadcasterModeratorEvent):

0 commit comments

Comments
 (0)