Python3 MySQL 数据库连接 - PyMySQL 驱动
[TOC]
PyMySQL 是在 Python3.x 版本中用于连接 MySQL 服务器的一个库,Python2 中则使用 mysqldb
准备要求:mysql8.0以上版本及python3.0以上版本。
使用PyMySQL链接数据库
实例: 链接 Mysql 的 testdb 数据库:
代码语言:javascript复制#!/usr/bin/python3
import pymysql
# 打开数据库连接
db = pymysql.connect(host='localhost',user='root',password='123456',database='testdb')
# 使用 cursor() 方法创建一个游标对象 cursor
cursor = db.cursor()
# 使用 execute() 方法执行 SQL 查询
cursor.execute("SELECT VERSION()")
# 使用 fetchone() 方法获取单条数据.
data = cursor.fetchone()
print("Database version : %s " % data)
# 关闭数据库连接
db.close()
pymysql.connect()连接数据库函数
pymysql.Connect()参数说明
host(str) | MySQL服务器地址 |
---|---|
port(int) | MySQL服务器端口号 |
user(str) | 用户名 |
passwd(str) | 密码 |
db(str) | 数据库名称 |
charset(str) | 连接编码 |
connection对象:
使用pymysql.Connect()生成的是一个connection对象,Connection对象也称为数据库连接对象,Connection对象的功能是负责对数据源的连接。
其功能包括:获取游标 cursor()
,commit()
提交当前事务,rollback()
回滚当前事务,close()
关闭连接。
cursor():获取游标
要想操作数据库,光连接数据是不够的,必须拿到操作数据库的游标,才能进行后续的操作,比如读取数据、添加数据。通过获取到的数据库连接实例conn下的cursor()方法来创建游标。游标用来接收返回结果
cursor返回一个游标实例对象,其中包含了很多操作数据的方法
execute(op) | 执行一个数据库的查询命令 |
---|---|
fetchone() | 取得结果集的下一行 |
fetchmany(size) | 获取结果集的下几行 |
fetchall() | 获取结果集中的所有行 |
rowcount() | 返回数据条数或影响行数 |
close() | 关闭游标对象 |
对数据库的各种操作
创建类操作:
创建数据库
代码语言:javascript复制import pymysql
#打开数据库连接,不需要指定数据库,因为需要创建数据库
conn = pymysql.connect('localhost',user = "root",passwd = "123456")
#获取游标
cursor=conn.cursor()
#创建pythonBD数据库
cursor.execute('CREATE DATABASE IF NOT EXISTS pythonDB DEFAULT CHARSET utf8 COLLATE utf8_general_ci;')
cursor.close()#先关闭游标
conn.close()#再关闭数据库连接
print('创建pythonBD数据库成功')
创建数据库表
建立数据库连接后,可以使用创建的游标的execute()
方法将数据库表或记录创建到数据库表中。
import pymysql
#打开数据库连接
conn = pymysql.connect(host="localhost",user = "root",passwd = "28283860",db = "pythondb")
#获取游标
cursor=conn.cursor()
print(cursor)
#创建user表
cursor.execute('drop table if exists user')
sql="""CREATE TABLE IF NOT EXISTS `user` (
`id` int(11) NOT NULL AUTO_INCREMENT,
`name` varchar(255) NOT NULL,
`age` int(11) NOT NULL,
PRIMARY KEY (`id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8 AUTO_INCREMENT=0"""
cursor.execute(sql)
cursor.close()#先关闭游标
conn.close()#再关闭数据库连接
print('创建数据表成功')
插入操作:
当要将记录创建到数据库表中时,需要执行INSERT
操作。
插入单条数据:
代码语言:javascript复制import pymysql
#打开数据库连接,不指定数据库
conn=pymysql.connect(host='localhost',user='root',password='28283860')
conn.select_db('pythondb')
cur=conn.cursor()#获取游标
#创建user表
cur.execute('drop table if exists user')
sql="""CREATE TABLE IF NOT EXISTS `user` (
`id` int(11) NOT NULL AUTO_INCREMENT,
`name` varchar(255) NOT NULL,
`age` int(11) NOT NULL,
PRIMARY KEY (`id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8 AUTO_INCREMENT=0"""
cur.execute(sql)
insert=cur.execute("insert into user values(1,'tom',18)")
print('添加语句受影响的行数:',insert)
#另一种插入数据的方式,通过字符串传入值
sql="insert into user values(%s,%s,%s)"
cur.execute(sql,(3,'kongsh',20))
cur.close()
conn.commit()
conn.close()
print('sql执行成功')
select_db(“数据库名”) 可以用这种方式在数据库连接建立后指定数据库
插入多条数据:
代码语言:javascript复制import pymysql
#打开数据库连接,同时指定数据库
conn=pymysql.connect(host='localhost',user='root',password='28283860',database='pythondb')
cur=conn.cursor()#获取游标
#另一种插入数据的方式,通过字符串传入值
sql="insert into user values(%s,%s,%s)"
insert=cur.executemany(sql,[(4,'wen',20),(5,'tom',10),(6,'test',30)])
print ('批量插入返回受影响的行数:',insert)
cur.close()
conn.commit()
conn.close()
print('sql执行成功')
注意:批量插入多条sql语句采用的是executemany(sql,args)函数,返回受影响的行数。args参数是一个包含多个元组的列表,每个元组对应一条mysql中的一条数据。这里的%s不需要加引号,否则插入数据的数据会类型错误
查询操作:
代码语言:javascript复制使用execute()函数得到的只是受影响的行数,并不能真正拿到查询的内容。cursor对象还提供了3种提取数据的方法:fetchone、fetchmany、fetchall.。每个方法都会导致游标动,所以必须注意游标的位置。 cursor.fetchone():获取游标所在处的一行数据,返回元组,没有返回None cursor.fetchmany(size):接受size行返回结果行。如果size大于返回的结果行的数量,则会返回cursor.arraysize条数据。 cursor. fetchall():接收全部的返回结果行。
import pymysql
#打开数据库连接
conn=pymysql.connect(host='localhost',user='root',password='28283860',database='pythondb')
#获取游标
cur=conn.cursor()
cur.execute("select * from user;")
while 1:
res=cur.fetchone()
if res is None:
#表示已经取完结果集
break
print (res)
cur.close()
conn.commit()
conn.close()
print('sql执行成功')
注意:从execute()函数的查询结果中取数据,以元组的形式返回游标所在处的一条数据,如果游标所在处没有数据,将返回空元组,该数据执行一次,游标向下移动一个位置。fetchone()函数必须跟exceute()函数结合使用,并且在exceute()函数之后使用
代码语言:javascript复制#fetchone()函数实例:
import pymysql
#打开数据库连接
conn=pymysql.connect(host='localhost',user='root',password='28283860')
conn.select_db('pythondb')
#获取游标
cur=conn.cursor()
cur.execute("select * from user")
#取3条数据
resTuple=cur.fetchmany(3)
print(type(resTuple))
for res in resTuple:
print(res)
cur.close()
conn.commit()
conn.close()
print('sql执行成功')
注意:从exceute()函数结果中获取游标所在处的size条数据,并以元组的形式返回,元组的每一个元素都也是一个由一行数据组成的元组,如果size大于有效的结果行数,将会返回cursor.arraysize条数据,但如果游标所在处没有数据,将返回空元组。查询几条数据,游标将会向下移动几个位置。fetmany()函数必须跟exceute()函数结合使用,并且在exceute()函数之后使用
代码语言:javascript复制import pymysql
#打开数据库连接
conn=pymysql.connect(host='localhost',user='root',password='28283860')
conn.select_db('pythondb')
#获取游标
cur=conn.cursor()
cur.execute("select * from user")
#取所有数据
resTuple=cur.fetchall()
print(type(resTuple))
print ('共%d条数据'%len(resTuple))
cur.close()
conn.commit()
conn.close()
print('sql执行成功')
注意:获取游标所在处开始及以下所有的数据,并以元组的形式返回,元组的每一个元素都也是一个由一行数据组成的元组,如果游标所在处没有数据,将返回空元组。执行完这个方法后,游标将移动到数据库表的最后
更新操作:
单条数据更新操作:
代码语言:javascript复制#打开数据库连接
conn=pymysql.connect(host='localhost',user='root',password='28283860')
conn.select_db('pythondb')
#获取游标
cur=conn.cursor()
#更新一条数据
update=cur.execute("update user set age=100 where name='kongsh'")
print ('修改后受影响的行数为:',update)
#查询一条数据
cur.execute('select * from user where name="kongsh";')
print(cur.fetchone())
cur.close()
conn.commit()
conn.close()
print('sql执行成功')
多条数据更新操作:
代码语言:javascript复制import pymysql
conn=pymysql.connect(host='localhost',user='root',password='28283860')#打开数据库连接
conn.select_db('pythondb')#获取游标
cur=conn.cursor()
cur.execute("select * from user where name in ('kongsh','wen');")#更新前查询所有数据
print('更新前的数据为:')
for res in cur.fetchall():
print (res)
print ('*'*40)#更新2条数据
sql="update user set age=%s where name=%s"
update=cur.executemany(sql,[(15,'kongsh'),(18,'wen')])
#更新2条数据后查询所有数据
cur.execute("select * from user where name in ('kongsh','wen');")
print('更新后的数据为:')
for res in cur.fetchall():
print (res)
cur.close()
conn.commit()
conn.close()
print('sql执行成功')
删除数据:
删除单条数据:
代码语言:javascript复制import pymysql
#打开数据库连接
conn=pymysql.connect(host='localhost',user='root',password='28283860')#打开数据库连接
conn.select_db('pythondb')#获取游标
cur=conn.cursor()
cur.execute("select * from user;")#删除前查询所有数据
print('删除前的数据为:')
for res in cur.fetchall():
print (res)
print ('*'*40)
cur.execute("delete from user where id=1")#删除1条数据
#删除后查询所有数据
cur.execute("select * from user;")
print('删除后的数据为:')
for res in cur.fetchall():
print (res)
cur.close()
conn.commit()
conn.close()
print('sql执行成功')
删除多条数据:
代码语言:javascript复制import pymysql
#打开数据库连接
conn=pymysql.connect(host='localhost',user='root',password='28283860')
conn.select_db('pythondb')
cur=conn.cursor()#获取游标
cur.execute("select * from user;")#删除前查询所有数据
print('删除前的数据为:')
for res in cur.fetchall():
print (res)
print ('*'*40)#删除2条数据
sql="delete from user where id=%s"
cur.executemany(sql,[(3),(4)])
cur.execute("select * from user;")#删除后查询所有数据
print('删除后的数据为:')
for res in cur.fetchall():
print (res)
cur.close()
conn.commit()
conn.close()
print('sql执行成功')
事务回滚
代码语言:javascript复制import pymysql
conn=pymysql.connect(host='localhost',user='root',password='28283860')#打开数据库连接
conn.select_db('pythondb')
cur=conn.cursor()#获取游标
cur.execute("select * from user;")#修改前查询所有数据
print('修改前的数据为:')
for res in cur.fetchall():
print (res)
print ('*'*40)
cur.execute("update user set name='xiaoxiaoxiaoxiaoren' where id=5")#更新表中第1条数据
cur.execute("select * from user;")#修改后查询所有数据
print('修改后的数据为:')
for res in cur.fetchall():
print (res)
print ('*'*40)
conn.rollback()#回滚事务
cur.execute("select * from user;")
print('回滚事务后的数据为:')
for res in cur.fetchall():
print (res)
cur.close()
conn.commit()
conn.close()
print('sql执行成功')