使用套接字和Argparse的Python HTTP响应

import socket
import argparse

def send_http_request(url, method='GET', headers=None):
    # 解析URL
    host, path = parse_url(url)
    
    # 建立与服务器的连接
    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    sock.connect((host, 80))
    
    # 构造HTTP请求
    request = f"{method} {path} HTTP/1.1\r\n"
    if headers:
        for header, value in headers.items():
            request += f"{header}: {value}\r\n"
    request += "\r\n"
    
    # 发送HTTP请求
    sock.sendall(request.encode())
    
    # 接收服务器返回的HTTP响应
    response = b""
    while True:
        data = sock.recv(1024)
        if not data:
            break
        response += data
    
    # 关闭套接字连接
    sock.close()
    
    return response.decode()

def parse_url(url):
    # 解析URL，提取主机和路径
    # 这里简化处理，不考虑HTTPS和端口号
    parts = url.split('/')
    host = parts[2]
    path = '/' + '/'.join(parts[3:])
    return host, path

if __name__ == '__main__':
    # 解析命令行参数
    parser = argparse.ArgumentParser(description='Python HTTP Client')
    parser.add_argument('url', help='URL of the HTTP server')
    parser.add_argument('-m', '--method', default='GET', help='HTTP request method')
    parser.add_argument('-H', '--header', action='append', help='HTTP request headers')
    args = parser.parse_args()
    
    # 构造请求头字典
    headers = {}
    if args.header:
        for header in args.header:
            key, value = header.split(':')
            headers[key.strip()] = value.strip()
    
    # 发送HTTP请求并打印响应
    response = send_http_request(args.url, args.method, headers)
    print(response)