Python异步编程中解决同步阻塞与多服务集成指南

pip install vkreal

import asyncio
import discord
from discord.ext import commands
import vkreal # 导入异步VK库

# Discord Bot的初始化与命令定义
client = commands.Bot(command_prefix='!', intents=discord.Intents.all())

@client.event
async def on_ready():
    print('Discord Bot已连接！')

@client.event
async def on_message(message):
    if message.author == client.user:
        return
    await client.process_commands(message) # 确保处理Discord命令

@client.command(pass_context=True)
async def hi(ctx: commands.Context):
    await ctx.send('你好！')

# VK API的凭证信息 (请替换为你的实际信息)
VK_LOGIN = 'your_vk_login'
VK_PASSWORD = 'your_vk_password'
VK_APP_ID = 'your_vk_app_id' # 如果需要，或者直接使用token
VK_CHAT_ID = 123456789 # 替换为你的VK聊天ID

# 使用vkreal进行VK会话初始化
# 注意：vkreal通常推荐使用access_token进行认证，而不是用户名密码
# 如果你必须使用用户名密码，vkreal也支持，但获取token是更推荐的方式
# 假设你已经有了一个VK access token
VK_ACCESS_TOKEN = 'your_vk_access_token' # 替换为你的VK访问令牌

# 初始化vkreal会话和Long Poll
session = vkreal.VkApi(token=VK_ACCESS_TOKEN)
vk = session.api_context()
longpoll = vkreal.VkLongPoll(session, loop=asyncio.get_event_loop()) # 传入事件循环

async def vk_longpoll_listener():
    """
    异步监听VK Long Poll事件，并将消息转发到Discord。
    """
    print('开始监听VK Long Poll事件...')
    async for event in longpoll.listen(): # 关键：使用async for进行异步迭代
        # 打印事件类型，用于调试
        # print(f"VK Event Type: {event['type']}")

        # 根据你的原始逻辑处理VK消息事件
        if event['type'] == vkreal.VkEventType.MESSAGE_NEW and event['from_chat'] and event['chat_id'] == VK_CHAT_ID:
            user_id = event['user_id']
            message_text = event['text']
            attachments = event.get('attachments', {}) # 使用.get()避免KeyError

            # 获取用户信息 (vkreal的API调用也是异步的)
            user_info_list = await vk.users.get(user_ids=user_id)
            user_info = user_info_list[0]
            user_name = f"{user_info['first_name']} {user_info['last_name']}"

            # 假设Discord转发的目标频道ID
            DISCORD_CHANNEL_ID = 123456789012345678 # 替换为你的Discord频道ID
            await client.wait_until_ready() # 确保Discord客户端已准备好
            channel = client.get_channel(DISCORD_CHANNEL_ID)

            if channel:
                # 检查消息中是否包含@all或@everyone
                is_everyone_mention = '@all' in message_text or '@everyone' in message_text
                # 检查是否有附件（vkreal的附件结构可能与vk_api略有不同，需要根据实际情况调整）
                has_attachment = bool(attachments) # 简单判断是否有附件

                if is_everyone_mention:
                    if has_attachment:
                        await channel.send(f"{user_name} » {message_text} [附件] @everyone")
                    else:
                        await channel.send(f"{user_name} » {message_text} @everyone")
                else:
                    if has_attachment:
                        await channel.send(f"{user_name} » {message_text} [附件]")
                    else:
                        await channel.send(f"{user_name} » {message_text}")
            else:
                print(f"错误：未找到Discord频道 ID: {DISCORD_CHANNEL_ID}")
        else:
            # 可以根据需要处理其他VK事件
            pass

async def main():
    """
    主函数，同时运行Discord Bot和VK Long Poll监听任务。
    """
    async with client:
        # 在Discord Bot启动前，创建VK监听任务并将其添加到事件循环
        client.loop.create_task(vk_longpoll_listener())
        # 启动Discord Bot
        await client.start('YOUR_DISCORD_BOT_TOKEN') # 替换为你的Discord Bot令牌

if __name__ == '__main__':
    # 运行主异步函数
    asyncio.run(main())