import csv

def createTable(csv_file, table_name, output_file):
 # 读取 CSV 文件
 with open(csv_file, newline='', encoding='utf-8') as csvfile:
 reader = csv.reader(csvfile)
 columns = next(reader) # 获取第一行作为列名
 data = [row for row in reader] # 获取剩余行作为数据

 # 生成 CREATE TABLE 语句
 create_statement = f"CREATE TABLE {table_name} (\n"
 for col in columns:
 create_statement += f" {col} VARCHAR(255),\n" # 假设所有列类型为 VARCHAR(255)
 create_statement = create_statement.rstrip(',\n') + "\n);" # 移除最后一个逗号,并结束语句

 # 将 CREATE TABLE 语句保存到文件
 with open(output_file, 'w') as f:
 f.write(create_statement)

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


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