self_example/Spider/Chapter04_数据存储/saveMySQL.py

34 lines
724 B
Python

# -*- encoding:utf-8 -*-
'''
@Author : dingjiawen
@Date : 2023/12/6 15:03
@Usage :
@Desc :
'''
import pymysql
data = {
'id': '20120001',
'name': 'Bob',
'age': 20
}
# 通过字典动态构建插入语句
table = 'students'
keys = ', '.join(data.keys())
values = ', '.join(['%s'] * len(data))
db = pymysql.connect(host='localhost', user='root',
password=None, port=3306, db='spiders')
cursor = db.cursor()
sql = 'INSERT INTO {table}({keys}) VALUES ({values})'.format(
table=table, keys=keys, values=values)
try:
if cursor.execute(sql, tuple(data.values())):
print('Successful')
db.commit()
except Exception as e:
print('Failed', e)
db.rollback()
db.close()