Python+Pytest+Yaml+Request+Allure框架源代码之(一)common公共方法封装

common模块:

在这里插入图片描述

  • get_path.py:获取路径方法
# -*- coding: UTF-8 -*-
import os

# 项目根目录
BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))

# 配置文件目录
CONFIG_DIR = os.path.join(BASE_DIR,'config')

# 测试用例文件目录
TESTCASES_DIR = os.path.join(BASE_DIR,'testcases')

#data文件目录
DATA_DIR = os.path.join(BASE_DIR,'data')

#日志文件目录
LOGS_DIR=os.path.join(BASE_DIR,'logs')


if __name__ == '__main__':
    print(LOGS_DIR)
  • logger_util.py:日志封装
import logging
import time
from common.get_path import *
from common.yaml_util import read_file


class LoggerUitl:
    def create_log(self,logger_name='log'):
        # 创建一个日志对象
        self.logger = logging.getLogger(logger_name)
        # 设置全局的日志级别(DEBUG<INFO<WARNING<ERROR<CRITICAL)
        self.logger.setLevel(logging.DEBUG)
        # 防止日志重复
        if not self.logger.handlers:
            #------------文件日志--------------
            # 获取日志文件的名称
            self.file_log_path = LOGS_DIR+'/'+ read_file('/config/config.yml','log','log_name') + str(int(time.time()))+".log"
            # 创建文件日志的控制器
            self.file_handler = logging.FileHandler(self.file_log_path,encoding='utf-8')
            # 设置文件日志的级别
            file_log_level= str(read_file('/config/config.yml','log','log_level')).lower()
            if file_log_level == 'debug':
                self.file_handler.setLevel(logging.DEBUG)
            elif file_log_level == 'info':
                self.file_handler.setLevel(logging.INFO)
            elif file_log_level == 'waring':
                self.file_handler.setLevel(logging.WARNING)
            elif file_log_level == 'error':
                self.file_handler.setLevel(logging.ERROR)
            elif file_log_level == 'critical':
                self.file_handler.setLevel(logging.CRITICAL)
            # 设置文件日志的格式
            self.file_handler.setFormatter(logging.Formatter(read_file('/config/config.yml','log','log_format')))
            # 将控制器加入到日志对象
            self.logger.addHandler(self.file_handler)

            #------------控制台日志--------------
            # 创建控制台日志的控制器
            self.console_handler = logging.StreamHandler()
            # 设置控制台日志的级别
            console_log_level= read_file('/config/config.yml','log','log_level').lower()
            if console_log_level == 'debug':
                self.console_handler.setLevel(logging.DEBUG)
            elif console_log_level == 'info':
                self.console_handler.setLevel(logging.INFO)
            elif console_log_level == 'waring':
                self.console_handler.setLevel(logging.WARNING)
            elif console_log_level == 'error':
                self.console_handler.setLevel(logging.ERROR)
            elif console_log_level == 'critical':
                self.console_handler.setLevel(logging.CRITICAL)
            # 设置控制台日志的格式
            self.console_handler.setFormatter(logging.Formatter(read_file('/config/config.yml','log','log_format')))
            # 将控制器加入到日志对象
            self.logger.addHandler(self.console_handler)

        return self.logger

# 函数:输出正常日志
def my_log(log_massage):
    LoggerUitl().create_log().info(log_massage)

# 函数:输出错误日志
def error_log(log_massage):
    LoggerUitl().create_log().error(log_massage)
    raise Exception(log_massage)


if __name__ == '__main__':
    my_log('zhangweixu')


- parameters_until.py:传参方式方法封装

import csv
import json
import traceback
import yaml
from common.get_path import *
from common.logger_util import error_log


# 读取csv文件
def read_csv_file(csv_file):
    '''c参数说明'''
    csv_list = []
    path = BASE_DIR+"/"+csv_file
    with open(path,encoding='utf-8') as f:
        csv_data = csv.reader(f)
        for row in csv_data:
            csv_list.append(row)
    return csv_list

