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

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

服务器 参考代码如下:

from socket import *
import sys
def 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程序设计有所帮助。

来源：https://blog.csdn.net/kai46385076/article/details/100006148