Python怎样利用pymysql封装项目通用的连接和查询

蜗牛互联网技术资讯2022-07-251600

这篇文章主要介绍了Python怎样利用pymysql封装项目通用的连接和查询，具有一定借鉴价值，需要的朋友可以参考下。下面就和我一起来看看吧。

前言

一个项目通常都需要有数据库，而对于python这门语言，除了一些框架自带orm或者扩展的orm(像django自带orm,flask则需要扩展的orm),使用orm必然有他的好处，但毫无疑问你要花时间学习这个orm,那么接下来阿牛带你们用pymysql简单分装一个通用的连接，关闭和查询！

pymysql 介绍与安装

PyMySQL 是在 Python3.x 版本中用于连接 MySQL 服务器的一个库，Python2中则使用mysqldb。

在终端使用以下指令安装：

pip3 install PyMySQL

pymysql 的使用

import  pymysql

# 打开数据库连接，password为你的数据库密码，db是数据库名
db = pymysql.connect(host="127.0.0.1",
                     user="root",
                     password=" ",
                     db=" ",
                     charset="utf8")

# 使用 cursor() 方法创建一个游标对象
cursor = db.cursor()

# 使用 execute()  方法执行 SQL 查询 
cursor.execute("SELECT VERSION()")

# 使用 fetchone() 方法获取单条数据.
data = cursor.fetchone()

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

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

password为你的数据库密码，db是数据库名，操作前请先建立好你的mysql数据库。

Python怎样利用pymysql封装项目通用的连接和查询 python ios小火箭免费节点二维码第1张

如图，我连接成功并且获取到了我数据库的版本！

方法execute执行我们的sql语句。

在获取执行的结果时，可以指定获取的结果的条数，可以使用的选项如下：

fetchone()         # 取得检索结果的一条数据

fetchmany(n)       # 取得检索结果的n条数据

fetchall()         # 取得检索结果的所有数据

需要注意的是，与读取文件时的指针类似，如果在同一段代码中，先使用fetchone()获取检索结果的第一条数据，然后再使用fetchmany(2)的话，指针会在检索结果的当前位置向后读取执行结果，而不会从头开始重新读取检索的结果。

获取到的结果是元组。，如下图：

Python怎样利用pymysql封装项目通用的连接和查询 python ios小火箭免费节点二维码第2张

封装项目通用的连接和查询

请自行写入密码和数据库

import pymysql

# 创建连接
#return: 连接，游标
def get_conn():

    # 创建连接
    conn = pymysql.connect(host="127.0.0.1",
                           user="root",
                           password="",
                           db="",
                           charset="utf8")
    # 创建游标
    cursor = conn.cursor()# 执行完毕返回的结果集默认以元组显示
    return conn, cursor


# 关闭游标，连接
def close_conn(conn, cursor):
    cursor.close()
    conn.close()


def query(sql,*args):
    """
    封装通用查询
    :param sql:
    :param args:
    :return: 返回查询到的结果，((),(),)的形式
    """
    conn, cursor = get_conn()
    cursor.execute(sql,args)
    res = cursor.fetchall()
    close_conn(conn, cursor)
    return res

因此，我们在查询时只需要调用query()函数就行，*args可以让其传入任意个参数或者不传参数，只需要sql语句中的占位符和参数个数对应就可以！

使用如下图所示

Python怎样利用pymysql封装项目通用的连接和查询 python ios小火箭免费节点二维码第3张