本文Python 操作 MySQL 数据库需要是使用到 PyMySQL 驱动

• PyMySQL 是在 Python3.x 版本中用于连接 MySQL 服务器的一个库，Python2 中则使用 mysqldb。
• PyMySQL 遵循 Python 数据库 API v2.0 规范，并包含了 pure-Python MySQL 客户端库。

Python 操作 MySQL 前提是要安装好 MySQL 数据库并能正常连接使用，安装步骤详见下文。

操作流程

1. 安装 MySQL 数据库
2. pymysql 模块安装与使用
3. 获取数据库的连接
4. 执行 sql 语句或存储过程
5. 关闭数据库连接

安装 MySQL 数据库

注意： 安装过程我们需要通过开启管理员权限来安装，否则会由于权限不足导致无法安装。

一、Linux 上安装 MySQL

1、检测系统是否自带安装 MySQL:

  rpm -qa | grep mysql

2、如果查看到系统有安装 MySQL，那可以选择进行卸载:

rpm -e mysql…　　# 普通删除模式
rpm -e --nodeps mysql…　　# 强力删除模式，如果使用上面命令删除时，提示有依赖的其它文件，则用该命令可以对其进行强力删除

3、安装 MySQL

首先需要先下载 MySQL 安装包，官网下载地址 下载对应版本即可,或直接在网上拉取并安装：

wget http://repo.mysql.com/mysql-community-release-el7-5.noarch.rpm # 获取安装包
rpm -ivh mysql-community-release-el7-5.noarch.rpm                    # 安装
yum update
yum install mysql-server  

权限设置：

chown mysql:mysql -R /var/lib/mysql

初始化 MySQL：

mysqld --initialize

启动 MySQL：

systemctl start mysqld

查看 MySQL 运行状态：

systemctl status mysqld

4、修改默认密码

Mysql安装成功后，默认的root用户密码为空，你可以使用以下命令来创建root用户的密码：

mysqladmin -u root password "new_password";

登陆：

mysql -uroot -p'new_password'

三、创建数据库

创建数据库：

create database test_db;

查看数据库：

mysql> show databases;
+--------------------+
| Database           |
+--------------------+
| information_schema |
| mysql              |
| performance_schema |
| sys                |
| test_db            |
+--------------------+
5 rows in set (0.00 sec)

mysql> use test_db;
Database changed

pymysql 模块安装与数据 CURD

一、安装

PyMySQL 模块使用 pip命令进行安装：

pip3 install PyMySQL

假如系统不支持 pip 命令，可以使用以下方式安装：

$ git clone https://github.com/PyMySQL/PyMySQL
$ cd PyMySQL/
$ python3 setup.py install

二、pymysql 连接数据库

pymysql .connect 函数：连接上数据库

# 导入模块
import pymysql

# 打开数据库连接

conn = pymysql.connect(
    host="127.0.0.1",
    user="root",
    password="123456",
    database="test_db",
    charset="utf8")

print(conn)
print(type(conn))

输出结果显示如下：表面数据库连接成功

<pymysql.connections.Connection object at 0x00000000022A54A8>
<class 'pymysql.connections.Connection'>

使用 pymysql 的 connect() 方法连接数据库，connect 参数解释如下：

• host：MySQL服务的地址，若数据库在本地上，使用 localhost 或者127.0.0.1。如果在其它的服务器上，则写对应的 IP地址
• port：服务的端口号，默认为3306，不写则为默认值。
• user：登录数据库的用户名
• passwd：登录 MySQL 的密码
• db：数据库名
• charset：设置为 utf8 编码，解决存汉字乱码问题

conn.cursor():获取游标

如果要操作数据库，光连接数据是不够的，咱们必须拿到操作数据库的游标，才能进行后续的操作，游标的主要作用是用来接收数据库操作后的返回结果，比如读取数据、添加数据。通过获取到的数据库连接实例 conn 下的 cursor() 方法来创建游标，实例如下：

# 导入模块
import pymysql

# 打开数据库连接
conn = pymysql.connect(
    host="127.0.0.1",
    user="root",
    password="123456",
    database="test_db",
    charset="utf8")

# print(conn)
# print(type(conn))
# 获取连接下的游标
cursor_test = conn.cursor()
print(cursor_test)

输出结果为：

<pymysql.cursors.Cursor object at 0x000000000A544B00>

cursor 返回一个游标实例对象，其中包含了很多操作数据的方法，如执行sql语句，sql 执行命令：execute()和executemany()

execute(query,args=None):

• 函数作用：执行单条的sql语句，执行成功后返回受影响的行数
• 参数说明：query：要执行的sql语句，字符串类型args：可选的序列或映射，用于query的参数值。如果args为序列，query中必须使用%s做占位符；如果args为映射，query中必须使用%(key)s做占位符

executemany(query,args=None)：

• 函数作用：批量执行sql语句，比如批量插入数据，执行成功后返回受影响的行数
• 参数说明：query：要执行的sql语句，字符串类型args：嵌套的序列或映射，用于query的参数值

其他游标对象如下表：

完整数据库连接操作实例如下：

# 导入模块
import pymysql

# 打开数据库连接
conn = pymysql.connect(
    host="127.0.0.1",
    user="root",
    password="123456",
    database="test_db",
    charset="utf8")
# print(conn)
# print(type(conn))

# 获取连接下的游标
cursor_test = conn.cursor()
print(cursor_test)

# 使用 execute()  方法执行 SQL 查询，查询数据库版本
cursor_test.execute("SELECT VERSION()")

