如何使用python將文本文件導入SQLite3表

Question

我目前有電影的txt文件，其中包含movieID，電影名稱和類型。 例：

1,Toy Story (1995),Animation|Children's|Comedy
2,Jumanji (1995),Adventure|Children's|Fantasy
3,Grumpier Old Men (1995),Comedy|Romance
4,Waiting to Exhale (1995),Comedy|Drama
5,Father of the Bride Part II (1995),Comedy

我正在嘗試將此文件導入到使用Python創建的SQLite3表中：

import sqlite3

conn = sqlite3.connect('CSC 452 Final.db')
c=conn.cursor()

刪除表

c.execute('''DROP TABLE IF EXISTS temp_movie''')
print("Table temp_movie dropped.")

為電影創建表

c.execute('''CREATE TABLE temp_movie (
    MovieID          NUMERIC (5),
    movie_title_year VARCHAR2 (255),
    movie_categories VARCHAR2 (255),
    CONSTRAINT movies_movieID_PK PRIMARY KEY (
        MovieID
    ))''')
print("Table temp_movie successfully created.")
conn.commit()

將分隔符為'，'的文本文件導入此數據庫的最佳方法是什么？

Answer 1

你可以在閱讀txt文件和分裂的, ：

file_data = [i.strip('\n').split(',') for i in open('filename.txt')]
new_data = [[int(a), *b] for a, *b in file_data] #convert string id to integer
conn = sqlite3.connect('CSC 452 Final.db')
c = conn.cursor()
c.executemany('INSERT INTO temp_movie VALUES (?, ?, ?)', new_data)

Answer 2

我本來可以使用python csv模塊來做到這一點的：

with open(<txt file>,'rb') as f: 
    dic_reader = csv.DictReader(f)
    value_list = [(each_row['MovieID'], each_row['movie_title_year'], each_row['movie_title_year']) for each_row in dic_reader]

c.executemany("INSERT INTO t (MovieID, movie_title_year, movie_title_year) VALUES (?, ?);", value_list)
conn.commit()
conn.close()