python自动化测试(5)--操作mysql数据库

一、数据库操作的基本流程

1、安装并导入操作mysql数据库的模块：pymysql

2、与数据库建立连接：conn = pymysql.connect(系列参数)

　　常用的系列参数包含：

　　host、user、password、port、database、charset(编码方式)、cursorclass(用于指定查询结果的数据格式，默认是tuple)

3、创建游标：用于操作sql语句

　　cur = conn.cursor()

4、执行sql语句

　　cur.execute(sql语句)    # 返回值为查询结果的数量

5、获取查询的结果

　　5.1、获取单条查询结果

　　　　one = cur.fetchone()

　　5.2、查询所有的查询结果

　　　　all = cur.fetchall()

6、关闭连接资源

　　cur.close()

　　conn.close()

7、说明：

　　在执行数据库查询操作之前，为了确保查询的数据准确，需要单独执行一条 self.conn.commit()语句。

二、代码实现

import pymysql

# 1、建立连接
conn = pymysql.connect(
    host="api.lemonban.com",
    user="future",
    password="123456",
    port=3306,
    database="futureloan",
    charset="utf8",
    cursorclass=pymysql.cursors.DictCursor       # 将查询的结果使用字典进行保存(默认使用元组进行保存)
)

# 2、创建游标
cur = conn.cursor()

# 3、执行sql语句
sql = "select * from member limit 3"
cur.execute(sql)      # 返回值为查询结果的数量
# 4、获取单条查询的结果
one = cur.fetchone()
print("单条查询的结果为：",one)
# 5、获取所有的查询结果
all = cur.fetchall()
for item in all:
    print(item)
    pass
# 6、关闭连接资源
cur.close()
conn.close()

三、封装数据库操作流程

要求：在建立数据库连接时，参数通过配置文件获取

import pymysql

from integration1.Commons.handle_config import handleConfig

class HandleDB():

    def __init__(self):
        # 建立连接
        self.conn = pymysql.connect(
            host=handleConfig.get("mysql","host"),
            user=handleConfig.get("mysql","user"),
            password=handleConfig.get("mysql","password"),
            port=handleConfig.getint("mysql","port"),
            database=handleConfig.get("mysql","database"),
            charset="utf8",
            cursorclass=pymysql.cursors.DictCursor
        )
        # 创建游标
        self.cur = self.conn.cursor()
        pass
    # 获取1条数据
    def select_one_data(self,sql):
        self.cur.execute(sql)
        return self.cur.fetchone()
    # 获取所有数据
    def select_all_data(self,sql):
        self.cur.execute(sql)
        return self.cur.fetchall()
    #获取查询的条数
    def get_count(self,sql):
        return self.cur.execute(sql)

    def update_data(self,sql):
        self.cur.execute(sql)  # 更新数据库
        self.conn.commit()     # 提交
        pass

    # 关闭操作
    def close(self):
        self.cur.close()    # 关闭游标
        self.conn.close()   # 关闭连接
        pass

if __name__ == '__main__':
    sql = "select * from member limit 3"
    handleDB = HandleDB()
    count = handleDB.get_count(sql)
    print(count)
    one = handleDB.select_one_data(sql)
    print(one)
    all = handleDB.select_all_data(sql)
    print(all)

今后，如果我们需要操作mysql的数据库资源时，只需要导入该模块即可。通过实例化HandleDB，就可以进行相应的查询操作。

四、应用

要求：随机生成一个手机号码，要求生成的手机号码未在数据库中注册

思路：

　　1、先生成符合要求的11位手机号码

　　2、通过手机号字段查询数据库，根据查询结果确定手机号码是否已在数据库中注册：

　　　　2.1、若已存在，重复执步骤1和2

　　　　2.2、若不存在，则返回生成的手机号码

import random

from integration1.Commons.handle_db import HandleDB    # 导入操作数据库的类

prefix = [133, 149, 153, 173, 177, 180, 181, 189, 199,
          130, 131, 132, 145, 155, 156, 166, 171, 175, 176, 185, 186, 166,
          134, 135, 136, 137, 138, 139, 147, 150, 151, 152, 157, 158, 159, 172, 178, 182, 183, 184, 187, 188, 198
          ]

def get_new_phone():
    handleDB = HandleDB()
    while True:
        # 生成电话号码
        phone = __generate_phone()
        #校验生成的号码是否已存在数据库中
        sql = "select * from member where mobile_phone = '{}'".format(phone)
        count = handleDB.get_count(sql)
        if count == 0:   # 如果生成的手机号码在数据库中不存在，则返回该手机号码
            handleDB.close()
            return phone
    pass

def __generate_phone():
    # 生成电话号码的前缀
    pre = prefix[random.randint(0,len(prefix)-1)]
    phone = str(pre)
    for _ in range(8):
        phone += str(random.randint(0,9))
        pass
    return phone
    pass

if __name__ == '__main__':
    print(get_new_phone())