mysql 表数据生成insert脚本

CREATE TABLE users (
    id INT AUTO_INCREMENT PRIMARY KEY,
    name VARCHAR(100),
    email VARCHAR(100)
);

INSERT INTO users (name, email) VALUES ('Alice', 'alice@example.com');

INSERT INTO users (name, email) VALUES 
('Bob', 'bob@example.com'),
('Charlie', 'charlie@example.com'),
('David', 'david@example.com');

name,email
Alice,alice@example.com
Bob,bob@example.com
Charlie,charlie@example.com
David,david@example.com

import csv

def generate_insert_script(csv_file, table_name):
    with open(csv_file, newline='') as csvfile:
        reader = csv.DictReader(csvfile)
        script = f"INSERT INTO {table_name} (name, email) VALUES\n"
        for row in reader:
            script += f"('{row['name']}', '{row['email']}'),\n"
        script = script.rstrip(',\n') + ";"
    return script

csv_file = 'users.csv'
table_name = 'users'
insert_script = generate_insert_script(csv_file, table_name)
print(insert_script)