@@ -194,17 +194,27 @@ async def reaction(
194194 ):
195195 assert isinstance (bot , KBot )
196196 if isinstance (mid , str ):
197+ msg_id = mid
198+ else :
199+ _mid : MessageCreateReturn = cast (MessageCreateReturn , mid )
200+ if not _mid .msg_id :
201+ return
202+ msg_id = _mid .msg_id
203+ if isinstance (context , Target ):
204+ if context .private :
205+ if delete :
206+ return await bot .directMessage_deleteReaction (msg_id = msg_id , emoji = emoji .name or emoji .id )
207+ return await bot .directMessage_addReaction (msg_id = msg_id , emoji = emoji .name or emoji .id )
197208 if delete :
198- await bot .message_deleteReaction (msg_id = mid , emoji = emoji .name or emoji .id )
199- else :
200- await bot . message_addReaction ( msg_id = mid , emoji = emoji . name or emoji . id )
201- _mid : MessageCreateReturn = cast ( MessageCreateReturn , mid )
202- if not _mid . msg_id :
203- return
209+ return await bot .message_deleteReaction (msg_id = msg_id , emoji = emoji .name or emoji .id )
210+ return await bot . message_addReaction ( msg_id = msg_id , emoji = emoji . name or emoji . id )
211+ if isinstance ( context , PrivateMessageEvent ):
212+ if delete :
213+ return await bot . directMessage_deleteReaction ( msg_id = msg_id , emoji = emoji . name or emoji . id )
214+ return await bot . directMessage_addReaction ( msg_id = msg_id , emoji = emoji . name or emoji . id )
204215 if delete :
205- await bot .message_deleteReaction (msg_id = _mid .msg_id , emoji = emoji .name or emoji .id )
206- else :
207- await bot .message_addReaction (msg_id = _mid .msg_id , emoji = emoji .name or emoji .id )
216+ return await bot .message_deleteReaction (msg_id = msg_id , emoji = emoji .name or emoji .id )
217+ return await bot .message_addReaction (msg_id = msg_id , emoji = emoji .name or emoji .id )
208218
209219 def get_reply (self , mid : Any ):
210220 _mid : MessageCreateReturn = cast (MessageCreateReturn , mid )
0 commit comments