csvファイルの読み込みと挿入時のエラー

Question

前提・実現したいこと

入門Python3(オライリー・ジャパン)p272の復習課題8-7において、前問の8-6で作成したbooks.csvを読み出し、そのデータをbookテーブルに挿入したいです。ちなみに8-6は「sqlite3モジュールを用いて、books.dbというSQLiteデータベースを作り、その中にtitle(文字列）、'year'（整数）というフィールドを持つbookというテーブルを作ろう」です。正しく動作すれば以下のように表示されます。

#8-6(bookというテーブルを作成)
>>> import sqlite3
>>> db = sqlite3.connect('books.db')
>>> curs = db.cursor()
>>> curs.execute('''create table book (title text, author text, year int)''')
<sqlite3.Cursor object at 0x1090561f0>
>>> db.commit()
#8-7(book.csvを読み出し、そのデータをbookテーブルに挿入)
>>> import csv
>>> import sqlite3
>>> ins_str = 'insert into book values(?, ?, ?)'
>>> with open('books.csv', 'rt') as infile:
...     books = csv.DictReader(infile)
...     for book in books:
...             curs.execute(ins_str,  (book['title'], book['author'], book['year']))
... 
<sqlite3.Cursor object at 0x1007b21f0>
<sqlite3.Cursor object at 0x1007b21f0>
<sqlite3.Cursor object at 0x1007b21f0>
<sqlite3.Cursor object at 0x1007b21f0>
<sqlite3.Cursor object at 0x1007b21f0>

発生している問題・エラーメッセージ

以下のようにKeyErrorが発生します。

該当のソースコード

>>> import csv
>>> import sqlite3
>>> ins_str = 'insert into book values(?, ?, ?)'
>>> with open('books.csv', 'rt') as infile:
...     books = csv.DictReader(infile)
...     for book in books:
...             curs.execute(ins_str,  (book['title'], book['author'], book['year']))
... 
Traceback (most recent call last):
  File "<stdin>", line 4, in <module>
KeyError: 'title'

補足情報（FW/ツールのバージョンなど）

PCはMacを用いています。また、8-6の前問の8-5では以下のようにbooks.csvというCSVファイルを作成しています。

>>> text = '''title, author, year
... The Weirdstone of Brisingamen, Alan Garner, 1960
... Perdido Street Station, China Mieville, 2000
... Thud!, Terry Pratchett, 2005
... The Spellmen Files, Lisa Lutz, 2007
... Small Gods, Terry Pratchett, 1992
... '''
>>> with open('books.csv', 'wt') as outfile:
...     outfile.write(text)
... 
213

Comments

[kuma_kuma_]

一回CSVの4行目
Thud!
を
Thud
にして試してみれば？

[toast-uz]

エラーの前にprint(book)として、bookの内容を確認してください。'title'というキーが、何か別のものに変わっているようです。

[hamazon]

お二人ともご回答ありがとうございます。
８-５でCSVファイルを作成する際に、不要なスペースをカンマの後ろに入力していたのが原因だったようで、そちらを修正した所動きました！
初歩的なミスでした...お手数おかけしました。

Answer 1

８−５でCSVファイル作成の際にカンマの後ろに不要なスペースを入力していた事が原因か。以下のように訂正したら作動した。

>>> text = '''title,author,year
... The Weirdstone of Brisingamen,Alan Garner,1960
... Perdido Street Station,China Mieville,2000
... Thud!,Terry Pratchett,2005
... The Spellman Files,Lisa Lutz,2007
... Small Gods,Terry Pratchett,1992
... '''
>>> with open('books.csv', 'wt') as outfile:
...     outfile.write(text)
... 
201
>>> import sqlite3
>>> db = sqlite3.connect('books.db')
>>> curs = db.cursor()
>>> curs.execute('''create table book (title text, author text, year int)''')
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
sqlite3.OperationalError: table book already exists
>>> db.commit()
>>> import csv
>>> import sqlite3
>>> ins_str = 'insert into values(?, ?, ?)'
>>> with open('books.csv', 'rt') as infile:
...     books = csv.DictReader(infile)
...     for book in books:
...             curs.execute(ins_str, (book['title'], book['author'], book['year']))
... 
Traceback (most recent call last):
  File "<stdin>", line 4, in <module>
sqlite3.OperationalError: near "values": syntax error
>>> ins_str = 'insert into book values(?, ?, ?)'
>>> with open('books.csv', 'rt') as infile:
...     books = csv.DictReader(infile)
...     for book in books:
...             curs.execute(ins_str, (book['title'], book['author'], book['year']))
... 
<sqlite3.Cursor object at 0x100c2bd50>
<sqlite3.Cursor object at 0x100c2bd50>
<sqlite3.Cursor object at 0x100c2bd50>
<sqlite3.Cursor object at 0x100c2bd50>
<sqlite3.Cursor object at 0x100c2bd50>
>>> db.commit()

Comments

[toast-uz]

その要因は気づいていましたが、KeyError: 'title'ではなく、KeyError: 'author'が出るはずです。カンマの後のスペース削除で改善されるのは、'author'と'year'であり、'title'は変わりませんので。

元のコードでなぜKeyError: 'title'が出たのか知りたいので、質問文にコメントしたprint(book)を教えていただけますでしょうか？私にて再現させても、KeyError: 'author'が出てしまい、そのために質問をしていました。