现在是用的大部分是python 3.x
,之前在python 2.x
使用的MySQLdb
不支持python 3.x
,故记录此篇关于pymysql
的用法,以备以后查询,如果有新的用法会随时更新。
安装:pip install pymysql
写入数据数据
import pymysql
#port要求是int型
db = pymysql.connect(host='localhost', passwd = '123456', port = 3301, charset='utf8', db='test_db')
#创建游标,所有的数据库操作都是基于游标的
cursor = db.cursor()
sql = "INSERT INTO person_info SET name='John', age=18"
effect_row = cursor.execute(sql) #这时候还没有真正提交到数据库,因为默认是auto_commit=false
print('effect row:',effect_row)
db.commit() #提交数据
cursor.close() #关闭游标
db.close() #关闭数据库连接
"""
effect row: 1
"""
查询数据
import pymysql
db = pymysql.connect(host='localhost', passwd = '123456', port = 3301, charset='utf8', db='test_db')
cursor = db.cursor()
sql = "SELECT * FROM person_info"
query_num = cursor.execute(sql) #返回的是查询到的行数
print('get num:', query_num)
record = cursor.fetchone() #获取一行数据
print('fetchone:')
print(record)
records = cursor.fetchmany(2) #获取接下来的两行
print('fetchmany:')
print(records)
records = cursor.fetchall() #获取剩下的
print('fetchall:')
print(records)
cursor.close()
db.close()
"""
('get num:', 5)
fetchone:
(1, u'John', 18)
fetchmany:
((2, u'Tom', 20), (3, u'Mary', 21))
fetchall:
((4, u'Sally', 22), (5, u'Bob', 23))
"""
设置事务自动提交
pymysql默认事务是不会自动提交的,必须我们在创建db对象的时候指定autocommit为True
import pymysql
#autocommit=True,打开自动提交
db = pymysql.connect(host='localhost', passwd = '123456', port = 3301, charset='utf8', db='test_db', autocommit=True)
cursor = db.cursor()
sql = "INSERT INTO person_info SET name='David', age=18"
effect_row = cursor.execute(sql) #这时候还没有真正提交到数据库,因为默认是auto_commit=false
print('effect row:',effect_row)
cursor.close() #关闭游标
db.close() #关闭数据库连接
"""
effect row: 1
"""
获取自增ID
如果执行的多次语句,只能获取最后一个语句执行时对应生成的id
import pymysql
db = pymysql.connect(host='localhost', passwd = '123456', port = 1280, charset='utf8', db='test_db', autocommit=True)
cursor = db.cursor()
sql = "INSERT INTO person_info SET name=%s, age=%s"
effect_row = cursor.executemany(sql, [('Kaci', '16'),('Harry', '25')])
print('effect row:',effect_row)
print('last id:', cursor.lastrowid) 通过cursor对象的属性来获取
cursor.close() #关闭游标
db.close() #关闭数据库连接
"""
effect row: 2
last id: 8
"""
获取字典类型数据
默认获取的是元组类型,可以通过设置cursor的参数来使得获取的结果是字典类型
import pymysql
db = pymysql.connect(host='localhost', passwd = '123456', port = 1280, charset='utf8', db='test_db', autocommit=True)
cursor = db.cursor(cursor=pymysql.cursors.DictCursor) #设置类型
sql = "SELECT * FROM person_info"
query_num = cursor.execute(sql) #返回的是查询到的行数
records = cursor.fetchmany(3)
print(records)
cursor.close() #关闭游标
db.close() #关闭数据库连接
"""
[{'id': 1, 'name': 'John', 'age': 18}, {'id': 2, 'name': 'Tom', 'age': 20}, {'id': 3, 'name': 'Mary', 'age': 21}]
"""