# 读取yaml文件
def read_file(yml_file):
    try:
        path = BASE_DIR+yml_file
        with open(path,encoding='utf-8') as f:
            caseinfo = yaml.load(f,Loader=yaml.FullLoader)
            if len(caseinfo)>=2:
                return caseinfo
            else:
                caseinfo_keys = dict(*caseinfo).keys()
                if 'parameters' in caseinfo_keys:
                    new_caseinfo = analysis_parameters(*caseinfo)
                    return new_caseinfo
                else:
                    return caseinfo
    except Exception as f:
        error_log("读取用例文件报错:异常信息:%s"%str(traceback.format_exc()))


# 分析参数化
def analysis_parameters(caseinfo):
    try:
        caseinfo_keys = dict(caseinfo).keys()
        if 'parameters' in caseinfo_keys:
            for key, value in dict(caseinfo['parameters']).items():
                caseinfo_str = json.dumps(caseinfo)
                key_list = str(key).split('-')
                # 规范csv数据的写法
                length_flag = True
                csv_list = read_csv_file(value)
                one_row_data = csv_list[0]
                for csv_data in csv_list:
                    if len(csv_data) != len(one_row_data):
                        length_flag = False
                        break
                # 解析
                new_caseinfo = []
                if length_flag:
                    for x in range(1, len(csv_list)):  # x代表行
                        temp_caseinfo = caseinfo_str
                        for y in range(0, len(csv_list[x])):  # y代表列
                            if csv_list[0][y] in key_list:
                                temp_caseinfo = temp_caseinfo.replace("$csv{" + csv_list[0][y] + "}", csv_list[x][y])
                        new_caseinfo.append(json.loads(temp_caseinfo))
                return new_caseinfo
        else:
            return caseinfo
    except Exception as f:
        error_log("分析parameters参数异常:异常信息:%s"%str(traceback.format_exc()))


if __name__ == '__main__':
    print(read_file('/testcases/weixin/get_token.yml'))


  • requests_util.py:请求方式方法封装
# -*- coding: UTF-8 -*-
import json
import re
import traceback
import jsonpath
import requests
# from common.parameters_until import read_file
from common.logger_util import my_log, error_log
from common.yaml_util import *
from debugtalk import DebugTalk


