利用Python的socket编程简易实现TCP下载器

2018-02-04 21:04:02 Python 阅读 (2617) 评论(0)

      最近在学习Python的socket编程,简易实现了一个tcp下载器的功能。宇宙惯例,先上代码

一. 客户端代码(client.py)

import socket


def main():
    # 创建套接字
    tcp_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    # 链接服务器
    tcp_socket.connect(("192.168.21.59", 8090))
    # 获取下载的文件名字
    file_name = input("请输入你想下载的名字:")
    # 将文件名字发送到服务器
    tcp_socket.send(file_name.encode("utf-8"))
    # 接收服务器发过来的数据
    recv_data = tcp_socket.recv(1024)
    if recv_data:
        with open("download_" + file_name, "wb") as f:
            f.write(recv_data)
        print("文件下载完成")
    else:
        print("服务器找不到该文件(%s)" % file_name)
    # 关闭套接字
    tcp_socket.close()


if __name__ == '__main__':
    main()

二. 服务器端代码(server.py)

import socket


def send_file(client_socket, client_addr):
    """
    服务器端发送文件给客户端函数
    :param client_socket: 服务器端与客户端建立链接的socket
    :param client_addr: 客户端地址
    """
    # 接收客户端需要下载的文件名
    file_name = client_socket.recv(1024).decode("utf-8")
    print("客户端(%s)需要下载文件是:%s" % (str(client_addr), file_name))
    file_content = None
    # 打开文件,读取数据
    try:
        f = open(file_name, "rb")
        file_content = f.read()
        f.close()
    except Exception as ret:
        print("服务器找不到该文件(%s)" % file_name)

    # 发送文件给客户端
    if file_content:
        client_socket.send(file_content)


def main():
    # 创建socket
    tcp_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    # 绑定服务器端本地信息(ip,端口)
    tcp_socket.bind(("", 8090))
    # 设置套接字为被动监听模式
    tcp_socket.listen(128)
    while True:
        # 等待客户端的链接
        new_client_socket, client_addr = tcp_socket.accept()
        # 调用发送文件函数
        send_file(new_client_socket, client_addr)
        # 关闭当前客户端套接字
        new_client_socket.close()

    # 如无客户端请求,关闭服务器端套接字
    tcp_socket.close()


if __name__ == '__main__':
    main()

三. TCP介绍

   TCP是一种面向连接的、可靠的、基于字节流的传输层通信协议。这种偏理解概念性的东西,我怕我的只言片语会给读者带来误区,我就不赘述了,但是给大家几个传送门去理解这些协议:


在各编程语言里的socket都聚合实现了tcp,udp等协议。

四.程序说明

首先,运行服务器端程序,再运行客户端程序与服务端建立链接。客户端程序发起一个(下载)请求,服务器端会尝试open这个文件名,如果抛出异常,说明服务器端没有客户端尝试下载的文件。如果服务器端存在该文件,服务器端的程序会读写该文件,并发送给客户端,客户端接收到后,写到一个本地的新文件里。

这就是程序的运行流程说明(注释足够详细,相信你应该能看懂)



评论