Överväg att bygga frågan dynamiskt för att säkerställa att antalet platshållare matchar din tabell och ditt CSV-filformat. Sedan är det bara att se till att din tabell och CSV-fil är korrekt, istället för att kontrollera att du skrivit tillräckligt med ?
platshållare i din kod.
Följande exempel antar
- CSV-filen innehåller kolumnnamn på första raden
- Anslutningen är redan byggd
- Filnamnet är
test.csv
- Tabellnamnet är
MyTable
- Python 3
...
with open ('test.csv', 'r') as f:
reader = csv.reader(f)
columns = next(reader)
query = 'insert into MyTable({0}) values ({1})'
query = query.format(','.join(columns), ','.join('?' * len(columns)))
cursor = connection.cursor()
for data in reader:
cursor.execute(query, data)
cursor.commit()
Om kolumnnamn inte ingår i filen:
...
with open ('test.csv', 'r') as f:
reader = csv.reader(f)
data = next(reader)
query = 'insert into MyTable values ({0})'
query = query.format(','.join('?' * len(data)))
cursor = connection.cursor()
cursor.execute(query, data)
for data in reader:
cursor.execute(query, data)
cursor.commit()