Python使用PyMysql操作数据库

安装

pip install -U pymysql

连接数据库

连接数据库有两种不同的格式

直接使用参数

代码如下

import pymysql.cursors

connection = pymysql.Connect(
		host='127.0.0.1',
		port=3306,
		user='root',
		password='root',
		db='spider',
		charset='utf8mb4',
		cursorclass=pymysql.cursors.DictCursor
)

使用字典存储参数

代码如下

import pymysql.cursors

DbConfig = {
	'host'       : '127.0.0.1',
	'port'       : 3306,
	'user'       : 'root',
	'password'   : 'root',
	'db'         : 'spider',
	'charset'    : 'utf8mb4',
	'cursorclass': pymysql.cursors.DictCursor
}

connection = pymysql.Connect(**DbConfig)    # 作为关键字参数传入

查询数据

直接上代码

try:
	with connection.cursor() as cursor:
		select_sql = "select * from spider_tasks"
		cursor.execute(select_sql)
		result = cursor.fetchall()
		print(result)
	connection.commit()
finally:
	connection.close()

如果未设置自动提交,需要手动提交修改

得到的是多个字段组成的列表,也可以使用fetchone()获取单条记录

fetchmany(num)能获取到指定数目的记录,不过不建议这么做,应该在SQL语句中进行限制,以减少数据库的压力

记得最后关闭数据库连接

其他的数据库操作类似于此,不再赘述