劝君惜取少年时

PyMysql使用详解

在编写小脚本时，PyMysql是快速连接并操作数据库的一个不错选择。

安装

pip3 install PyMysql# 可使用 pip list 查看此环境安装了哪些第三方库pip list | grep PyMysql

使用

import pymysql

连接数据库

使用connect函数创建连接对象，此连接对象提供关闭数据库、事务回滚等操作
一般传参为：host, user, password, port(默认为3306), database（想要连接的数据库名）

db = pymysql.connect(host='127.0.0.1', user='root', password='pwd', database='database_name')

连接对象的常见方法：

操作数据库

操作数据库主要使用cursor游标对象，对数据库进行操作
cursor游标对象的常见方法

• fetchone()，fetchmany(size)，fetchall() 三者的返回值都是元组 或 元组的嵌套

查

查询主要获取结果集，注意fetchone 得到的是元组， fetchmany(size)，fetchall()得到的是元组的嵌套
注意:
如果既要使用python的字符串解析%，sql语句中又包含sql的模糊查询占位符%, 模糊查询使用%%即可，这样不会报错

import pymysql# 创建数据库连接对象db = pymysql.connect(host='127.0.0.1', user='root', password='pwd', database='database_name')# 创建游标对象cursor = db.cursor()table_name = 'map_point'sql = "SELECT * FROM %s WHERE username LIKE 'DL%%' " % table_nametry:cursor.execute(sql)# 执行sql语句，也可执行数据库命令，如：show tablesresult = cursor.fetchall()# 所有结果print(result)except Exception as e:db.rollback()print("查询失败")print(e)finally:cursor.close()# 关闭当前游标db.close()# 关闭数据库连接

循环查询：
在一些场景中，需要进行循环查询，判断，此时在执行完execute查询语句后，务必进行commit提交，否则每次查询到的数据都是之前查询结果的快照，也就是旧数据。
或者采用第二种方案，创建connect连接时，增添autocommit=True属性，自动进行commit提交。

增，删，改

增删改需要有提交事务的操作，查不需要提交事务，但如果循环查询，务必提交事务，否则结果都是重复的
创建连接对象——db= pymysql()
创建cursor游标对象——cur = db.cursor()
数据操作——cur.execute(sql)
提交连接事务——db.commit()
关闭cursor游标对象——cur.close()
关闭连接对象——db.close( )

import pymysql# 创建数据库连接对象db = pymysql.connect(host='127.0.0.1', user='root', password='pwd', database='database_name')# 创建游标对象cursor = db.cursor()table_name = 'table_name 'user_id = 'yyy'user_no = 'xxx'sql = "UPDATE %s SET user_no = '%s' WHERE user_id = '%s'" % (table_name, user_no, user_id)try:cursor.execute(sql)# 执行sql语句，也可执行数据库命令，如：show tablesdb.commit()# 增删改，必须执行事务print("数据更新成功")except Exception as e:db.rollback()# 若出现失败，进行回滚print("数据更新失败")print(e)finally:cursor.close()# 关闭当前游标db.close()# 关闭数据库连接

使用循环批量更新

db = pymysql.connect(host='127.0.0.1', user='root', password='pwd', database='database_name')table_name = 'table_name'update_list = ['xxx2', 'xxxx3']condition_list = ['xxx', 'xxx1']# 条件集合，更新集合长度相等时可使用，可根据其他情况重新编写cursor = db.cursor()for i in range(len(condition_list)):sql = "UPDATE %s SET user_no = '%s' WHERE user_id = '%s'" % (table_name, update_list[i], condition_list[i])print('执行sql语句：' + sql)try:cursor.execute(sql)db.commit()print("数据更新成功" + str(i+1) + '条')except Exception as e:db.rollback()print("数据更新失败")print(e)cursor.close()db.close()

使用executemany(query, param) 批量更新

params为每条记录的维度，可为嵌套数组和元组
注意：sql语句中需要更改的数据不管什么类型，统一使用%s作为占位符，不需要加引号

# 创建数据库连接对象db = pymysql.connect(host='127.0.0.1', user='root', password='pwd', database='database_name')# 创建游标对象cursor = db.cursor()update_list = ['a', 'b']condition_list = ['a1', 'b1']# 条件数组和需要更新的数据数组是分开的，因为param需要以每条的维度，所以可以使用拉链函数合并params = zip(update_list, condition_list) # 或 param = [['a', 'a1'], ['b', 'b1']]sql = "UPDATE map_point SET storageLocation_no = %s WHERE position_id = %s"# 注意 这里的 占位%s 都不用加引号try:cursor.executemany(sql, params)# 执行sql语句db.commit()# 执行事务print("数据批量更新成功")except Exception as e:db.rollback()print("数据更新失败")print(e)finally:cursor.close()# 关闭当前游标db.close()# 关闭数据库连接