tcp文件下载

服务器端

import socket


def send_file_2_client(new_client_socket,client_addr):
    # 接收信息
    file_name = new_client_socket.recv(1024).decode("utf-8")

    print("客户%s要下载的文件是：%s" % (str(client_addr), file_name))

    # 定义发送的内容
    send_content = None

    # 打开文件
    try:
        f = open(file_name,"rb")
        send_content = f.read()
        f.close()
    except Exception as ret:
        print("没有该文件信息")

    # 不做判断 如果没有该文件会报错：TypeError: a bytes-like object is required, not 'NoneType'
    if send_content:
        new_client_socket.send(send_content)


def main():
    # 创建套接字
    tcp_server_socket = socket.socket(socket.AF_INET,socket.SOCK_STREAM)

    # 绑定端口
    tcp_server_socket.bind(("",7788))

    # 被动监听
    tcp_server_socket.listen(128)

    while True:
        # 接收客户端信息
        new_client_socket,client_addr = tcp_server_socket.accept()

        # while True:
        send_file_2_client(new_client_socket,client_addr)


        # 关闭
        new_client_socket.close()
    # 关闭套接字
    tcp_server_socket.close()

客户端

import socket


def main():
    # 创建
    tcp_client_socket = socket.socket(socket.AF_INET,socket.SOCK_STREAM)

    # 连接服务器
    # ip和port可以写死 也可以用input() 输入
    server_ip = input("请输入服务器IP：")
    server_port = int(input("请输入端口："))
    tcp_client_socket.connect((server_ip,server_port))

    # 发送信息
    file_name = input("请输入要下载的文件名：")
    tcp_client_socket.send(file_name.encode("utf-8"))

    # 接收对方发来的数据
    recv_data = tcp_client_socket.recv(1024)
    print(recv_data)
    if recv_data:
        with open("[接收]" + file_name,"wb") as f:
            f.write(recv_data)

    # 关闭套接字
    tcp_client_socket.close()