Python之PyMySQL的使用:
在python3.x中,可以使用pymysql来MySQL数据库的连接,并实现数据库的各种操作,本次博客主要介绍了pymysql的安装和使用方法。
PyMySQL的安装
一、.windows上的安装方法:
在python3.6中,自带pip3,所以在python3中可以直接使用pip3去安装所需的模块:
pip3 install pymysql -i https://pypi.douban.com/simple
二、.linux下安装方法:
1.tar包下载及解压
下载tar包
wget https://pypi.python.org/packages/29/f8/919a28976bf0557b7819fd6935bfd839118aff913407ca58346e14fa6c86/PyMySQL-0.7.11.tar.gz#md5=167f28514f4c20cbc6b1ddf831ade772
解压并展开tar包
tar xf PyMySQL-0.7.11.tar.gz
2.安装
[root@localhost PyMySQL-0.7.11]# python36 setup.py install
数据库的连接
本次测试创建的数据及表:
#创建数据库及表,然后插入数据
mysql> create database dbforpymysql;
mysql> create table userinfo(id int not null auto_increment primary key,username varchar(10),passwd varchar(10))engine=innodb default charset=utf8;
mysql> insert into userinfo(username,passwd) values('frank','123'),('rose','321'),('jeff',666);#查看表内容
mysql> select * from userinfo;
+----+----------+--------+
| id | username | passwd |
+----+----------+--------+
| 1 | frank | 123 |
| 2 | rose | 321 |
| 3 | jeff | 666 |
+----+----------+--------+
3 rows in set (0.00 sec)
连接数据库:
import pymysql#连接数据库
db = pymysql.connect("localhost","root","LBLB1212@@","dbforpymysql")#使用cursor()方法创建一个游标对象
cursor = db.cursor()#使用execute()方法执行SQL语句
cursor.execute("SELECT * FROM userinfo")#使用fetall()获取全部数据
data = cursor.fetchall()#打印获取到的数据
print(data)#关闭游标和数据库的连接
cursor.close()
db.close()#运行结果
((1, 'frank', '123'), (2, 'rose', '321'), (3, 'jeff', '666'))
要完成一个MySQL数据的连接,在connect中可以接受以下参数:
参数
cursor其实是调用了cursors模块下的Cursor的类,这个模块主要的作用就是用来和数据库交互的,当你实例化了一个对象的时候,你就可以调用对象下面的各种绑定方法:
一些绑定方法
数据库操作
一、数据库增删改操作
commit()方法:在数据库里增、删、改的时候,必须要进行提交,否则插入的数据不生效。
import pymysql
config={"host":"127.0.0.1","user":"root","password":"LBLB1212@@","database":"dbforpymysql"
}
db = pymysql.connect(**config)
cursor = db.cursor()
sql = "INSERT INTO userinfo(username,passwd) VALUES('jack','123')"
cursor.execute(sql)
db.commit() #提交数据
cursor.close()
db.close()
或者在execute提供插入的数据
import pymysql
config={"host":"127.0.0.1","user":"root","password":"LBLB1212@@","database":"dbforpymysql"
}
db = pymysql.connect(**config)
cursor = db.cursor()
sql = "INSERT INTO userinfo(username,passwd) VALUES(%s,%s)"
cursor.execute(sql,("bob","123"))
db.commit() #提交数据
cursor.close()
db.close()
MySQL的注入问题
executemany():用来同时插入多条数据:
import pymysql
config={"host":"127.0.0.1","user":"root","password":"LBLB1212@@","database":"dbforpymysql"
}
db = pymysql.connect(**config)
cursor = db.cursor()
sql = "INSERT INTO userinfo(username,passwd) VALUES(%s,%s)"
cursor.executemany(sql,[("tom","123"),("alex",'321')])
db.commit() #提交数据
cursor.close()
db.close()
execute()和executemany()都会返回受影响的行数:
sql = "delete from userinfo where username=%s"
res = cursor.executemany(sql,("jack",))
print("res=",res)
#运行结果
res= 1
当表中有自增的主键的时候,可以使用lastrowid来获取最后一次自增的ID:
import pymysql
config={"host":"127.0.0.1","user":"root","password":"LBLB1212@@","database":"dbforpymysql"
}
db = pymysql.connect(**config)
cursor = db.cursor()
sql = "INSERT INTO userinfo(username,passwd) VALUES(%s,%s)"
cursor.execute(sql,("zed","123"))
print("the last rowid is ",cursor.lastrowid)
db.commit() #提交数据
cursor.close()
db.close()#运行结果
the last rowid is 10
二、数据库的查询操作
这里主要介绍三个绑定方法:
fetchone():获取下一行数据,第一次为首行;
fetchall():获取所有行数据源
fetchmany(4):获取下4行数据
先来查看表的内容:
mysql> select * from userinfo;
+----+----------+--------+
| id | username | passwd |
+----+----------+--------+
| 1 | frank | 123 |
| 2 | rose | 321 |
| 3 | jeff | 666 |
| 5 | bob | 123 |
| 8 | jack | 123 |
| 10 | zed | 123 |
+----+----------+--------+
6 rows in set (0.00 sec)
import pymysql
config={"host":"127.0.0.1","user":"root","password":"LBLB1212@@","database":"dbforpymysql"
}
db = pymysql.connect(**config)
cursor = db.cursor()
sql = "SELECT * FROM userinfo"
cursor.execute(sql)
res = cursor.fetchone() #第一次执行
print(res)
res = cursor.fetchone() #第二次执行
print(res)
cursor.close()
db.close()#运行结果
(1, 'frank', '123')
(2, 'rose', '321')
使用fetchall():
import pymysql
config={"host":"127.0.0.1","user":"root","password":"LBLB1212@@","database":"dbforpymysql"
}
db = pymysql.connect(**config)
cursor = db.cursor()
sql = "SELECT * FROM userinfo"
cursor.execute(sql)
res = cursor.fetchall() #第一次执行
print(res)
res = cursor.fetchall() #第二次执行
print(res)
cursor.close()
db.close()
#运行结果
((1, 'frank', '123'), (2, 'rose', '321'), (3, 'jeff', '666'), (5, 'bob', '123'), (8, 'jack', '123'), (10, 'zed', '123'))
()
可以看到,第二次获取的时候,什么数据都没有获取到,这个类似于文件的读取操作。
默认情况下,我们获取到的返回值是元组,只能看到每行的数据,却不知道每一列代表的是什么,这个时候可以使用以下方式来返回字典,每一行的数据都会生成一个字典:
cursor = db.cursor(cursor=pymysql.cursors.DictCursor) #在实例化的时候,将属性cursor设置为pymysql.cursors.DictCursor
使用fetchall获取所有行的数据,每一行都被生成一个字典放在列表里面:
import pymysql
config={"host":"127.0.0.1","user":"root","password":"LBLB1212@@","database":"dbforpymysql"
}
db = pymysql.connect(**config)
cursor = db.cursor(cursor=pymysql.cursors.DictCursor)
sql = "SELECT * FROM userinfo"
cursor.execute(sql)
res = cursor.fetchall()
print(res)
cursor.close()
db.close()
#运行结果
[{'id': 1, 'username': 'frank', 'passwd': '123'}, {'id': 2, 'username': 'rose', 'passwd': '321'}, {'id': 3, 'username': 'jeff', 'passwd': '666'}, {'id': 5, 'username': 'bob', 'passwd': '123'}, {'id': 8, 'username': 'jack', 'passwd': '123'}, {'id': 10, 'username': 'zed', 'passwd': '123'}]
这样获取到的内容就能够容易被理解和使用了!
在获取行数据的时候,可以理解开始的时候,有一个行指针指着第一行的上方,获取一行,它就向下移动一行,所以当行指针到最后一行的时候,就不能再获取到行的内容,所以我们可以使用如下方法来移动行指针:
cursor.scroll(1,mode='relative') # 相对当前位置移动
cursor.scroll(2,mode='absolute') # 相对绝对位置移动
第一个值为移动的行数,整数为向下移动,负数为向上移动,mode指定了是相对当前位置移动,还是相对于首行移动
例如:
sql = "SELECT * FROM userinfo"
cursor.execute(sql)
res = cursor.fetchall()
print(res)
cursor.scroll(0,mode='absolute') #相对首行移动了0,就是把行指针移动到了首行
res = cursor.fetchall() #第二次获取到的内容
print(res)#运行结果
[{'id': 1, 'username': 'frank', 'passwd': '123'}, {'id': 2, 'username': 'rose', 'passwd': '321'}, {'id': 3, 'username': 'jeff', 'passwd': '666'}, {'id': 5, 'username': 'bob', 'passwd': '123'}, {'id': 8, 'username': 'jack', 'passwd': '123'}, {'id': 10, 'username': 'zed', 'passwd': '123'}]
[{'id': 1, 'username': 'frank', 'passwd': '123'}, {'id': 2, 'username': 'rose', 'passwd': '321'}, {'id': 3, 'username': 'jeff', 'passwd': '666'}, {'id': 5, 'username': 'bob', 'passwd': '123'}, {'id': 8, 'username': 'jack', 'passwd': '123'}, {'id': 10, 'username': 'zed', 'passwd': '123'}]
上下文管理器
在python的文件操作中支持上下文管理器,在操作数据库的时候也可以使用:
import pymysql
config={"host":"127.0.0.1","user":"root","password":"LBLB1212@@","database":"dbforpymysql"
}
db = pymysql.connect(**config)
with db.cursor(cursor=pymysql.cursors.DictCursor) as cursor: #获取数据库连接的对象sql = "SELECT * FROM userinfo" cursor.execute(sql)res = cursor.fetchone()print(res)cursor.scroll(2,mode='relative')res = cursor.fetchone()print(res)cursor.close()
db.close()#运行结果
{'id': 1, 'username': 'frank', 'passwd': '123'}
{'id': 5, 'username': 'bob', 'passwd': '123'}
Cursor常用API
1、cursor.execute(query, args=None):执行sql语句。
参数:
query (str):sql语句。
args (tuple, list or dict):sql语句中如果有变量,或者格式化输出,会在这里填充数据。
Returns:返回一个int值,表示sql语句影响到的行数。
下面演示插入数据的四种方式:
import pymysql.cursors
# 连接数据库
connect = pymysql.connect(
host=‘127.0.0.1’,
user=‘root’,
password=‘123’,
db=‘demo_temp’,
charset=‘utf8’,
cursorclass=pymysql.cursors.DictCursor
)
with connect.cursor() as cursor: # 创建游标
# 第一种方式向表中插入数据
sql = “”"
INSERT INTO users (user,passwd) VALUES (‘123’,‘123’)
“”"
cursor.execute(sql)
# 第二种方式向表中插入数据
sql = “”"
INSERT INTO users (user,passwd) VALUES (%s,%s)
“”"
cursor.execute(sql,[‘456’,‘456’])
# 第三种方式向表中插入数据
sql = “”"
INSERT INTO users (user,passwd) VALUES (%s,%s)
“”"
cursor.execute(sql, (‘789’, ‘789’))
# 第四种方式向表中插入数据
sql = “”"
INSERT INTO users (user,passwd) VALUES (%(a)s,%(b)s)
“”"
cursor.execute(sql, {‘a’:‘abc’,‘b’:‘123’}) # 这里的’a’和’b’要和上面VALUES中对应
connect.commit() # 插入数据后必须更新
# 查询数据
sql = “”"
select * from users
“”"
cursor.execute(sql)
ret = cursor.fetchall() # 提取查询数据
print(ret)
connect.close() # 关闭数据库连接
2、cursor.executemany(query, args):可以执行多条sql语句。
参数:
query :sql语句
args:可迭代对象
Returns:返回sql语句影响到的行数。
下面是方法实例:
import pymysql.cursors
# 连接数据库
connect = pymysql.connect(
host=‘127.0.0.1’,
user=‘root’,
password=‘123’,
db=‘demo_temp’,
charset=‘utf8’,
cursorclass=pymysql.cursors.DictCursor
)
with connect.cursor() as cursor: # 创建游标
# 插入多条数据
sql = “”"
INSERT INTO users (user,passwd) VALUES (%s,%s)
“”"
ret_row = cursor.executemany(sql,[[‘a’,‘123’],[‘b’,‘234’],[‘c’,‘456’]]) # 在args中传入一个可迭代对象。
connect.commit() # 插入数据后必须更新
print(f’当前受影响了{ret_row}行’)
# 查询数据
sql = “”"
select * from users
“”"
cursor.execute(sql)
ret = cursor.fetchall() # 提取查询数据
print(ret)
connect.close() # 关闭数据库连接
3、cursor.fetchone():从查询语句中获取一条查询结果。
4、cursor.fetchall():获取所有查询到的结果。
5、cursor.fetchmany(size):根据size获取相应的行数。
6、cursor.mogrify(query, args=None):返回执行的sql语句,下面是示例:
import pymysql.cursors
# 连接数据库
connect = pymysql.connect(
host=‘127.0.0.1’,
user=‘root’,
password=‘123’,
db=‘demo_temp’,
charset=‘utf8’,
cursorclass=pymysql.cursors.DictCursor
)
with connect.cursor() as cursor: # 创建游标
sql = “”"
select * from users
“”"
cursor.execute(sql)
ret_sql = cursor.mogrify(sql) # 返回执行的sql语句
cursor.fetchmany(2) # 提取查询数据
print(ret_sql)
connect.close() # 关闭数据库连接
7、cursor.close():关闭游标,关闭游标后,游标内所有数据将被清空。