diff --git a/sukuinote/plugins/loli.py b/sukuinote/plugins/loli.py index 6388f98..57a183f 100644 --- a/sukuinote/plugins/loli.py +++ b/sukuinote/plugins/loli.py @@ -11,14 +11,17 @@ async def loli(client, message): query = ' '.join(message.command) results = await client.get_inline_bot_results(bot.username or bot.id, query) result = results.results[0] + to_reply = message + if not getattr(message.reply_to_message, 'empty', True): + to_reply = message.reply_to_message if result.type != 'photo': - await message.reply_text({'message': result.send_message.message, 'entities': result.send_message.entities}, parse_mode='through') + await to_reply.reply_text({'message': result.send_message.message, 'entities': result.send_message.entities}, parse_mode='through') return photo = Photo._parse(client, result.photo) try: - await message.reply_cached_media(photo.file_id, caption={'message': result.send_message.message, 'entities': result.send_message.entities}, parse_mode='through') + await to_reply.reply_cached_media(photo.file_id, caption={'message': result.send_message.message, 'entities': result.send_message.entities}, parse_mode='through') except Forbidden: - await message.reply_text({'message': result.send_message.message, 'entities': result.send_message.entities}, parse_mode='through') + await to_reply.reply_text({'message': result.send_message.message, 'entities': result.send_message.entities}, parse_mode='through') help_dict['loli'] = ('Loli', '''{prefix}loli [keywords] - Gets a possibly nsfw image of a loli, thanks to lolicon.app