python:tcp模拟文件下载

#coding=utf-8
import socket

#客户端发送请求
def downLoad():
    #创建套接字
    tcp_scoket=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
    tcp_scoket.bind(("", 8001))
    # 获取服务器IP 和端口,连接服务器
    tcp_scoket.connect(("ip",8000))
    # 获取下载的文件名 将文件名发送到服务器
    tcp_scoket.send("111.txt".encode("utf-8"))
    # 接收数据
    recv_data=tcp_scoket.recv(1024)
    # 接收数据存到文件中
    if recv_data:
        with open("[新]111.txt","wb") as f:
            f.write(recv_data)
    # 关闭套接字
    tcp_scoket.close()

#获取文件的函数
def sendFile2Client(client_scoket,client_addr):
    # 接收下载的文件名
    recv_data = client_scoket.recv(1024).decode("utf-8")
    print("客户端%s要下载文件:%s" % (str(client_addr), recv_data))
    # 打开文件读取数据
    try:
        f=open(recv_data,"rb")
        content=f.read()
        f.close()
    except Exception as ret:
        print("没有要下载的文件%s" % recv_data)
    # 数据发给客户端
    if content:
        client_scoket.send(content)

#服务器发送文件
def sendFile():
    # 创建套接字
    tcp_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    # 绑定端口，接收数据时会从绑定的端口发送，不会再生成随机端口
    tcp_socket.bind(("", 8000))
    # listen使套接字变成可以被动链接
    tcp_socket.listen(128)
    # 为多个客户端服务while True:
    # accept等待客户端的链接,返回一个客户端元组
    client_scoket, client_addr = tcp_socket.accept()
    print(client_addr)
    #调用函数
    sendFile2Client(client_scoket,client_addr)
    # 关闭套接字
    client_scoket.close()
    tcp_socket.close()

if __name__ == '__main__':
    #downLoad()
    sendFile()