class Requestutil:

    session = requests.session()

    def __init__(self):
        self.base_url =""
        self.last_headers={}

    # 规范功能测试YAML测试用例文件的写法
    def analysis_yaml(self,caseinfo):
        try:
            # 1.必须有的四个一级关键字:name,base_url,requests,validate
            caseinfo_keys = dict(caseinfo).keys()
            if 'name' in caseinfo_keys and 'base_url' in caseinfo and 'request' in caseinfo and 'validate' in caseinfo:
                # 2.request关键字必须包含两个二级关键字:method,url
                request_keys = dict(caseinfo['request']).keys()
                if 'method' in request_keys and 'url' in request_keys:
                    # 参数(params,data,json),请求头,文件上传这些都不能约束.
                    name = caseinfo['name']
                    self.base_url = caseinfo['base_url']
                    method = caseinfo['request']['method']
                    del caseinfo['request']['method']
                    url = caseinfo['request']['url']
                    del caseinfo['request']['url']
                    headers = None
                    if jsonpath.jsonpath(caseinfo,'$..headers'):
                        headers = caseinfo['request']['headers']
                        del caseinfo['request']['headers']
                    files = None
                    if jsonpath.jsonpath(caseinfo, '$..files'):
                        files = caseinfo['request']['files']
                        for key,value in dict(files).items():
                            files[key] = open(value,'rb')
                        del caseinfo['request']['files']
                    # 把method,url,headers,files这四个数据从caseinfo['request']去掉之后再把剩下的传给kwargs
                    res = self.send_request(name=name,method=method,url=url,headers=headers,files=files,**caseinfo['request'])
                    return_text = res.text
                    status_code = res.status_code
                    my_log("响应文本信息:%s"%return_text)
                    my_log("响应json信息:%s"%res.json())
                    # 提取接口关联的变量,既要支持正则表达式,又要支持json提取
                    if 'extract' in caseinfo_keys:
                        for key,value in dict(caseinfo['extract']).items():
                            # 正则表达式提取
                            if '(.*?)' in value or '(.+?)' in value:
                                ze_value = re.search(value,return_text)
                                if ze_value:
                                    extract_data = {key:ze_value.group(1)}
                                    write_file('/config/extract.yml',extract_data)
                                    print(extract_data)
                            else:   # json提取
                                return_json = res.json()  # 前提是要返回json格式
                                extract_data = {key:return_json[value]}
                                write_file('/config/extract.yml', extract_data)
                                print(extract_data)
                    # 断言的封装
                    yq_result = caseinfo['validate']
                    sj_result = res.json()
                    self.validate_result(yq_result, sj_result, status_code)
                else:
                    error_log('request关键字必须包含两个二级关键字:method,url')
            else:
                error_log('必须有的四个一级关键字:name,base_url,request,validate')
        except Exception as f:
            error_log("分析YAML文件异常:异常信息:%s" % str(traceback.format_exc()))

    #  统一替换方法,data可以是url(string),也可以是参数(字典,字典中包含有列表),也可以是请求头(字典).
    def replace_value(self,data):
        # 字典类型转换成字符串
        if data and isinstance(data,dict):  # 如果data不为空并且数据类型为字典
            str_data = json.dumps(data)
        else:
            str_data = data
        # 替换值
        for i in range(1, str_data.count('{{') + 1):
            if "{{" in str_data and "}}" in str_data:
                start_index = str_data.index("{{")
                end_index = str_data.index("}}",start_index)
                old_value = str_data[start_index:end_index + 2]
                new_value = read_file("/config/extract.yml", old_value[2:-2])
                str_data = str_data.replace(old_value, new_value)
        # 还原数据类型
        if data and isinstance(data,dict):  # 如果data不为空并且数据类型为字典
            data = json.loads(str_data)
        else:
            data = str_data
        return data

    #  统一替换方法,data可以是url(string),也可以是参数(字典,字典中包含有列表),也可以是请求头(字典).
    def replace_load(self, data):
        # 字典类型转换成字符串
        if data and isinstance(data, dict): # 如果data不为空并且数据类型为字典
            str_data = json.dumps(data)
        else:
            str_data = data
        # 替换值
        for i in range(1, str_data.count('${') + 1):
            if "${" in str_data and "}" in str_data:
                start_index = str_data.index("${")
                end_index = str_data.index("}", start_index)
                old_value = str_data[start_index:end_index + 1]
                function_name = old_value[2:old_value.index('(')]
                args_value = old_value[old_value.index('(')+1:old_value.index(')')]
                # 反射(通过一个函数的字符串直接去调用这个方法)
                new_value = getattr(DebugTalk(),function_name)(*args_value.split(','))
                str_data = str_data.replace(old_value, str(new_value))
        # 还原数据类型
        if data and isinstance(data, dict):   # 如果data不为空并且数据类型为字典
            data = json.loads(str_data)
        else:
            data = str_data
        return data

    # 统一发送请求
    def send_request(self,name,method,url,headers=None,files=None,**kwargs):
        try:
            # 处理method
            self.last_method = str(method).lower()
            # 处理基础路径
            self.url=self.replace_load(self.base_url) + self.replace_value(url)
            # 处理请求头
            if headers and isinstance(headers,dict):
                self.last_headers=self.replace_value(headers)
            # 最核心的地方:请求数据如何去替换:可能是params,data,json
            for key,value in kwargs.items():
                if key in ['params','data','json']:
                    # 替换{{}}格式
                    value = self.replace_value(value)
                    # 替换${}格式
                    value = self.replace_load(value)
                    kwargs[key] = value
            # 收集日志
            my_log('-----------------接口请求开始-----------------')
            my_log("接口名称:%s"%name)
            my_log("请求方式:%s"%self.last_method)
            my_log("请求路径:%s"%self.url)
            my_log("请求头:%s"%self.last_headers)
            if 'params' in kwargs.keys():
                my_log("请求参数:%s"%kwargs['params'])
            elif 'data' in kwargs.keys():
                my_log("请求参数:%s"%kwargs['data'])
            elif 'json' in kwargs.keys():
                my_log("请求参数:%s"%kwargs['json'])
            my_log("文件上传:%s"%files)
            # 发送请求
            res = Requestutil.session.request(method=self.last_method,url=self.url,headers=self.last_headers,**kwargs)
            # print(res.request.headers)
            # print(res.text)
            # print(res.json())
            return res
        except Exception as f:
            error_log("发送请求异常:异常信息:%s"%str(traceback.format_exc()))

    # 断言封装
    def validate_result(self,yq_result,sj_result,status_code):
        try:
            '''
            :param yq_result:预期结果
            :param sj_result:实际结果
            :param status_code:实际状态码
            :return:
            '''
            # 收集日志
            my_log("预期结果:%s"%yq_result)
            my_log("实际结果:%s"%sj_result)
            #判断是否断言成功,0成功,1失败
            flag = 0
            # 解析ƒ
            if yq_result and isinstance(yq_result,list):
                for yq in yq_result:
                    for key,value in dict(yq).items():
                        # 判断断言方式
                        if key=='equals':
                            for assert_key,assert_value in dict(value).items():
                                if assert_key=='status_code':
                                    if status_code!=assert_value:
                                        flag=flag+1
                                        error_log("断言失败:"+assert_key+"不等于"+str(assert_value)+"")
                                else:
                                    key_list = jsonpath.jsonpath(sj_result,'$..%s'%assert_key)
                                    if key_list:
                                        if assert_value not in key_list:
                                            flag = flag + 1
                                            error_log("断言失败:"+assert_key+"不等于"+str(assert_value)+"")
                                    else:
                                        flag = flag + 1
                                        error_log("断言失败:返回结果中不存在"+assert_key+"")
                        elif key=='contains':
                            if value not in json.dumps(sj_result):
                                flag = flag + 1
                                error_log("断言失败:返回结果中不包含字符串"+value+"")
                        else:
                            error_log('框架不支持此断言方式')
            assert flag==0
            my_log('接口请求成功')
            my_log('-----------------接口请求结束-----------------\n')
        except Exception as f:
            my_log('接口请求失败')
            my_log('-----------------接口请求结束-----------------\n')
            error_log("断言异常:异常信息:%s" % str(traceback.format_exc()))


