Python连接数据库，从基础的增删改查到实战，一步到位!(我连网站都是我自己搭建的)

为什么要做python连接mysql，一般是解决什么问题的

自动化测试中，经常需要向数据库添加或删除数据，也需要验证测试数据和数据库的数据是否一致。这个时候就需要用Python连接Mysql

安装PyMySQL

pip install PyMySQL

连接MySql

import pymysql
# 打开数据库连接
db = pymysql.connect("localhost", "root", "111223", "study_date")

# 使用 cursor() 方法创建一个游标对象 cursor
cursor = db.cursor()
# 使用 execute()  方法执行 SQL 查询
cursor.execute("SELECT VERSION()")
# 使用 fetchone() 方法获取单条数据.
data = cursor.fetchone()
print("Database version : %s " % data)
# 关闭数据库连接
db.close()

数据库基本操作

增加数据

insert 语句可以用来将一行或多行数据插到数据库表中, 使用的一般形式如下:

insert into 表名 [(列名1, 列名2, 列名3, ...)] values (值1, 值2, 值3, ...);

import pymysql
 
# 打开数据库连接
db = pymysql.connect("localhost", "root", "111223", "study_date")
# 使用cursor()方法获取操作游标,以字典的方式获取数据，原数据类型为元组
cursor = db.cursor(cursor=pymysql.cursors.DictCursor)
insert_sql = "insert into studys(id, name, age) values(3, '骑着乌龟赶猪', 35)"
# 执行sql语句
cursor.execute(insert_sql) 
# 提交到数据库执行 
db.commit() 
# 关闭数据库连接
db.close()

删除数据

delete 语句用于删除表中的数据

delete from 表名称 where 删除条件;

import pymysql

# 打开数据库连接
db = pymysql.connect("localhost", "root", "111223", "study_date")

# 使用cursor()方法获取操作游标,以字典的方式获取数据，原数据类型为元组
cursor = db.cursor(cursor=pymysql.cursors.DictCursor)
# SQL 删除数据
del_sql = "delete from studys where id=3"
# 执行sql语句
cursor.execute(del_sql)
# 提交到数据库执行
db.commit()
# 关闭数据库连接
db.close()

修改数据

update 语句可用来修改表中的数据

update 表名称 set 列名称=新值 where 更新条件;
update studys set age=25 where id=1

import pymysql

# 打开数据库连接
db = pymysql.connect("localhost", "root", "111223", "study_date")
# 使用cursor()方法获取操作游标,以字典的方式获取数据，原数据类型为元组
cursor = db.cursor(cursor=pymysql.cursors.DictCursor)
# SQL 修改数据
updata_sql = "update studys set age=30 where id=2"
# 执行sql语句
cursor.execute(updata_sql)
# 提交到数据库执行
db.commit()
# 关闭数据库连接
db.close()

查询数据

语法：

fetchone()

例如要查询 students 表中所有学生的名字和年龄, 输入语句
select name, age from studys

fetchone()获取一行数据

# 导入模块
import pymysql

# 打开数据库连接     数据库地址
db = pymysql.connect("localhost", "root", "111223", "study_date")
# 使用cursor()方法获取操作游标,以字典的方式获取数据，原数据类型为元组
cursor = db.cursor(cursor=pymysql.cursors.DictCursor)
# 使用 execute()方法执行 SQL 查询
# 通配符，意思是查询表里所有内容
cursor.execute("select * from studys")
# 使用 fetchone() 方法获取一行数据.
data = cursor.fetchone()
print(data)
# 关闭数据库连接
db.close()

电商项目实战–判断主界面商品分类是否与数据库一致

传送门如何搭建网站教程

第1步：获取界面分类数据

 #! /usr/bin/python3
 -*- coding:utf-8 -*-
 @Time    : 2020/8/21 9:26
 @公众号  : 码上开始

from selenium import webdriver

url = "http://localhost:8080/Shopping/index.jsp"
driver = webdriver.Chrome()
driver.get(url)
# 复数形式定位定位
els = driver.find_elements_by_xpath("/html/body/table[5]/tbody/tr/td[2]/table[1]/tbody/tr/td/a")
# 空列表用于存放数据
list = [ ]
for i in els:
    # 获取text文本然后添加到列表
    list.append(i.text)
print("界面数据：", list)

第2步：获取数据库商品分类数据

db = pymysql.connect(host="localhost", user="root", password="111223", database="db_shopping")
cur = db.cursor(cursor=pymysql.cursors.DictCursor)
# 从tb_bigtype数据库中选择bigName列表
sql = "select bigName from tb_bigtype"
cur.execute(sql)
db.commit()
cur.execute(sql)
data = cur.fetchall()
db_list = []
print("数据库数据：",data)
for i in data:
    for j in i.values():
        db_list.append(j)
print(db_list)

第3步：两组数据进行断言

第一种方法：

# 反向排序列表
list.reverse()
if db_list == list:
    print("一致")
else:
    print("不一致")

第二种方式断言：
断言>>集合>>求交集

if set(list) & set(db_list):
     print("两者一致")
 else:
     print("不一致")

课后练习

1. 将电商实战代码封装后再进行断言
2. 将数据库的操作封装在一个类中，并且把连接数据库、增、删、改、查的操作封装为方法

---

如果你不想再体验一次自学时找不到学习资料，没人解答问题，坚持几天便放弃的话，可以加我们的软件测试交流群【902061117】