7.爬虫数据存入数据库通用连接

修改数据库和表名就可以存入对应的数据库和表中

import pymysql

# 连接数据库
db = pymysql.connect(host="127.0.0.1", port=3306, user="root",
                     password="root", database="python1901", charset="utf8")

# 获取游标
cursor = db.cursor()
# 指定数据表
table = 'user'

# 要插入数据库的数据
item ={
    "name": "龙泽",
    "age": 18,
    "sex": 1,
}

# 往表中插入数据
keys = ",".join(item.keys())
values = ",".join(["%s"] * len(item))
sql = f"insert into {table}({keys}) values({values})"

try:
    cursor.execute(sql,(tuple(item.values())))
    print("数据存储成功!")
    db.commit()
except Exception as e:
    print(e)
    db.rollback()
    print("数据存储失败!")