作为练习,我将从API中提取数据并将其插入psql数据库。我最初遵循默认限制每拉1000个条目,但决定我想尝试获得所有的数据,这是大约40K行。经过一点实验,我可以拉4800,但然后我得到以下结果:
Traceback (most recent call last):
File "data_pull.py", line 19, in <module>
postgres_db.Bike_Count.insert_many(data).execute()
File "/usr/local/lib/python3.5/dist-packages/peewee.py", line 3516, in execute
cursor = self._execute()
File "/usr/local/lib/python3.5/dist-packages/peewee.py", line 2901, in _execute
sql, params = self.sql()
File "/usr/local/lib/python3.5/dist-packages/peewee.py", line 3484, in sql
return self.compiler().generate_insert(self)
File "/usr/local/lib/python3.5/dist-packages/peewee.py", line 2084, in generate_insert
value = row_dict[field]
KeyError: <peewee.IntegerField object at 0x7f5b32c2c7f0>
数据拉.py
import json, requests, peewee
import postgres_db
endpoint = 'https://data.seattle.gov/resource/4xy5-26gy.json?$limit=4800'
response = requests.get(endpoint, headers={'X-App-Token': '(REMOVED)'})
if response.status_code == 200:
data = json.loads(response.text)
postgres_db.Bike_Count.create_table(True)
postgres_db.Bike_Count.insert_many(data).execute()
邮递员
import peewee
psql_db = peewee.PostgresqlDatabase('database', user='my_username')
class Bike_Count(peewee.Model):
date = peewee.DateTimeField()
fremont_bridge_sb = peewee.IntegerField()
fremont_bridge_nb = peewee.IntegerField()
class Meta:
database = psql_db
我在网上看过这些表格,以为其中的一个条目有问题,但我找不到任何明显的问题。谢谢你的帮助。
最佳答案
我在本地尝试了您的代码(删除了应用程序令牌和4800限制),它按预期工作:
id | date | fremont_bridge_sb | fremont_bridge_nb
------+---------------------+-------------------+-------------------
1 | 2017-01-09 06:00:00 | 28 | 55
2 | 2017-01-04 20:00:00 | 19 | 10
3 | 2017-01-18 13:00:00 | 18 | 18
4 | 2017-01-06 11:00:00 | 22 | 15
5 | 2017-01-27 11:00:00 | 39 | 38
6 | 2017-01-08 14:00:00 | 6 | 10
7 | 2017-01-06 23:00:00 | 8 | 3
8 | 2017-01-27 13:00:00 | 45 | 35
...
当我在附加限制的情况下运行它时,我注意到API返回的一行只包含一个
date
键(缺少fremont_bridge_nb和fremont_bridge_sb字段)。Peewee要求大容量插入的每一行都有相同的键,所以问题是Peewee希望找到所有3个键。