python如何连接数据库？一文详解pymysql的用法。

news2025/1/19 23:25:21

使用Python连接数据库是常用的操作，那么在Python代码中取操作数据库呢？接下来介绍一个包：pymysql .它能帮我们在代码中连接MySQL数据库进行各种操作。

1.常用数据库模块

在做自动化测试时，我们经常会查库的需求，那么该如何使用python去查找数据库呢？这里介绍一个python的第三方包PyMySQL . 通过它使用python语言可以连接mysql数据库。

以下为安装和验证：

#安装： pip install pymysql
#验证： pip show pymysql

2.数据库操作流程

首先我们可以通过查看客户端工具连接数据库的流程：

如果使用pymysql进行连接数据库的流程具体如下：

导包
创建数据库连接
获取游标对象
执行操作
关闭游标对象
关闭数据库连接

将工具和pymysql的连接流程放在一起做个比对，将会是如下：

3.使用pymysql操作

3.1 建立连接

如果想使用pymysql操作数据库，你就需要调用pymysql中的connect()方法建立连接。

# 建立连接对象 
conn = pymysql.connect(host=None,user=None,password="",database=None,port=0,charset="")

# 参数说明：
    host :数据库服务器地址
    user :登录用户名
    password : 密码
    database : 数据库名称
    port : 数据库连接端口，默认为3306时刻不填写。
    charset : 数据库编码，一般设置为utf8

案例：使用pymysql进行建立连接数据库，连接school数据库

conn = pymysql.connect(host="localhost",user="root",password="root",database="school",charset='utf8')

3.2常用方法

常用的操作数据库方法：

cursor() : 获取游标对象
close() : 关闭对象
execute(sql) : 执行SQL语句，sql参数为需要传入的SQL语句。
fetchone() : 查询一条数据返回，从指针的下一条开始查一条数据。。
fetchmany(num) : 查询num条数据返回
fetchall() : 获取全部的返回行
rowcount : 获取execute()方法执行后影响的行数

将以上的方法和操作步骤进行对应

3.3项目案例

对数据库表数据进行查询

"""
连接数据库的流程 ：
1. 导包 ：import pymysql
2. 建立连接 ： connect(host,user,password,port,database,chaset)
3. 创建游标 ： cursor()
4. 执行SQL语句
    （1） 执行SQL ：execute(sql)
    (2) 查询一条数据 : fetchone()
    (3) 查询多条数据 ： fetchmany(number)  , number代表的是传入多条数据
    （4） 查询所有数据 ： fetchall()
5. 关闭游标 ： close()
6. 关闭连接 ： close()
"""

# 1. 导包
import pymysql

# 2. 建立连接对象
connect = pymysql.connect(host='localhost',user='root',password='root',database='school',charset='utf8')

print(connect)

# 3.  创建游标
cursor = connect.cursor()

# 4. 执行SQL语句
sql = "select * from students"
res = cursor.execute(sql)
print(res)

# 需求1 ：查询所有的数据
# all_data = cursor.fetchall()
# print(all_data)				, 查询结果返回一个元组，每一条数据也是一个元组

# 需求2 ： 查询一条数据
# one_data = cursor.fetchone()
# print(one_data)				,返回的这条数据为元组 

# 需求3 ： 查询5条数据
five_data = cursor.fetchmany(5)
print(five_data)				# 查询结果返回一个元组，每一条数据也是一个元组

# 5. 关闭游标
cursor.close()

# 6. 关闭连接对象
connect.close()

# 总结 ：以上的三个查询方法，若只返回一条数据 ，这一条数据就在一个元组中。若返回多条数据 ，每一条数据是构成一个元组 ，然后多条数据又放在一个大的元组中。

对数据库表数据进行增删改


# 1. 导包
import pymysql

# 2. 建立连接对象
connect = pymysql.connect(host='localhost',user='root',password='root',database='school',charset='utf8')

print(connect)

# 3.  创建游标
cursor = connect.cursor()

# 4. 执行SQL语句
#需求1 ：数据的修改
# sql = "update students set age = 54 where name = '王伟'"
# res = cursor.execute(sql)
# print(res)

# 需求2数据的插入
# sql = "insert into students values(27,'李四',19,'男','2班',345678943,'天津')"
# res = cursor.execute(sql)
# print(res)

# 需求3 ：数据的删除
sql = "delete from students where name = '李四'"
res = cursor.execute(sql)
print(res)


# 5. 关闭游标
cursor.close()

# 6. 关闭连接对象
connect.close()

# 总结 ： 使用cursor.execute()方法就可以进行增删改，只需要将SQL语句传入该方法中即可。

4.数据库封装

这里的数据库封装其实就是将项目的数据库连接对象封装进来，大家不需要每次进行初始化连接，而是执行调用一个方法就可以执行相应的SQL语句即可。

当然数据库封装应该是最灵活的一种情况，10个人有10个人的封装法，这里只给出一种最基本的封装方式。具体如下：


# 数据库类
import pymysql
from setting import DB_INFO
from loguru import logger


class MysqlClient(object):

    # 初始化方法，建立连接
    def __init__(self):
        self.conn = pymysql.connect(**DB_INFO)  # 初始化数据库连接

    # 执行SQL语句的增删改
    def execute(self,sql):
        try:
            with self.conn.cursor() as cursor:
                cursor.execute(sql)
                self.conn.commit()
                logger.info("执行的SQL语句:{}".format(sql))
        except Exception as e:
            logger.error("执行SQL错误，报错原因{},执行SQL:{}".format(e,sql))
            self.conn.rollback()


    # 执行查询
    def get_data(self,sql):
        try:
            with self.conn.cursor() as cursor:
                cursor.execute(sql)
                result = cursor.fetchall()
                logger.info("执行的SQL语句:{}".format(sql))
                return result
        except Exception as e:
            logger.error("执行SQL错误:{},执行SQL:{}".format(e, sql))
            
            
    # 关闭对象
    def close(self):
        self.conn.close()