对csv的Sqlalchemy查询

from sqlalchemy import create_engine, Table, Column, Integer, String, MetaData, select

engine = create_engine('sqlite:///data.db')  # 数据库引擎，这里使用SQLite作为示例
metadata = MetaData()

csv_table = Table('csv_data', metadata,
                  Column('id', Integer, primary_key=True),
                  Column('name', String),
                  Column('age', Integer),
                  Column('city', String)
                  )

with open('data.csv', 'r') as file:
    connection = engine.connect()
    for line in file:
        data = line.strip().split(',')  # 假设CSV文件的字段以逗号分隔
        query = csv_table.insert().values(name=data[0], age=int(data[1]), city=data[2])
        connection.execute(query)
    connection.close()

connection = engine.connect()
query = select([csv_table])
result = connection.execute(query)
for row in result:
    print(row)
connection.close()