PyMySQL使用

PyMySQL使用

前言：Pymysql库是python对MySQL数据库读取和操作，使用mysql管理测试用例会大大提升测试自动化的程度。

PyMysql连接数据库

pip install PyMySQL==1.0.2

PyMySQL方法	说明
pymysql.connect(host,user,password,database,charset,port)	连接数据库，返回一个数据库对象
cursor()	游标对象，后续才能执行SQL语句
execute（）	执行SQL语句
fetchone（）	获取一条语句
cursor.close()	关闭游标
db.close()	关闭数据库

import pymysql

# 创建数据库对象
db = pymysql.connect(
        # 设置数据库主机的地址
        host = "127.0.0.1",
        # 设置数据库的用户名
        user = "root",
        # 设置数据库
        password = "123456",
        # 设置数据库的名称
        database = "dbtest",
        # 设置数据库的字符集
        charset = "utf8",
      # 设置数据库的端口号
        port = 3306
)
# 创建 SQL 游标对象，游标对象主要用来执行 SQL 语句
cursor = db.cursor()
# 要执行的 SQL 语句
sql = "select * from stu"
# 使用 execute() 方法执行 SQL 语句
cursor.execute(sql)
# 使用 fetchone() 方法一次性获取一条数据
res = cursor.fetchone()
# 打印获取的数据
print(res)
print(type(res)) #数据为元组类型
# 关闭游标对象
cursor.close()
# 关闭 database 对象
db.close()

获取查询数据

• 使用pymysql.connect连接数据库，返回一个数据库对象
• 使用cursor = db.cursor（）获得游标对象，主要用来执行SQL语句
• cursor.execute(sql) 执行SQL语句，若是查询操作结果存放在cursor中
• 使用cursor.fetchon(), fetchall() 来获得查询结果
  • fetchone() 方法一次性获取一条记录，每一条记录是一个元组形式，每获取一次，游标自动往前一行，所以每次都会自动获取下一条记录
  • fetchmany(数字) 一次性指定多条记录其中数字为获取的条数，游标也会按照数字自动下移
  • fetchall() 方法一次性获取表中所有记录，游标移动到底部，再次获取讲获取不到数据

# 字典游标
cursor = db.cursor(pymysql.cursors.DictCursor) #  获取游标将以字典形式返回数据

• 使用字典游标是为了方便取数据，sql中表形式存在便于取某个字段的值

数据回写与回滚机制

• db.commit() 提交事务（理解为数据保存）

  # 创建 SQL 游标对象，游标对象主要用来执行 SQL 语句
  cursor = db.cursor()
  # 要执行的 SQL 语句 
  sql = "INSERT INTO stu set sid='4',sname='张三',ssex='男';"
  # 使用 execute() 方法执行 SQL 语句
  cursor.execute(sql)
  
  # 注意mysql 默认有事务所以要提交数据
  db.commit()
  

• db.rollback() 回滚一般和try except结合使用

  例如: 同时插入二条数据，要么同时成功要么同时失败。

  sql001 = "INSERT INTO stu set sid='4',sname='李四',ssex='男';"
  sql002 = "INSERT INTO stu set sid='4',sname='小花',ssex='女';"
  # 使用 execute() 方法执行 SQL 语句
  try:
      cursor.execute(sql001)# 执行这条是成功的
      cursor.execute(sql002)# 执行这条是失败的因为主键重复了
  	# 注意mysql 默认有事务所以要提交数据
      db.commit()
  except:
      #try 中执行语句出现异常就一起撤回！
      print("业务流程失败，去除sql语句的影响")
      db.rollback()
  

删除数据和修改数据

• 不可以直接删除被外键引用的数据

sql001 = "delete from stu where sname='范冰';"
sql002="update stu set sname='晓明' where sid=3;"

# 使用 execute() 方法执行 SQL 语句
try:
    cursor.execute(sql001)# 执行这条是成功的
    cursor.execute(sql002)
    db.commit()
except :
    #try 中执行语句出现异常就一起撤回！
    print("业务流程失败，去除sql语句的影响")
    db.rollback()