python socket 简约下载器

一、下载文件 客户端

#下载文件，客户端
import socket
def main():
    #1、创建套接字
    tcp_socket=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
    #2、获取服务器的ip port
    dest_ip=input("请输入下载服务器的ip：")
    dest_port=int(input("请输入下载服务器的port："))
    #3、链接服务器
    tcp_socket.connect((dest_ip,dest_port))
    #4、获取下载的文件名
    download_file_name=input("请输入要下载的文件名字：")
    #5、将文件名字发送到服务器
    tcp_socket.send(download_file_name.encode("utf-8"))
    #6、接受收文件中的数据
    recv_data=tcp_socket.recv(1024*1024)#下载个小一点的文件，1M大小（1024等价于1K）
    #7、保存接收到的数据到一个文件中
    with open("[新]"+download_file_name,"wb") as f:#以wb的方式打开写二进制，注意recv的数据是二进制故保持一致
        f.write(recv_data)
    #8、关闭套接字
    tcp_socket.close()
if __name__=="__main__":
    main()

二、下载文件 服务器

#下载文件 服务器
import socket
def main():
    #创建socket作用是监听
    tcp_server_socket=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
    #本地信息bind
    address=('',7788)
    #绑定ip和port
    tcp_server_socket.bind(address)
    #使用socket创建的套接字默认的属性是主动的，使用listen将其变为被动的，这样就可以接收别人的链接了
    tcp_server_socket.listen(128)#设置为正常可发接信息状态
    #如果有新的客户端来链接服务器，那么就产生一个新的套接字专门为这个客户端服务
    client_socket,clientAddr=tcp_server_socket.accept()#等待行动相当于停在这里阻塞状态，只要有一个客户端链接到则立马解阻塞
    #接收对方发送来的数据
    file_name=client_socket.recv(1024).decode("utf-8")
    print('客户端(%s)需要下载的文件是：%s'%(str(clientAddr),file_name))
    #发送一些数据到客户端
    client_socket.send("thank you!".encode('utf-8'))
    #关闭套接字
    client_socket.close()
    tcp_server_socket.close()
if __name__=="__main__":
    main()

三、结果