if __name__ == '__main__':
    # url = "/cgi-bin/tags/update?access_token={{access_token}}&a=c{{csrf_token}}"
    # for i in range(1,url.count("{{")+1):
    #     if "{{" in url and "}}" in url:
    #         start_index = url.index("{{")
    #         end_index = url.index("}}")
    #         old_value = url[start_index:end_index+2]
    #         new_value = read_file('/config/extract.yml',old_value[2:-2])
    #         url = url.replace(old_value,new_value)
    #
    #         print(old_value,new_value)
    #         print(url)


    # dict_data = {'name': '获取access_token统一鉴权码', 'base_url': 'https://api.weixin.qq.com', 'request': {'method': 'GET', 'url': '/cgi-bin/token', 'params': {'grant_type': 'client_credential', 'appid': 'wx9b755d429f6fb216', 'secret': 'b963db0b97c8487b0cb920a240bd78e3'}}, 'validate': [{'eq': ['status_code', 200]}]}
    # # print(dict_data.pop('name'))
    # del dict_data['name']
    # print(dict_data)

    json_data = {"tag": {"id": 100, "name": "CesareCheung${get_random_number(100000,999999)}" }}
    result = Requestutil('base', 'base_weixin_url').replace_load(json_data)
    print(result)


  • yaml_util.py:文件读取写入方法
# -*- coding: UTF-8 -*-
import os
import yaml
from common.get_path import *


# 读取yml文件
def read_file(yml_file,one_node=None,two_node=None):
    path = BASE_DIR+yml_file
    with open(path,encoding='utf-8') as f:
        value = yaml.load(f,Loader=yaml.FullLoader)
        if one_node and two_node:
            return value[one_node][two_node]
        elif one_node:
            return value[one_node]
        else:
            return value

# 写入yml文件
def write_file(yml_file,data):
    path = BASE_DIR+yml_file
    with open(path,encoding='utf-8',mode='a') as f:
        yaml.dump(data, stream=f,allow_unicode=True)

# 清空yml文件
def clean_file(yml_file):
    path = BASE_DIR+yml_file
    with open(path,encoding='utf-8',mode='w') as f:
        f.truncate()

if __name__ == '__main__':
    # print(read_file('/config.yml',"base",'base_info_url'))
    print(read_file('/config/config.yml','log','log_name'))



本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:/a/736042.html

