一个项目通常都需要有数据库,而对于python这门语言,除了一些框架自带orm或者扩展的orm(像django自带orm,flask则需要扩展的orm),使用orm必然有他的好处,但毫无疑问你要花时间学习这个orm,那么接下来阿牛带你们用pymysql简单分装一个通用的连接,关闭和查询!
PyMySQL 是在 Python3.x 版本中用于连接 MySQL 服务器的一个库,Python2中则使用mysqldb。
在终端使用以下指令安装:
pip3 install 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数据库。
如图,我连接成功并且获取到了我数据库的版本!
方法execute执行我们的sql语句。
在获取执行的结果时,可以指定获取的结果的条数,可以使用的选项如下:
fetchone() # 取得检索结果的一条数据 fetchmany(n) # 取得检索结果的n条数据 fetchall() # 取得检索结果的所有数据
需要注意的是,与读取文件时的指针类似,如果在同一段代码中,先使用fetchone()获取检索结果的第一条数据,然后再使用fetchmany(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语句中的占位符和参数个数对应就可以!
使用如下图所示
只要你的sql过关,pymsql也可以帮助你完成项目!不一定需要使用orm哦!