import os
import requests
import json
"""測(cè)試數(shù)據(jù)路徑管理"""
SCRIPTS_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
DATAS_DIR = os.path.join(SCRIPTS_DIR, "datas")
Testcase_md= os.path.join(DATAS_DIR,'testcase.md')
"""示例代碼"""
def sendGpt_getmd(content):
    # 設(shè)置請(qǐng)求頭中的token
    headers = {
        "Content-Type": "application/json"
    }
    # payload= 設(shè)置請(qǐng)求主體中的參數(shù)和數(shù)據(jù)，根據(jù)請(qǐng)求的參數(shù)格式傳參即可
    payload = {"參數(shù)": "參數(shù)值"}
    # 發(fā)送POST請(qǐng)求
    response = requests.post("https://xxxxx", headers=headers,
                             data=json.dumps(payload))
    # 解析響應(yīng)結(jié)果
    if response.status_code == 200:
        result = response.text
        print(result)
        with open(Testcase_md, "w", encoding="utf-8") as f:
            f.write(result)
    else:
        print("請(qǐng)求失??！")
if __name__ == '__main__':
  # 示例用法:keydes為功能標(biāo)題，casecontentTemp為發(fā)送給gpt的模板話術(shù)，在使用時(shí)可以將keydes設(shè)置為變量，由前端傳參。另外可以根據(jù)需要修改casecontentTemp內(nèi)容
  # 僅包含一級(jí)標(biāo)題，二級(jí)標(biāo)題，無(wú)序列表格式的Markdown形式：該格式不可修改，因?yàn)楹罄m(xù)需要以Markdown格式轉(zhuǎn)換成xmind文件
  keydes='上傳附件'
  casecontentTemp="僅包含一級(jí)標(biāo)題，二級(jí)標(biāo)題，無(wú)序列表格式的Markdown形式發(fā)送給我。請(qǐng)寫(xiě)出 一級(jí)標(biāo)題為'"+keydes+"'的測(cè)試用例，包含功能測(cè)試（正向/逆向/異常場(chǎng)景），性能測(cè)試，安全測(cè)試，兼容測(cè)試"
  sendGpt_getmd(casecontentTemp)

import markdown
from bs4 import BeautifulSoup
import xmind
def md_to_xmind(md_file, xmind_file):
    # 讀取MD文件
    with open(md_file, 'r', encoding='utf-8') as f:
        md = f.read()
    # 解析MD文件
    html = markdown.markdown(md)
    # 創(chuàng)建XMind文件
    workbook = xmind.load(xmind_file)
    # 獲取根節(jié)點(diǎn)
    root_topic = workbook.getPrimarySheet().getRootTopic()
    # 遞歸添加節(jié)點(diǎn)
    def add_topic(parent_topic, node):
        # 添加節(jié)點(diǎn)
        topic = parent_topic.addSubTopic()
        title = node.get('title', '')
        topic.setTitle(title)
        # 添加文本
        if 'html' in node:
            topic.setPlainNotes(node['html'])
        # 遞歸添加子節(jié)點(diǎn)
        if 'children' in node:
            for child in node['children']:
                add_topic(topic, child)
    # 解析HTML，并添加節(jié)點(diǎn)
    soup = BeautifulSoup(html, 'html.parser')
    rootmap_node = {'children': []}
    root_node = None
    current_node = None
    for tag in soup.children:
        if tag.name == 'h1':
            # 創(chuàng)建根節(jié)點(diǎn)
            root_node = {'title': tag.string, 'children': []}
            current_node = root_node
        elif tag.name == 'h2':
            new_node = {'title': tag.string, 'children': []}
            root_node['children'].append(new_node)
            current_node = new_node
        elif tag.name == 'p':
            current_node['html'] = str(tag)
        elif tag.name == 'ul':
            for li in tag.children:
                text = li.text.strip()
                if len(text) > 0:
                    li_node = {'title': text, 'children': []}
                    current_node['children'].append(li_node)
        elif tag.name == 'ol':
            for li in tag.children:
                text = li.text.strip()
                if len(text) > 0:
                    li_node = {'title': text, 'children': []}
                    current_node['children'].append(li_node)
    # 添加節(jié)點(diǎn)
    for node in root_node['children']:
        add_topic(root_topic, node)
    # 修改根節(jié)點(diǎn)的名稱(chēng)
    root_topic.setTitle(root_node['title'])
    # 保存XMind文件
    xmind.save(workbook, xmind_file)
if __name__ == '__main__':
  # 示例用法 example.md為步驟1生成的文件，通過(guò)md_to_xmind方法調(diào)用將.md文件轉(zhuǎn)換成xmind文件
  md_to_xmind('example.md', 'example.xmind')
 

import os
from scripts.datas.mdtoxmind import md_to_xmind
from scripts.datas.sendGpt import sendGpt_getmd
"""測(cè)試數(shù)據(jù)路徑管理"""
SCRIPTS_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
DATAS_DIR = os.path.join(SCRIPTS_DIR, "datas")
Testcase_md= os.path.join(DATAS_DIR,'testcase.md')
Testcase_xmind= os.path.join(DATAS_DIR,'testcase.xmind')
def oneTocase(keydes):
   casecontentTemp="僅包含一級(jí)標(biāo)題，二級(jí)標(biāo)題，無(wú)序列表格式的Markdown形式發(fā)送給我。請(qǐng)寫(xiě)出 一級(jí)標(biāo)題為'"+keydes+"'的測(cè)試用例，包含功能測(cè)試（正向/逆向/異常場(chǎng)景），性能測(cè)試，安全測(cè)試，兼容測(cè)試"
    sendGpt_getmd(casecontentTemp)
    md_to_xmind(Testcase_md, Testcase_xmind)
if __name__ == '__main__':
  # 示例用法
  keydes='上傳附件'
  oneTocase(keydes)