Pysqlite получить имена столбцов из CSV-файла

Question

У меня есть CSV-файл, из которого я пытаюсь загрузить данные в базу данных pysqlite. Я не могу найти способ извлечь первую строку файла и автоматически поместить ее в базу данных как заголовки столбцов таблицы. Я должен вводить имена «вручную» в самом коде, который подходит для 1-2 столбцов, но становится громоздким с десятками или сотнями столбцов. Вот мой код:

import sqlite3
import csv

f_n = 'test_data_1.csv'
f = open( f_n , 'r' )
csv_reader = csv.reader(f)
header = next( csv_reader )

sqlite_file = 'survey_test_db.sqlite'
table_name01 = 'test_survey_1'

field_01 = 'analyst_name'
field_type_01 = 'text'
field_02 = 'question_name'
field_type_02 = 'text'

conn = sqlite3.connect( sqlite_file )
c = conn.cursor()
c.execute('CREATE TABLE {tn}({nf_01} {ft_01},{nf_02} {ft_02})'\
.format(tn = table_name01 , nf_01 = field_01 , ft_01 = field_type_01, nf_02 = field_02 , ft_02 = field_type_02 ))

for row in csv_reader:
 c.execute("INSERT INTO test_survey_1 VALUES (?,?)",row)
f.close()

for row in c.execute('SELECT * FROM test_survey_1'):
 print(row)

conn.commit()
conn.close()

Answer 1

Вы можете использовать pandas, чтобы прочитать ваш CSV-файл в DataFrame, а затем экспортировать это к sqlite.

import sqlite3
import pandas as pd

sqlite_file = 'survey_test_db.sqlite'
table_name01 = 'test_survey_1'

conn = sqlite3.connect(sqlite_file)

pd.read_csv('test_data_1.csv').to_sql(table_name01, con=con)

Answer 2

c.execute('CREATE TABLE {tn}({fieldlist})'.format(
    tn=table_name01, 
    fieldlist=', '.join('{} TEXT'.format(name) for name in header),
))

Или используйте ORM, который разработан, чтобы облегчить подобные вещи. Пример SQLAlchemy:

t = Table(table_name01, meta, *(Column(name, String()) for name in header))
t.create()