Python—实现文件下载器

补充知识点

1. tcp服务器一般情况下都需要绑定，否则客户端找不到这个服务器
2. tcp客户端一般不绑定，因为是出动链接服务器，所以只要确定好服务器的ip、port等信息就好，本地客户端可以随机
3. tcp服务器中通过listen可以将socket创建出来的主动套接字变为被动的，这是tcp服务器时必须要做的
4. 当客户端需要链接服务器时，就需要使用connect进行链接，udp是不需要链接的而是直接发送的，但是tcp必须先链接，只有链接成功才可以通信
5. 当一个tcp客户端连接服务器时，服务器端会有一个新的套接字，这个套接字是用来标记这个客户端，单独为这个客户端服务
6. listen后的套接字是被动套接字，用来接收新的客户端的链接请求，而accept返回的新套接字是标记这个客户端的
7. 关闭listen后的套接字意味这被动套接字关闭，会导致新的客户端不能够链接服务器，但是之前已经链接成功的客户端正常通信
8. 关闭accept返回的套接字意味着这个客户端已经服务完毕
9. 当客户端的套接字调用close后，服务器会recv解阻塞，并其返回长度为0，因此服务器可以通过返回数据的长度来区别客户端是否已经下线

文件下载器客户端

#!/usr/bin/env python
# coding=utf-8
import socket


def main():
    #1. 创建套接字
    tcp_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    
    #2. 获取服务器ipport
    dest_ip = input("please input server ip:")
    dest_port = int(input("please input server port"))

    #3. 链接服务器
    tcp_socket.connect((dest_ip, dest_port))

    #4. 获取下载文件的名字
    download_file_name = input("please input filename:")

    #5. 将文件名发送到服务器
    tcp_socket.send(download_file_name.encode("utf-8"))

    #6. 接收文件中的数据
    recv_data = tcp_socket.recv(1024) # 1024---->1k   1024*1024---->1k*1024=1M
    
    if recv_data:
        #7. 保存接收到的数据到一个文件中
        with open("[新]" + download_file_name, "wb") as f:
            f.write(recv_data)

    #8. 关闭套接字
    tcp_socket.close()

if __name__ == "__main__":
    main()

文件下载器服务器

#!/usr/bin/env python
# coding=utf-8
import socket

def send_file_2_client(new_client_socket, client_addr):
    #1. 接收客户端下载的文件名
    # 接收客户端发送的要下载的文件名
    file_name = new_client_socket.recv(1024).decode("utf-8")
    print("客户端（%s）需要下载的文件是：%s" % (str(client_addr), file_name))
    
    file_content = None
    #2. 打开文件，读取数据
    try:
        f = open(file_name, "rb")
        file_content = f.read()
        f.close()
    except Exception as ret:
        print("没有要下载的文件（%s）" % file_name)

    #3. 发送文件数据给客户端
    if file_content:
        #new_client_socket.send("hahhhahahah".encode("utf-8"))
        new_client_socket.send(file_content)

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

    #2. 绑定本地信息
    tcp_server_socket.bind(("", 7788))

    #3. 让默认的套接字由主动变为被动
    tcp_server_socket.listen(128)

    while True:
        #4. 等待客户端的链接
        new_client_socket, client_addr = tcp_server_socket.accept()
        
        #5 发送文件到客户端
        send_file_2_client(new_client_socket, client_addr)

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



if __name__ == "__main__":
    main()