如何使用PyMySQL操作mysql数据库？

作者 Airy

在工作和生活中我们用Python处理数据的情况并不少见，而且很多情况是从数据库取数据，比如MySQL，这里我来分享下简单的Python操作MySQL的库pymysql。

安装与应用

适用环境

python版本 >=2.6或3.3

mysql版本>=4.1

安装

可以使用pip安装也可以手动下载安装。

使用pip安装，在命令行执行如下命令：

pip install PyMySQL(大写不行换小写)

手动安装，请先下载。下载地址：https://github.com/PyMySQL/PyMySQL/tarball/pymysql-X.X。

其中的X.X是版本（目前可以获取的最新版本是0.7.6）。

下载后解压压缩包。在命令行中进入解压后的目录，执行如下的指令：

python setup.py install

建议使用pip安装。

使用示例

连接数据库如下：

import pymysql.cursors

config = {

'host':'127.0.0.1',

'port':3306,

'user':'root',

'password':'zhyea.com',

'db':'employees',

'charset':'utf8mb4',

'cursorclass':pymysql.cursors.DictCursor,

}

# Connect to the database

connection = pymysql.connect(**config)

插入数据：

执行sql语句前需要获取cursor，因为配置默认自动提交，故在执行sql语句后需要主动commit，最后不要忘记关闭连接：

from datetime import date, datetime, timedelta

import pymysql.cursors

#连接配置信息

config = {

'host':'127.0.0.1',

'port':3306,

'user':'root',

'password':'zhyea.com',

'db':'employees',

'charset':'utf8mb4',

'cursorclass':pymysql.cursors.DictCursor,

}

# 创建连接

connection = pymysql.connect(**config)

# 获取明天的时间

tomorrow = datetime.now().date() + timedelta(days=1)

# 执行sql语句

try:

with connection.cursor() as cursor:

# 执行sql语句，插入记录

sql = 'INSERT INTO employees (first_name, last_name, hire_date, gender, birth_date) VALUES (%s, %s, %s, %s, %s)'

cursor.execute(sql, ('Robin', 'Zhyea', tomorrow, 'M', date(1989, 6, 14)));

# 没有设置默认自动提交，需要主动提交，以保存所执行的语句

connection.commit()

finally:

connection.close();

执行查询：

import datetime

import pymysql.cursors

#连接配置信息

config = {

'host':'127.0.0.1',

'port':3306,

'user':'root',

'password':'zhyea.com',

'db':'employees',

'charset':'utf8mb4',

'cursorclass':pymysql.cursors.DictCursor,

}

# 创建连接

connection = pymysql.connect(**config)

# 获取雇佣日期

hire_start = datetime.date(1999, 1, 1)

hire_end = datetime.date(2016, 12, 31)

# 执行sql语句

try:

with connection.cursor() as cursor:

# 执行sql语句，进行查询

sql = 'SELECT first_name, last_name, hire_date FROM employees WHERE hire_date BETWEEN %s AND %s'

cursor.execute(sql, (hire_start, hire_end))

# 获取查询结果

result = cursor.fetchone()

print(result)

# 没有设置默认自动提交，需要主动提交，以保存所执行的语句

connection.commit()

finally:

connection.close();

这里的查询只取了一条查询结果，查询结果以字典的形式返回：

result = cursor.fetchmany(2)

不过不建议这样使用，最好在sql语句中设置查询的记录总数。获取全部结果集可以使用fetchall方法：

result = cursor.fetchall()

因为只有两条记录，所以上面提到的这两种查询方式查到的结果是一样的：

[{‘last_name’: ‘Vanderkelen’, ‘hire_date’: datetime.date(2015, 8, 12), ‘first_name’: ‘Geert’}, {‘last_name’: ‘Zhyea’, ‘hire_date’: datetime.date(2015, 8, 21), ‘first_name’: ‘Robin’}]

在django中使用

在django中使用是我找这个的最初目的。目前同时支持python3.4、django1.8的数据库backend并不好找。这个是我目前找到的最好用的。

设置DATABASES和官方推荐使用的MySQLdb的设置没什么区别：

关键是这里：我们还需要在站点的__init__.py文件中添加如下的内容：

import pymysql

pymysql.install_as_MySQLdb()

以上就是Python中pymysql库的简单使用，谢谢。