在退出之前，如何在python中使用websockets创建3次尝试？

import asyncio
import websockets

async def connect_with_retry(uri, max_retries=3):
    retries = 0
    while retries < max_retries:
        try:
            async with websockets.connect(uri) as websocket:
                print("WebSocket连接成功！")
                # 在这里进行WebSocket通信操作
                break
        except websockets.exceptions.ConnectionError:
            print(f"WebSocket连接失败，正在进行第 {retries+1} 次重试...")
            retries += 1
            await asyncio.sleep(1)  # 等待1秒后进行下一次尝试
    else:
        print("WebSocket连接尝试次数已达到上限，无法连接。")

# 使用示例
websocket_uri = "ws://example.com/websocket"
asyncio.get_event_loop().run_until_complete(connect_with_retry(websocket_uri))