#!/usr/bin/env/ python3
# -*- coding:utf-8 -*-
"""
@Project: apiAutoTest
@File  :recording.py
@Author:zy7y
@Date  :2021/5/21 22:07
@Desc  : 录制接口，生成用例文件
基于mitmproxy实现，会包含css/html/png等后缀链接
参考资料：
https://blog.wolfogre.com/posts/usage-of-mitmproxy/
https://www.cnblogs.com/liuwanqiu/p/10697373.html
"""

import json

import mitmproxy.http
import xlwt

# 上传文件接口不能录入文件参数 , excel单元格限制： Exception: String longer than 32767 characters
from mitmproxy import ctx


class Counter:
    def __init__(self, filter_url: str, filename: str = "case_data1.xls"):
        """
        基于mitmproxy抓包生成用例数据
        :param filter_url: 需要过滤的url
        :param filename: 生成用例文件路径
        """
        self.url = filter_url
        self.excel_row = [
            "用例标题",
            "是否执行",
            "请求头",
            "接口地址",
            "请求方式",
            "入参关键字",
            "上传文件",
            "请求数据",
            "提取参数",
            "后置sql",
            "预期结果",
        ]
        self.cases = [self.excel_row]
        self.counter = 1
        self.file = filename

    def response(self, flow: mitmproxy.http.HTTPFlow):
        """
        mitmproxy抓包处理响应，在这里汇总需要数据, 过滤 包含指定url，并且响应格式是 json的
        :param flow:
        :return:
        """
        if (
            self.url in flow.request.url
            and "json" in flow.response.headers["Content-Type"]
        ):
            # 标题
            title = "mitmproxy录制接口" + str(self.counter)
            try:
                token = flow.request.headers["Authorization"]
            except KeyError:
                token = ""
            header = json.dumps({"Authorization": token})
            data = flow.request.text
            # 请求地址，config.yaml 里面基准环境地址 写 空字符串
            method = flow.request.method.lower()
            url = flow.request.url
            try:
                content_type = flow.request.headers["Content-Type"]
            except KeyError:
                content_type = ""
            if "form" in content_type:
                data_type = "data"
            elif "json" in content_type:
                data_type = "json"
            else:
                data_type = "params"
                if "?" in url:
                    data = url.split("?")[1]
            data = Counter.handle_form(data)
            # 预期结果
            try:
                expect = json.dumps(
                    {".": json.loads(flow.response.text)}, ensure_ascii=False
                )
            except Exception as e:
                ctx.log.error(e)
                expect = "{}"
            # 日志
            ctx.log.info(url)
            ctx.log.info(header)
            ctx.log.info(content_type)
            ctx.log.info(method)
            ctx.log.info(data)
            ctx.log.info(flow.response.text)
            case = [
                title,
                "FALSE",
                header,
                url.split("?")[0],
                method,
                data_type,
                "",
                data,
                "",
                "",
                expect,
            ]
            self.cases.append(case)
            self.counter += 1
            # 文件末尾追加
            self.excel_cases()

    def excel_cases(self):
        """
        对二维列表cases进行循环并将内容写入单元格中
        :return:
        """
        workbook = xlwt.Workbook()
        worksheet = workbook.add_sheet("用例数据")
        for x in range(len(self.cases)):
            for y in range(len(self.cases[x])):
                worksheet.write(x, y, self.cases[x][y])
        try:
            workbook.save(self.file)
        except Exception as e:
            print(e)

    @classmethod
    def handle_form(cls, data: str):
        """
        处理 Content-Type:	application/x-www-form-urlencoded
        默认生成的数据 username=admin&password=123456
        :param data: 获取的data 类似这样  username=admin&password=123456
        :return:
        """
        data_dict = {}
        if data.startswith("{") and data.endswith("}"):
            return data
        try:
            for i in data.split("&"):
                data_dict[i.split("=")[0]] = i.split("=")[1]
            return json.dumps(data_dict)
        except IndexError:
            return ""


addons = [Counter("https://gitee.com/zy7y/apiAutoTest")]

"""

mitmweb -s recording.py 启动
ctrl + C 停止 并生成完整用例
"""