跳转至

Python PyMySQL 教程

PyMySQL 教程 展示了如何使用 PyMySQL 模块在 Python 中对 MySQL 进行编程。

PyMySQL

PyMySQL 是基于 PEP 249 的纯 Python MySQL 客户端库。大多数公共 API 与 mysqlclient 和 MySQLdb 兼容。 PyMySQL 可与 MySQL 5.5+和 MariaDB 5.5+一起使用。

MySQL 是领先的开源数据库管理系统。 它是一个多用户，多线程的数据库管理系统。 MySQL 在网络上特别流行。

PyMySQL 安装

1
$ sudo pip3 install PyMySQL

我们使用pip3工具安装 PyMySQL。

PyMySQL 版本示例

在下面的示例中，我们获取 MySQL 的版本。

version.py 
 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
#!/usr/bin/python3
# -*- coding: utf-8 -*-

import pymysql

con = pymysql.connect('localhost', 'user17', 
    's$cret', 'mydb')

with con:

    cur = con.cursor()
    cur.execute("SELECT VERSION()")

    version = cur.fetchone()

    print("Database version: {}".format(version[0]))

在 MySQL 中，我们可以使用SELECT VERSION()获取 MySQL 的版本。

1
import pymysql

我们导入pymysql模块。

1
2
con = pymysql.connect('localhost', 'user17', 
    's$cret', 'mydb')

我们使用connect()连接到数据库。 我们传递四个参数：主机名，MySQL 用户名，密码和数据库名。

1
with con:

使用with关键字，Python 解释器会自动释放资源。 它还提供错误处理。

1
cur = con.cursor()

从连接对象，我们创建一个游标。 游标用于遍历结果集中的记录。

1
cur.execute("SELECT VERSION()")

我们调用游标的execute()方法并执行 SQL 语句。

1
version = cur.fetchone()

fetchone()方法获取查询结果集的下一行，返回单个序列，或者在没有更多数据可用时返回None。

1
print("Database version: {}".format(version[0]))

我们打印数据库的版本。

1
2
$ ./version.py 
Database version: 5.7.23-0ubuntu0.16.04.1

这是输出。

PyMySQL fetchAll

fetchAll()方法检索查询结果的所有（剩余）行，并将它们作为序列序列返回。

retrieve_all.py 
 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
#!/usr/bin/python3
# -*- coding: utf-8 -*-

import pymysql

con = pymysql.connect('localhost', 'user17', 
    's$cret', 'testdb')

with con: 

    cur = con.cursor()
    cur.execute("SELECT * FROM cities")

    rows = cur.fetchall()

    for row in rows:
        print("{0} {1} {2}".format(row[0], row[1], row[2]))

在示例中，我们从数据库表中检索所有城市。

1
cur.execute("SELECT * FROM cities")

该 SQL 语句从 citys 表中选择所有数据。

1
rows = cur.fetchall()

fetchall()方法获取所有记录。 它返回一个结果集。 从技术上讲，它是一个元组的元组。 每个内部元组代表表中的一行。

1
2
for row in rows:
    print("{0} {1} {2}".format(row[0], row[1], row[2]))

我们将数据逐行打印到控制台。

1
2
3
4
5
6
7
8
9
$ ./retrieve_all.py 
1 Bratislava 432000
2 Budapest 1759000
3 Prague 1280000
4 Warsaw 1748000
5 Los Angeles 3971000
6 New York 8550000
7 Edinburgh 464000
8 Berlin 3671000

这是输出。

PyMySQL 字典游标

默认游标以元组的元组返回数据。 当我们使用字典游标时，数据以 Python 字典的形式发送。 这样，我们可以通过列名称来引用数据。

dictionary_cursor.py 
 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
#!/usr/bin/python3
# -*- coding: utf-8 -*-

import pymysql
import pymysql.cursors

con = pymysql.connect(host='localhost',
        user='user17',
        password='s$cret',
        db='mydb',
        charset='utf8mb4',
        cursorclass=pymysql.cursors.DictCursor)

with con:

    cur = con.cursor()
    cur.execute("SELECT * FROM cities")

    rows = cur.fetchall()

    for row in rows:
        print(row["id"], row["name"])

在此示例中，我们使用字典游标获取 citys 表的第一行。

1
2
3
4
5
6
con = pymysql.connect(host='localhost',
        user='user17',
        password='s$cret',
        db='mydb',
        charset='utf8mb4',
        cursorclass=pymysql.cursors.DictCursor)

在connect()方法中，我们将pymysql.cursors.DictCursor值传递给cursorclass参数。

1
2
for row in rows:
    print(row["id"], row["name"])

我们通过“城市”表的列名称引用数据。

PyMySQL 列标题

接下来，我们将展示如何使用数据库表中的数据打印列标题。

column_headers.py 
 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
#!/usr/bin/python3
# -*- coding: utf-8 -*-

import pymysql

con = pymysql.connect('localhost', 'user17', 
    's$cret', 'testdb')

with con:

    cur = con.cursor()
    cur.execute("SELECT * FROM cities")

    rows = cur.fetchall()

    desc = cur.description

    print("{0:>3} {1:>10}".format(desc[0][0], desc[1][0]))

    for row in rows:    
        print("{0:3} {1:>10}".format(row[0], row[2]))

列名称被认为是元数据。 它们是从游标对象获得的。

1
desc = cur.description

游标的description属性返回有关查询的每个结果列的信息。

1
print("{0:>3} {1:>10}".format(desc[0][0], desc[1][0]))

在这里，我们打印并格式化表格的列名。

1
2
for row in rows:    
    print("{0:3} {1:>10}".format(row[0], row[2]))

我们遍历并打印数据。

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
$ ./column_headers.py 
 id       name
  1     432000
  2    1759000
  3    1280000
  4    1748000
  5    3971000
  6    8550000
  7     464000
  8    3671000

这是输出。

PyMySQL 预备语句

在编写预备语句时，我们使用占位符，而不是直接将值写入语句中。 预准备的语句可提高安全性和性能。

prepared_statement.py 
 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
#!/usr/bin/python3
# -*- coding: utf-8 -*-

import pymysql

con = pymysql.connect('localhost', 'user17', 
    's$cret', 'testdb')

# user input
myid = 4

with con:    

    cur = con.cursor()

    cur.execute("SELECT * FROM cities WHERE id=%s", myid) 

    cid, name, population  = cur.fetchone()
    print(cid, name, population)

在示例中，我们获得具有指定 ID 的行。

1
cur.execute("SELECT * FROM cities WHERE id=%s", myid)

我们使用由%s标记标识的占位符。 在执行 SQL 语句之前，将值绑定到它们的占位符。

1
2
$ ./prepared_statement.py 
4 Warsaw 1748000

这是输出。

PyMySQL 受影响的行

rowcount是只读的游标属性，它指定最后一条 SELECT，UPDATE 或 INSERT 语句产生的行数。

affected_rows.py 
 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
#!/usr/bin/python3
# -*- coding: utf-8 -*-

import pymysql

con = pymysql.connect('localhost', 'user17',
   's$cret', 'mydb')

with con:

    cur = con.cursor()
    cur.execute("SELECT * FROM cities WHERE id IN (1, 2, 3)")

    # rows = cur.fetchall()

    # for row in rows:
    #     print("{0} {1} {2}".format(row[0], row[1], row[2]))

    print("The query affected {} rows".format(cur.rowcount))

在示例中，我们有一条 SELECT 语句选择三行。

1
print("The query affected {} rows".format(cur.rowcount))

我们生成一条消息，显示受影响的行数。

1
2
$ ./affected_rows.py 
The query affected 3 rows

这是输出。