优化base_request.py，增加日志提示

2023-05-22 09:24:53 +08:00
parent e4516935a4
commit 5e87b2b5d2
3 changed files with 16 additions and 113 deletions
@@ -1,110 +0,0 @@
-# -*- coding: utf-8 -*-
-# @Version: Python 3.9
-# @Time    : 2023/2/1 14:16
-# @Author  : chenyinhua
-# @File    : data_handle.py
-# @Software: PyCharm
-# @Desc: 数据处理
-from loguru import logger
-from jsonpath import jsonpath
-import re
-from faker import Faker
-from string import Template
-from config.global_vars import GLOBAL_VARS
-
-faker = Faker()
-
-
-def case_data_replace(content):
-    """
-    用例数据替换的方法
-    :param content: 原始的字符串内容
-    return content： 替换表达式后的字符串
-    """
-    if content is None:
-        return None
-    logger.debug(f"-----Start-----开始进行字符串替换: 初始字符串为：{content}")
-    if len(content) != 0:
-        # safe_substitute() 方法会保留没有被替换的占位符，不会抛出 KeyError 异常。
-        # 所以，如果 content 中不存在占位符，使用 safe_substitute() 方法进行替换后，得到的结果和原始字符串是一样的。
-        content = Template(str(content)).safe_substitute(GLOBAL_VARS)
-        for func in re.findall('\\${(.*?)}', content):
-            content = content.replace('${%s}' % func, exec_func(func))
-            try:
-                content = content.replace('${%s}' % func, exec_func(func))
-            except Exception as e:
-                logger.error(f"-----END-----替换数据时出现了异常：{e}")
-        logger.debug(f"-----END-----字符串替换完成: 新字符串为：{content}")
-        return content
-
-
-def json_extractor(obj: dict, expr: str = '.'):
-    """
-    :param obj :json/dict类型数据
-    :param expr: 表达式, . 提取字典所有内容， $.test_api_case 提取一级字典case， $.test_api_case.data 提取case字典下的data
-    :return result: 提取的结果，未提取到返回 None
-    """
-    try:
-        result = jsonpath(obj, expr)[0]
-        logger.debug("提取响应内容成功，提取表达式为： {} 提取值为 {}".format(expr, result))
-    except Exception as e:
-        logger.exception(e)
-        logger.exception(f'未提取到内容，请检查表达式是否错误！提取表达式为：{expr} 提取数据为 {obj}')
-        result = None
-    return result
-
-
-def re_extract(obj: str, expr: str = '.'):
-    """
-    :param obj : 字符串数据
-    :param expr: 正则表达式
-    :return result: 提取的结果，未提取到返回 None
-    """
-    try:
-        result = re.findall(expr, obj)[0]
-    except Exception as e:
-        result = None
-        logger.exception(e)
-        logger.exception(f'未提取到内容，请检查表达式是否错误！提取表达式为：{expr} 提取数据为 {obj}')
-    return result
-
-
-def exec_func(func) -> str:
-    """
-    :params func 字符的形式调用函数
-    : return 返回的转换成函数执行的结果,已字符串格式返回
-    """
-    result = eval(func)
-    return str(result)
-
-
-# 将"[1,2,3]" 或者"{'k':'v'}" -> [1,2,3], {'k':'v'}
-def eval_data(data):
-    """
-    执行一个字符串表达式，并返回其表达式的值
-    """
-    try:
-        if hasattr(eval(data), "__call__"):
-            return data
-        else:
-            return eval(data)
-    except Exception:
-        return data
-
-
-def eval_data_process(data):
-    """
-    将数据中的字符串表达式处理后更新其值为表达式
-    """
-    # 如果目标数据是字符串，直接尝试eval
-    if isinstance(data, str):
-        data = eval_data(data)
-    # 如果目标数据是列表，遍历列表的每一个数据，再用递归的方法处理每一个item
-    if isinstance(data, list):
-        for index, item in enumerate(data):
-            data[index] = eval_data_process(eval_data(item))
-    # 如果目标数据是字典，遍历字典的每一个值，再用递归的方法处理每一个value
-    elif isinstance(data, dict):
-        for key, value in data.items():
-            data[key] = eval_data_process(eval_data(value))
-    return data
@@ -30,17 +30,30 @@ class BaseRequest:
        :return: 响应对象
        """
        try:
-            logger.info(f"-----Start-----进行接口请求，并获取接口响应数据，请求数据：{type(req_data)} || {req_data}-----")
+            logger.info(f"\n======================================================\n" \
+                        "-------------Start：请求前--------------------\n"
+                        f"用例标题: {req_data.get('title', None)}\n" \
+                        f"请求路径: {req_data.get('url', None)}\n" \
+                        f"请求方式: {req_data.get('method', None)}\n" \
+                        f"请求头:   {req_data.get('headers', None)}\n" \
+                        f"请求关键字: {req_data.get('pk', None)}\n" \
+                        f"请求内容: {req_data.get('payload', None)}\n" \
+                        f"请求文件: {req_data.get('files', None)}\n" \
+                        "=====================================================")
            res = cls.send_api_request(
                url=req_data.get("url"),
                method=req_data.get("method").lower(),
                pk=req_data.get("pk", None),
-                header=req_data.get("header"),
+                header=req_data.get("headers"),
                payload=req_data.get("payload"),
                files=req_data.get("files")
            )
-            logger.info(f"-----End-----接口请求结束，请求响应数据：{res.text}-----")
+            logger.info(f"\n======================================================\n" \
+                        "-------------End：请求后--------------------\n"
+                        f"响应数据: {res.text}\n" \
+                        "=====================================================")
        except requests.exceptions.RequestException as e:
+            logger.error(f"请求出错，{str(e)}")
            raise ValueError(f"请求出错，{str(e)}")

        return res