import csv

def insertTable(csv_file, table_name, output_file):
 # 读取 CSV 文件
 with open(csv_file, newline='', encoding='utf-8') as csvfile:
 reader = csv.DictReader(csvfile)
 columns = reader.fieldnames # 获取列名
 data = []

 for row in reader:
 if any(row.values()): # 检查行中是否有数据
 data.append(row)
 else:
 break # 如果遇到空行,停止读取

 # 生成 INSERT 语句
 insert_values = []
 for row in data:
 values = ', '.join([f"'{row[col]}'" for col in columns]) # 每行数据值
 insert_values.append(f"({values})")

 # 将 INSERT 语句保存到文件
 with open(output_file, 'w') as f:
 f.write(f"INSERT INTO {table_name} ({', '.join(columns)}) VALUES ")
 f.write(', '.join(insert_values))
 f.write(';')

 print(f"INSERT 语句已保存到 {output_file}")

if __name__ == '__main__':
 insertTable('../DataSet/test/Location1.csv', 'Location1', 'insert_data_script.txt')