python简单实现文件下载

1.TCP 文件下载客户端

import socket


def main():
    # 创建客户端套接字
    client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    # 连接服务器ip和端口
    server_ip = input("请输入服务器ip地址")
    server_port = int(input("请输入服务器端口号"))
    client_socket.connect((server_ip, server_port))
    file_name = input("请输入你要下载的文件名")
    # 发送客户端信息
    client_socket.send(file_name.encode())
    # 接收服务器信息
    while True:
        recv_data = client_socket.recv(1024)
        # print(recv_data.decode())
        if recv_data:
            file = open("新建"+file_name, "ab")
            file.write(recv_data)


    client_socket.close()

if __name__ == '__main__':
    main()

2.TCP 文件下载服务器

import socket


def get_file_content(file_name):
    try:
        file = open(file_name, "r")
        content = file.read()
    except:
        print("没有%s这个文件" % file_name)
    else:
        file.close()
        return content


def main():
    # 创建套接字
    server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    # 绑定服务器ip端口
    server_socket.bind(('192.168.113.62', 9995))
    # 将主动套接字转换为被动套接字
    server_socket.listen(128)
    while True:
        # 等待客户端发送信息
        client_socket, client_address = server_socket.accept()
        # 接收客户端发送的文件名
        rec_data = client_socket.recv(1024)
        file_name = rec_data.decode()
        file_content = get_file_content(file_name)
        if file_content:
            client_socket.send(file_content)
            print("文件下载完成")
            client_socket.close()
    server_socket.close()
if __name__ == '__main__':
    main()