mysql 导入显示进度

pv data.csv | mysqlimport -u username -p database_name table_name

import mysql.connector
import time

def load_data_with_progress(file_path, table_name):
    conn = mysql.connector.connect(user='username', password='password', host='localhost', database='database_name')
    cursor = conn.cursor()
    
    # 获取总行数
    cursor.execute(f"SELECT COUNT(*) FROM {table_name}")
    total_rows = cursor.fetchone()[0]
    
    # 开始导入
    start_time = time.time()
    with open(file_path, 'r') as file:
        cursor.execute(f"LOAD DATA INFILE '{file_path}' INTO TABLE {table_name}")
        while True:
            rows_loaded = cursor.rowcount
            progress = (rows_loaded / total_rows) * 100
            print(f"Progress: {progress:.2f}%")
            if rows_loaded == total_rows:
                break
            time.sleep(1)
    
    end_time = time.time()
    print(f"Total time taken: {end_time - start_time:.2f} seconds")
    
    cursor.close()
    conn.close()

load_data_with_progress('data.csv', 'table_name')