如若内容造成侵权/违法违规/事实不符,请联系我们进行投诉反馈qq邮箱809451989@qq.com,一经查实,立即删除!

相关文章

【数据结构导论】自考笔试题:伪代码练习题汇总 1

目录 一、开源项目推荐 二、线性表的基本运算在单链表上的实现 &#xff08;1&#xff09;初始化 &#xff08;2&#xff09;插入 p 指向的新结点的操作 &#xff08;3&#xff09;删除 *p 节点 三、循环链表 &#xff08;1&#xff09;在单链表中 &#xff08;2&…

双指针问题2

文章目录 1. 有效三角形的个数&#xff08;611&#xff09;2. 查找总价格为目标值的两个商品&#xff08;LCR179&#xff09;3. 三数之和&#xff08;15&#xff09;4. 四数之和&#xff08;18&#xff09; 1. 有效三角形的个数&#xff08;611&#xff09; 题目描述&#xff…

Stable Diffusion初体验——基于机器学习通过神经网络的强大AI平台

文章目录 前言最新热门活动&#xff01;&#xff01;平台介绍 一.创建应用 Stable Diffusion WebUI初始化上传模型&#xff0c;VAE&#xff0c;lora 介绍sd模型&#xff0c;vae&#xff0c;lora模型进入应用文生图工作区调参区图生图 结语小程序活动——6.20火热上线&#x1f5…

消息队列MQ相关面试题

消息队列MQ相关面试题 1 RabbitMQ 1.1 你们项目中哪里用到了RabbitMQ ? 难易程度&#xff1a;☆☆☆ 出现频率&#xff1a;☆☆☆☆ 我们项目中很多地方都使用了RabbitMQ , RabbitMQ 是我们项目中服务通信的主要方式之一 , 我们项目中服务通信主要有二种方式实现 : 通过Fei…

阿里云服务器618没想到这么便宜,买早了!

2年前&#xff0c;我买了个服务器&#xff0c;租用服务器&#xff08;ECS5&#xff09;和网络宽带&#xff08;1M&#xff09;&#xff0c;可以说是非常非常低的配置了。 当时5年的折扣力度最大&#xff0c;但是打完折后&#xff0c;价格依然要近3000多元。 最近看到阿里云618活…

LVGL8.3动画图像(太空人)

LVGL8.3 动画图像 1. 动画图像本质 我们知道电影属于视频&#xff0c;而电影的本质是将一系列动作的静态图像进行快速切换而呈现出动画的形式&#xff0c;也就是说动画本质是一系列照片。所以 lvgl 依照这样的思想而定义了动画图像&#xff0c;所以在 lvgl 中动画图像类似于普…

element-plus 表单组件 之element-form

elment-plus的表单组件的标签有el-form,el-form-item。 单个el-form标签内包裹若干个el-form-item,el-form-item包裹具体的表单组件&#xff0c;如输入框组件&#xff0c;多选组件&#xff0c;日期组件等。 el-form组件的主要作用是&#xff1a;提供统一的布局给其他表单组件&…

setInterval 定时任务执行时间不准验证

一般在处理定时任务的时候都使用setInterval间隔定时调用任务。 setInterval(() > {console.log("interval"); }, 2 * 1000);我们定义的是两秒执行一次&#xff0c;但是浏览器实际执行的间隔时间只多不少。这是由于浏览器执行 JS 是单线程模式&#xff0c;使用se…

apksigner jarsigner.md

关于作者&#xff1a;CSDN内容合伙人、技术专家&#xff0c; 从零开始做日活千万级APP。 专注于分享各领域原创系列文章 &#xff0c;擅长java后端、移动开发、商业变现、人工智能等&#xff0c;希望大家多多支持。 目录 一、导读二、概览三、apksigner3.2 为 APK 签名3.3 验证…

电商爬虫API的定制开发:满足个性化需求的解决方案

一、引言 随着电子商务的蓬勃发展&#xff0c;电商数据成为了企业决策的重要依据。然而&#xff0c;电商数据的获取并非易事&#xff0c;特别是对于拥有个性化需求的企业来说&#xff0c;更是面临诸多挑战。为了满足这些个性化需求&#xff0c;电商爬虫API的定制开发成为了解决…

2024年综合艺术与媒体传播国际会议(ICIAMC 2024)

