python基于TCP实现的文件下载器功能案例

Python  /  管理员 发布于 7年前   159

本文实例讲述了python基于TCP实现的文件下载器功能。分享给大家供大家参考，具体如下：

服务器 参考代码如下:

from socket import *import sysdef get_file_content(file_name):  """获取文件的内容"""  try:    with open(file_name, "rb") as f:      content = f.read()    return content  except:    print("没有下载的文件:%s" % file_name)def main():  if len(sys.argv) != 2:    print("请按照如下方式运行：python3 xxx.py 7890")    return  else:    # 运行方式为python3 xxx.py 7890    port = int(sys.argv[1])  # 创建socket  tcp_server_socket = socket(AF_INET, SOCK_STREAM)  # 本地信息  address = ('', port)  # 绑定本地信息  tcp_server_socket.bind(address)  # 将主动套接字变为被动套接字  tcp_server_socket.listen(128)  while True:    # 等待客户端的链接，即为这个客户端发送文件    client_socket, clientAddr = tcp_server_socket.accept()    # 接收对方发送过来的数据    recv_data = client_socket.recv(1024) # 接收1024个字节    file_name = recv_data.decode("utf-8")    print("对方请求下载的文件名为:%s" % file_name)    file_content = get_file_content(file_name)    # 发送文件的数据给客户端    # 因为获取打开文件时是以rb方式打开，所以file_content中的数据已经是二进制的格式，因此不需要encode编码    if file_content:      client_socket.send(file_content)    # 关闭这个套接字    client_socket.close()  # 关闭监听套接字  tcp_server_socket.close()if __name__ == "__main__":  main()

客户端 参考代码如下:

from socket import *def main():  # 创建socket  tcp_client_socket = socket(AF_INET, SOCK_STREAM)  # 目的信息  server_ip = input("请输入服务器ip:")  server_port = int(input("请输入服务器port:"))  # 链接服务器  tcp_client_socket.connect((server_ip, server_port))  # 输入需要下载的文件名  file_name = input("请输入要下载的文件名：")  # 发送文件下载请求  tcp_client_socket.send(file_name.encode("utf-8"))  # 接收对方发送过来的数据，最大接收1024个字节（1K）  recv_data = tcp_client_socket.recv(1024)  # print('接收到的数据为:', recv_data.decode('utf-8'))  # 如果接收到数据再创建文件，否则不创建  if recv_data:    with open("[接收]"+file_name, "wb") as f:      f.write(recv_data)  # 关闭套接字  tcp_client_socket.close()if __name__ == "__main__":  main()

更多关于Python相关内容可查看本站专题：《Python Socket编程技巧总结》、《Python数据结构与算法教程》、《Python函数使用技巧总结》、《Python字符串操作技巧汇总》、《Python入门与进阶经典教程》及《Python文件与目录操作技巧汇总》

希望本文所述对大家Python程序设计有所帮助。