14
14
from . import logging
15
15
from .api import BotAPI
16
16
from .robot import Robot
17
- from .types import gateway , channel , guild , user , session , reaction , interaction , forum
17
+ from .types import session
18
18
19
19
_log = logging .get_logger ()
20
20
@@ -36,7 +36,7 @@ def __init__(
36
36
):
37
37
self .dispatch = dispatch
38
38
self .state = ConnectionState (dispatch , api )
39
- self .parser : Dict [str , Callable [[gateway . WsContext , Any ], None ]] = self .state .parsers
39
+ self .parser : Dict [str , Callable [[dict ], None ]] = self .state .parsers
40
40
41
41
self ._connect = connect
42
42
self ._max_async = max_async
@@ -89,141 +89,141 @@ def __init__(self, dispatch: Callable, api: BotAPI):
89
89
self ._dispatch = dispatch
90
90
self .api = api
91
91
92
- def parse_ready (self , ctx : gateway . WsContext , data : gateway . ReadyEvent ):
92
+ def parse_ready (self , payload ):
93
93
self ._dispatch ("ready" )
94
94
95
- def parse_resumed (self , ctx : gateway . WsContext , data : gateway . ReadyEvent ):
95
+ def parse_resumed (self , payload ):
96
96
self ._dispatch ("resumed" )
97
97
98
98
# botpy.flags.Intents.guilds
99
- def parse_guild_create (self , ctx : gateway . WsContext , data : guild . GuildPayload ):
100
- _guild = Guild (self .api , ctx , data )
99
+ def parse_guild_create (self , payload ):
100
+ _guild = Guild (self .api , payload . get ( 'id' , None ), payload . get ( 'd' , {}) )
101
101
self ._dispatch ("guild_create" , _guild )
102
102
103
- def parse_guild_update (self , ctx : gateway . WsContext , data : guild . GuildPayload ):
104
- _guild = Guild (self .api , ctx , data )
103
+ def parse_guild_update (self , payload ):
104
+ _guild = Guild (self .api , payload . get ( 'id' , None ), payload . get ( 'd' , {}) )
105
105
self ._dispatch ("guild_update" , _guild )
106
106
107
- def parse_guild_delete (self , ctx : gateway . WsContext , data : guild . GuildPayload ):
108
- _guild = Guild (self .api , ctx , data )
107
+ def parse_guild_delete (self , payload ):
108
+ _guild = Guild (self .api , payload . get ( 'id' , None ), payload . get ( 'd' , {}) )
109
109
self ._dispatch ("guild_delete" , _guild )
110
110
111
- def parse_channel_create (self , ctx : gateway . WsContext , data : channel . ChannelPayload ):
112
- _channel = Channel (self .api , ctx , data )
111
+ def parse_channel_create (self , payload ):
112
+ _channel = Channel (self .api , payload . get ( 'id' , None ), payload . get ( 'd' , {}) )
113
113
self ._dispatch ("channel_create" , _channel )
114
114
115
- def parse_channel_update (self , ctx : gateway . WsContext , data : channel . ChannelPayload ):
116
- _channel = Channel (self .api , ctx , data )
115
+ def parse_channel_update (self , payload ):
116
+ _channel = Channel (self .api , payload . get ( 'id' , None ), payload . get ( 'd' , {}) )
117
117
self ._dispatch ("channel_update" , _channel )
118
118
119
- def parse_channel_delete (self , ctx : gateway . WsContext , data : channel . ChannelPayload ):
120
- _channel = Channel (self .api , ctx , data )
119
+ def parse_channel_delete (self , payload ):
120
+ _channel = Channel (self .api , payload . get ( 'id' , None ), payload . get ( 'd' , {}) )
121
121
self ._dispatch ("channel_delete" , _channel )
122
122
123
123
# botpy.flags.Intents.guild_members
124
- def parse_guild_member_add (self , ctx : gateway . WsContext , data : user . GuildMemberPayload ):
125
- _member = Member (self .api , ctx , data )
124
+ def parse_guild_member_add (self , payload ):
125
+ _member = Member (self .api , payload . get ( 'id' , None ), payload . get ( 'd' , {}) )
126
126
self ._dispatch ("guild_member_add" , _member )
127
127
128
- def parse_guild_member_update (self , ctx : gateway . WsContext , data : user . GuildMemberPayload ):
129
- _member = Member (self .api , ctx , data )
128
+ def parse_guild_member_update (self , payload ):
129
+ _member = Member (self .api , payload . get ( 'id' , None ), payload . get ( 'd' , {}) )
130
130
self ._dispatch ("guild_member_update" , _member )
131
131
132
- def parse_guild_member_remove (self , ctx : gateway . WsContext , data : user . GuildMemberPayload ):
133
- _member = Member (self .api , ctx , data )
132
+ def parse_guild_member_remove (self , payload ):
133
+ _member = Member (self .api , payload . get ( 'id' , None ), payload . get ( 'd' , {}) )
134
134
self ._dispatch ("guild_member_remove" , _member )
135
135
136
136
# botpy.flags.Intents.guild_messages
137
- def parse_message_create (self , ctx : gateway . WsContext , data : gateway . MessagePayload ):
138
- _message = Message (self .api , ctx , data )
137
+ def parse_message_create (self , payload ):
138
+ _message = Message (self .api , payload . get ( 'id' , None ), payload . get ( 'd' , {}) )
139
139
self ._dispatch ("message_create" , _message )
140
140
141
- def parse_message_delete (self , ctx : gateway . WsContext , data : gateway . MessagePayload ):
142
- _message = Message (self .api , ctx , data )
141
+ def parse_message_delete (self , payload ):
142
+ _message = Message (self .api , payload . get ( 'id' , None ), payload . get ( 'd' , {}) )
143
143
self ._dispatch ("message_delete" , _message )
144
144
145
145
# botpy.flags.Intents.guild_message_reactions
146
- def parse_message_reaction_add (self , ctx : gateway . WsContext , data : reaction . Reaction ):
147
- _reaction = Reaction (self .api , ctx , data )
146
+ def parse_message_reaction_add (self , payload ):
147
+ _reaction = Reaction (self .api , payload . get ( 'id' , None ), payload . get ( 'd' , {}) )
148
148
self ._dispatch ("message_reaction_add" , _reaction )
149
149
150
- def parse_message_reaction_remove (self , ctx : gateway . WsContext , data : reaction . Reaction ):
151
- _reaction = Reaction (self .api , ctx , data )
150
+ def parse_message_reaction_remove (self , payload ):
151
+ _reaction = Reaction (self .api , payload . get ( 'id' , None ), payload . get ( 'd' , {}) )
152
152
self ._dispatch ("message_reaction_remove" , _reaction )
153
153
154
154
# botpy.flags.Intents.direct_message
155
- def parse_direct_message_create (self , ctx : gateway . WsContext , data : gateway . DirectMessagePayload ):
156
- _message = DirectMessage (self .api , ctx , data )
155
+ def parse_direct_message_create (self , payload ):
156
+ _message = DirectMessage (self .api , payload . get ( 'id' , None ), payload . get ( 'd' , {}) )
157
157
self ._dispatch ("direct_message_create" , _message )
158
158
159
- def parse_direct_message_delete (self , ctx : gateway . WsContext , data : gateway . DirectMessagePayload ):
160
- _message = DirectMessage (self .api , ctx , data )
159
+ def parse_direct_message_delete (self , payload ):
160
+ _message = DirectMessage (self .api , payload . get ( 'id' , None ), payload . get ( 'd' , {}) )
161
161
self ._dispatch ("direct_message_delete" , _message )
162
162
163
163
# botpy.flags.Intents.interaction
164
- def parse_interaction_create (self , ctx : gateway . WsContext , data : interaction . InteractionPayload ):
165
- _interaction = Interaction (self .api , ctx , data )
164
+ def parse_interaction_create (self , payload ):
165
+ _interaction = Interaction (self .api , payload . get ( 'id' , None ), payload . get ( 'd' , {}) )
166
166
self ._dispatch ("interaction_create" , _interaction )
167
167
168
168
# botpy.flags.Intents.message_audit
169
- def parse_message_audit_pass (self , ctx : gateway . WsContext , data : gateway . MessageAuditPayload ):
170
- _message_audit = MessageAudit (self .api , ctx , data )
169
+ def parse_message_audit_pass (self , payload ):
170
+ _message_audit = MessageAudit (self .api , payload . get ( 'id' , None ), payload . get ( 'd' , {}) )
171
171
self ._dispatch ("message_audit_pass" , _message_audit )
172
172
173
- def parse_message_audit_reject (self , ctx : gateway . WsContext , data : gateway . MessageAuditPayload ):
174
- _message_audit = MessageAudit (self .api , ctx , data )
173
+ def parse_message_audit_reject (self , payload ):
174
+ _message_audit = MessageAudit (self .api , payload . get ( 'id' , None ), payload . get ( 'd' , {}) )
175
175
self ._dispatch ("message_audit_reject" , _message_audit )
176
176
177
177
# botpy.flags.Intents.audio_action
178
- def parse_audio_start (self , ctx : gateway . WsContext , data ):
179
- _audio = Audio (self .api , ctx , data )
178
+ def parse_audio_start (self , payload ):
179
+ _audio = Audio (self .api , payload . get ( 'id' , None ), payload . get ( 'd' , {}) )
180
180
self ._dispatch ("audio_start" , _audio )
181
181
182
- def parse_audio_finish (self , ctx : gateway . WsContext , data ):
183
- _audio = Audio (self .api , ctx , data )
182
+ def parse_audio_finish (self , payload ):
183
+ _audio = Audio (self .api , payload . get ( 'id' , None ), payload . get ( 'd' , {}) )
184
184
self ._dispatch ("audio_finish" , _audio )
185
185
186
- def parse_on_mic (self , ctx : gateway . WsContext , data ):
187
- _audio = Audio (self .api , ctx , data )
186
+ def parse_on_mic (self , payload ):
187
+ _audio = Audio (self .api , payload . get ( 'id' , None ), payload . get ( 'd' , {}) )
188
188
self ._dispatch ("on_mic" , _audio )
189
189
190
- def parse_off_mic (self , ctx : gateway . WsContext , data ):
191
- _audio = Audio (self .api , ctx , data )
190
+ def parse_off_mic (self , payload ):
191
+ _audio = Audio (self .api , payload . get ( 'id' , None ), payload . get ( 'd' , {}) )
192
192
self ._dispatch ("off_mic" , _audio )
193
193
194
194
# botpy.flags.Intents.public_guild_messages
195
- def parse_at_message_create (self , ctx : gateway . WsContext , data : gateway . MessagePayload ):
196
- _message = Message (self .api , ctx , data )
195
+ def parse_at_message_create (self , payload ):
196
+ _message = Message (self .api , payload . get ( 'id' , None ), payload . get ( 'd' , {}) )
197
197
self ._dispatch ("at_message_create" , _message )
198
198
199
- def parse_public_message_delete (self , ctx : gateway . WsContext , data : gateway . MessagePayload ):
200
- _message = Message (self .api , ctx , data )
199
+ def parse_public_message_delete (self , payload ):
200
+ _message = Message (self .api , payload . get ( 'id' , None ), payload . get ( 'd' , {}) )
201
201
self ._dispatch ("public_message_delete" , _message )
202
202
203
203
# botpy.flags.Intents.forums
204
- def parse_forum_thread_create (self , ctx : gateway . WsContext , data : forum . Thread ):
205
- _forum = Thread (self .api , ctx , data )
204
+ def parse_forum_thread_create (self , payload ):
205
+ _forum = Thread (self .api , payload . get ( 'id' , None ), payload . get ( 'd' , {}) )
206
206
self ._dispatch ("forum_thread_create" , _forum )
207
207
208
- def parse_forum_thread_update (self , ctx : gateway . WsContext , data : forum . Thread ):
209
- _forum = Thread (self .api , ctx , data )
208
+ def parse_forum_thread_update (self , payload ):
209
+ _forum = Thread (self .api , payload . get ( 'id' , None ), payload . get ( 'd' , {}) )
210
210
self ._dispatch ("forum_thread_update" , _forum )
211
211
212
- def parse_forum_thread_delete (self , ctx : gateway . WsContext , data : forum . Thread ):
213
- _forum = Thread (self .api , ctx , data )
212
+ def parse_forum_thread_delete (self , payload ):
213
+ _forum = Thread (self .api , payload . get ( 'id' , None ), payload . get ( 'd' , {}) )
214
214
self ._dispatch ("forum_thread_delete" , _forum )
215
215
216
- def parse_forum_post_create (self , ctx : gateway . WsContext , data : forum . Post ):
217
- self ._dispatch ("forum_post_create" , data )
216
+ def parse_forum_post_create (self , payload ):
217
+ self ._dispatch ("forum_post_create" , payload . get ( 'd' , {}) )
218
218
219
- def parse_forum_post_delete (self , ctx : gateway . WsContext , data : forum . Post ):
220
- self ._dispatch ("forum_post_delete" , data )
219
+ def parse_forum_post_delete (self , payload ):
220
+ self ._dispatch ("forum_post_delete" , payload . get ( 'd' , {}) )
221
221
222
- def parse_forum_reply_create (self , ctx : gateway . WsContext , data : forum . Reply ):
223
- self ._dispatch ("forum_reply_create" , data )
222
+ def parse_forum_reply_create (self , payload ):
223
+ self ._dispatch ("forum_reply_create" , payload . get ( 'd' , {}) )
224
224
225
- def parse_forum_reply_delete (self , ctx : gateway . WsContext , data : forum . Reply ):
226
- self ._dispatch ("forum_reply_delete" , data )
225
+ def parse_forum_reply_delete (self , payload ):
226
+ self ._dispatch ("forum_reply_delete" , payload . get ( 'd' , {}) )
227
227
228
- def parse_forum_publish_audit_result (self , ctx : gateway . WsContext , data : forum . AuditResult ):
229
- self ._dispatch ("forum_publish_audit_result" , data )
228
+ def parse_forum_publish_audit_result (self , payload ):
229
+ self ._dispatch ("forum_publish_audit_result" , payload . get ( 'd' , {}) )
0 commit comments