pymysql

现在是用的大部分是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}]
"""