Mysqldb 和 Python KeyError 处理

Question

我正在尝试向 MySQL 表中添加多个值，代码如下：

Try:
cursor.execute("INSERT INTO companies_and_charges_tmp (etags, company_id,  created, delivered, satisfied, status, description, persons_entitled) VALUES ('%s, %s, %s, %s, %s, %s, %s, %s')" % (item['etag'], ch_no, item['created_on'], item['delivered_on'], item['satisfied_on'], item['status'], item['particulars'][0]['description'], item['persons_entitled'][0]['name']))

Except KeyError:
    pass

问题是这段代码在循环中，有时正在插入的值之一会丢失，这将导致 Key Error 取消整个插入。

如何通过 KeyError，所以当与正在插入的项目之一相关的 KeyError 发生时，其他项目仍会添加到表中，而缺少的项目只是保留为 NULL？

Answer 1

您可以使用dict.get() method，如果在字典中找不到键，它将返回None。然后MySQL 驱动程序将在查询参数化步骤中将None 转换为NULL：

# handling description and name separately
try:
    description = item['particulars'][0]['description']
except KeyError:
    description = None

# TODO: violates DRY - extract into a reusable method?
try:
    name = item['persons_entitled'][0]['name']
except KeyError:
    name = None

cursor.execute("""
     INSERT INTO 
         companies_and_charges_tmp 
         (etags, company_id,  created, delivered, satisfied, status, description, persons_entitled) 
     VALUES 
         (%s, %s, %s, %s, %s, %s, %s, %s)""", 
     (item.get('etag'), ch_no, item.get('created_on'), item.get('delivered_on'), item.get('satisfied_on'), item.get('status'), description, name))