# 使用 fetchone() 方法返回一条数据.
data = cursor_test.fetchone()

print("Database version : %s " % data)

# 关闭数据库连接
conn.close()

以上结果输出为：

Database version : 5.7.28 

三、创建表

创建表代码如下：

import pymysql

# 打开数据库连接
conn = pymysql.connect(
    host="127.0.0.1",
    user="root",
    password="123456",
    database="test_db",
    charset="utf8")

# 获取连接下的游标
cursor_test = conn.cursor()

# 使用 execute() 方法执行 SQL，如果表存在则删除
cursor_test.execute("DROP TABLE IF EXISTS EMPLOYEE")

# 使用预处理语句创建表
sql = """CREATE TABLE user1 (
         FIRST_NAME  CHAR(20) NOT NULL,
         LAST_NAME  CHAR(20),
         AGE INT,  
         SEX CHAR(1),
         INCOME FLOAT )"""

cursor_test.execute(sql)

# 关闭数据库连接
conn.close()

如下所示数据库表创建成功：

mysql> desc user1;
+------------+----------+------+-----+---------+-------+
| Field      | Type     | Null | Key | Default | Extra |
+------------+----------+------+-----+---------+-------+
| FIRST_NAME | char(20) | NO   |     | NULL    |       |
| LAST_NAME  | char(20) | YES  |     | NULL    |       |
| AGE        | int(11)  | YES  |     | NULL    |       |
| SEX        | char(1)  | YES  |     | NULL    |       |
| INCOME     | float    | YES  |     | NULL    |       |
+------------+----------+------+-----+---------+-------+
5 rows in set (0.00 sec)

三、插入数据

插入数据实现代码：

import pymysql

# 打开数据库连接
conn = pymysql.connect(
    host="127.0.0.1",
    user="root",
    password="123456",
    database="test_db",
    charset="utf8")

# 获取连接下的游标
cursor_test = conn.cursor()

# 使用预处理语句创建表
sql = """INSERT INTO user1(FIRST_NAME,
         LAST_NAME, AGE, SEX, INCOME)
         VALUES ('Fei', 'Fei', 20, 'M', 1000)"""

try:
   # 执行sql语句
   cursor_test.execute(sql)
   # 提交到数据库执行
   conn.commit()
except:
   # 如果发生错误则回滚
   conn.rollback()

# 关闭数据库连接
conn.close()

插入数据结果：

mysql> select * from user1;
+------------+-----------+------+------+--------+
| FIRST_NAME | LAST_NAME | AGE  | SEX  | INCOME |
+------------+-----------+------+------+--------+
| Fei        | Fei       |   20 | M    |   1000 |
+------------+-----------+------+------+--------+
1 row in set (0.00 sec)

四、查询数据

Python查询Mysql使用 fetchone() 方法获取单条数据, 使用fetchall() 方法获取多条数据。

• fetchone(): 该方法获取下一个查询结果集。结果集是一个对象
• fetchall(): 接收全部的返回结果行.
• rowcount: 这是一个只读属性，并返回执行 execute()方法后影响的行数。

查询数据代码如下：

import pymysql


# 打开数据库连接
conn = pymysql.connect(
    host="127.0.0.1",
    user="root",
    password="123456",
    database="test_db",
    charset="utf8")

# 获取连接下的游标
cursor_test = conn.cursor()

sql = """
    select * from user1"""

try:
    # 执行 sql 语句
    cursor_test.execute(sql)
    # 显示出所有数据
    data_result = cursor_test.fetchall()
    for row in data_result:
        fname = row[0]
        lname = row[1]
        age = row[2]
        sex = row[3]
        income = row[4]
        # 打印结果
        print("fname=%s,lname=%s,age=%s,sex=%s,income=%s" % \
              (fname, lname, age, sex, income))
except:
    print("Error: unable to fetch data")

# 关闭数据库连接
conn.close()

输出结果：

fname=Fei,lname=Fei,age=20,sex=M,income=1000.0

五、数据库表更新操作

# 导入模块
import pymysql

# 打开数据库连接
conn = pymysql.connect(
    host="127.0.0.1",
    user="root",
    password="123456",
    database="test_db",
    charset="utf8")
# print(conn)
# print(type(conn))

# 获取连接下的游标
cursor_test = conn.cursor()

sql = "UPDATE user1 SET AGE = AGE + 1 WHERE SEX = '%c'" % ('M')

try:
    # 执行SQL语句
    cursor_test.execute(sql)
    # 提交到数据库执行
    conn.commit()
except:
    # 发生错误时回滚
    conn.rollback()

# 关闭数据库连接
conn.close()

六、删除操作

# 导入模块
import pymysql

# 打开数据库连接
conn = pymysql.connect(
    host="127.0.0.1",
    user="root",
    password="123456",
    database="test_db",
    charset="utf8")
# print(conn)
# print(type(conn))

# 获取连接下的游标
cursor_test = conn.cursor()

sql = "DELETE * FROM user1"

try:
    # 执行SQL语句
    cursor_test.execute(sql)
    # 提交到数据库执行
    conn.commit()
except:
    # 发生错误时回滚
    conn.rollback()

# 关闭数据库连接
conn.close()

七、错误处理

DB API中定义了一些数据库操作的错误及异常，下表列出了这些错误和异常:

总结

本文给大家介绍 Python 如何连接 Mysql 进行数据的增删改查操作，文章通过简洁的代码方式进行示例演示，给使用 Python 操作 Mysql 的工程师提供支撑。