2024年综合艺术与媒体传播国际会议(ICIAMC 2024) 2024 International Conference on Integrated Arts and Media Communication (ICIAMC 2024) 会议地点&#xff1a;贵阳&#xff0c;中国 网址&#xff1a;www.iciamc.com 邮箱: iciamcsub-conf.com 投稿主题请注明:ICIAMC…

使用AGG里面的clip_box函数裁剪画布, 绘制裁剪后的图形

// 矩形裁剪图片, 透明 void agg_testImageClipbox_rgba32(unsigned char* buffer, unsigned int width, unsigned int height) {// 创建渲染缓冲区 agg::rendering_buffer rbuf;// BMP是上下倒置的&#xff0c;为了和GDI习惯相同&#xff0c;最后一个参数是负值。rbuf.attach…

使用API有效率地管理Dynadot域名,为文件夹中的域名统一设置whois信息

关于Dynadot Dynadot是通过ICANN认证的域名注册商&#xff0c;自2002年成立以来&#xff0c;服务于全球108个国家和地区的客户&#xff0c;为数以万计的客户提供简洁&#xff0c;优惠&#xff0c;安全的域名注册以及管理服务。 Dynadot平台操作教程索引&#xff08;包括域名邮…

基于SSM+Vue+微信小程序的大学生就业平台系统+毕业论文

项目包含前台和后台两部分&#xff1a;多角色登录&#xff0c;功能完善&#xff0c;界面优美 前台主要功能实现&#xff1a;首页列表查看、求职信息管理、简历管理、面试邀请管理、个人中心等 后台主要功能实现&#xff1a;首页、个人中心、学生管理、企业管理、企业类型管理…

若依 ruoyi 显示隐藏搜索框 显示隐藏列

一、 显示隐藏搜索框 页面搜索关键字 showSearch&#xff0c;设置是否显示 隐藏&#xff1a; 显示&#xff1a; 二、自定义设置 显示隐藏列 1. 页面搜索关键字 right-toolbar&#xff0c;新增&#xff1a; :columns"columns" 2. js下 data(){return{}}中新增&am…

如何覆盖!important修饰的属性

最简单的方法 如果这个!important修饰的属性 是自己的写的&#xff0c;去掉这种写法&#xff0c;使用优先级的方式来写这个属性&#xff08;.outter .inner 的优先级就会比 。outter的优先级高&#xff09; 复杂的方法&#xff1a;用魔法打败魔法 但是这个样式来自于全局css&am…

【计算机毕业设计】185餐厅点餐微信小程序

&#x1f64a;作者简介&#xff1a;拥有多年开发工作经验&#xff0c;分享技术代码帮助学生学习&#xff0c;独立完成自己的项目或者毕业设计。 代码可以私聊博主获取。&#x1f339;赠送计算机毕业设计600个选题excel文件&#xff0c;帮助大学选题。赠送开题报告模板&#xff…

百度网盘的文件该怎么整理?不如试试这个整理工具

科学的文件架构 一键生成文件夹层级工具 极具妥帖的秩序感受 又是一周&#xff0c;好久不见&#xff0c;正琢磨着这次给大家带来点什么好东西&#xff0c;突然百度网盘的整理让我头疼不已&#xff0c;从我记事以来&#xff0c;这网盘已经整理过N遍了&#xff0c;总是乱了整理…

【权威出版/投稿优惠】2024年水利水电与能源环境科学国际会议(WRHEES 2024)

2024 International Conference on Water Resources, Hydropower, Energy and Environmental Science 2024年水利水电与能源环境科学国际会议 【会议信息】 会议简称&#xff1a;WRHEES 2024 大会时间&#xff1a;点击查看 截稿时间&#xff1a;点击查看 大会地点&#xff1a;…

day2-web安全漏洞攻防-基础-弱口令、HTML注入(米斯特web渗透测试)

day2-web安全漏洞攻防-基础-弱口令、HTML注入&#xff08;米斯特web渗透测试&#xff09; 1&#xff0c;漏洞2&#xff0c;弱口令3&#xff0c;爆破&#xff08;1&#xff09;Burpsuite&#xff08;2&#xff09;攻击类型 4&#xff0c;HTML针剂注入 1&#xff0c;漏洞 挖掘和利…