在python中sql连接失败后如何重试？

import time
import pymysql

def connect_to_database():
    connection = None
    max_retries = 3
    retry_count = 0
    
    while retry_count < max_retries:
        try:
            connection = pymysql.connect(host='localhost', user='username', password='password', database='database')
            print("Connected to the database!")
            break
        except pymysql.Error as e:
            print(f"Failed to connect to the database: {e}")
            retry_count += 1
            print(f"Retrying in 5 seconds... (Attempt {retry_count}/{max_retries})")
            time.sleep(5)
    
    if connection is None:
        print("Failed to connect to the database after multiple retries.")
    else:
        # 执行数据库操作
        # ...
        connection.close()

connect_to_database()