AI写代码：我让元宝将500+行开源c代码翻译成100+行python代码，完美实现功能-CSDN博客

本文链接：https://blog.csdn.net/zhfcmx1/article/details/139769291

关注公众号，赠送AI/Python/Linux资料，对AI智能体有兴趣的朋友也可以添加一起交流

webbench是一个网站的压力测试开源软件。我经常用这个软件对网站做压力测试，顺便借用产生的HTTP报文测试设备通信。

现在需要把webbench功能融入到测试代码中去，测试代码是用python写的。所以需要把webbench转成Python的实现。webbench的代码不复杂，但既然有了AI，这个活就交给AI来干。看看腾讯元宝能不能做到

在线网页地址：https://yuanbao.tencent.com/chat/naQivTmsDa

首先是提示词，让元宝变身编程语言转换专家。熟悉c以及Python代码编写。

然后我将webbench的源代码以文件的形式上传，让元宝学习，并要求转成Python代码

元宝只是提供一个类似伪代码的实现。要求继续完善

这一次元宝完善了代码，但是从代码来看，很多以前在c使用的函数并没有转换成python的实现。要求元宝用python代码来覆盖并且用python的库

这一次元宝提供的代码功能齐全

import socketimport multiprocessingimport argparseimport sysfrom time import time, strftimefrom multiprocessing import Lockfrom multiprocessing import Manager
# 全局变量http10 = 1  # 默认使用 HTTP/1.0method = 'GET'  # 默认方法为 GETclients = 1force = 0force_reload = 0proxyport = 80proxyhost = Nonebenchtime = 30url = ''success_count = 0failed_count = 0bytes_transferred = 0
# 请求模板def build_request(url, method='GET', force_reload=force_reload):    request = f"{method} {url} HTTP/1.1\r\n"    request += f"User-Agent: Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.3"    request += " Connection: close\r\n\r\n"    print(request)    return request.encode()
# 解析URLdef parse_url(url):    protocol, rest = url.split("://", 1)    host = rest.split('/')[0]    return protocol, host
# 基准测试核心函数def benchcore(host, port, request, benchtime,speed_counter,failed_counter,bytes_counter):    global success_count, failed_count, bytes_transferred    start_time = time()    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)    s.settimeout(5)    try:        s.connect((host, port))    except ConnectionRefusedError:        print("Connect to server failed. Aborting benchmark.")        return None    while (time() - start_time) < benchtime:        try:            s.sendall(request)            if force:                return            while True:                try:                    data = s.recv(1500)                except BaseException as e:                    continue                # print(data)                if not data:                    failed_counter['value'] += 1                    break                with Lock():                    bytes_counter['value'] += len(data)                    speed_counter['value'] += 1        except socket.error:            with Lock():                failed_counter['value'] += 1            return    s.close()
# 子进程执行的基准测试def child_benchcore(host, proxyport, request, benchtime,speed_counter,failed_counter,bytes_counter):    host=host    port=proxyport    request=request    benchtime=benchtime    benchcore(host, port, request, benchtime,speed_counter,failed_counter,bytes_counter)
# 解析命令行参数def parse_args():    parser = argparse.ArgumentParser(description="Simple Web Benchmark")    parser.add_argument("url", help="URL to benchmark")    parser.add_argument("-c", "--clients", type=int, default=1, help="Number of clients")    parser.add_argument("-t", "--time", type=int, default=30, help="Run benchmark for <sec> seconds")    parser.add_argument("-f", "--force", action="store_true", help="Don't wait for reply from server")    parser.add_argument("-r", "--reload", action="store_true", help="Send reload request - Pragma: no-cache")    parser.add_argument("--http09", action="store_true", help="Use HTTP/0.9 style requests")    parser.add_argument("--http10", action="store_true", help="Use HTTP/1.0 protocol")    parser.add_argument("--http11", action="store_true", help="Use HTTP/1.1 protocol")    return parser.parse_args()
# 主函数def main():    global benchtime, clients, force, force_reload, http10, url, proxyhost, proxyport    args = parse_args()    url = args.url    clients = args.clients    benchtime = args.time    force = args.force    force_reload = args.reload
    if args.http09:        http10 = 0    elif args.http10:        http10 = 1    elif args.http11:        http10 = 2
    _, host = parse_url(url)    if ':' in host:  # 如果URL包含端口，解析端口号        host, port_str = host.split(':')        proxyport = int(port_str)
    request = build_request(url, 'GET', force_reload)    manager = Manager()    speed_counter = manager.dict(value=0)    failed_counter = manager.dict(value=0)    bytes_counter = manager.dict(value=0)
    processes = []    for _ in range(clients):        p = multiprocessing.Process(target=child_benchcore, args=(host, proxyport, request, benchtime,speed_counter,failed_counter,bytes_counter,))        p.start()        processes.append(p)
    for p in processes:        p.join()
    print(f"Benchmark completed at {strftime('%Y-%m-%d %H:%M:%S')}\n")    speed = speed_counter['value']    failed = failed_counter['value']    bytes_received = bytes_counter['value']
    print(f"\nSpeed={speed} pages/min, {bytes_received / benchtime:.2f} bytes/sec.")    print(f"Requests: {speed} succeeded, {failed} failed.")
if __name__ == "__main__":    main()