python影片教學欄位介紹實作檔案傳輸的高效方法。

透過網路通訊Socket模組實現檔案傳輸

實現的效果如下的Gif所示，就是網路通訊Socket模組實作檔案下載傳輸。

服務端

首先需要取得本機ip，這裡服務端採用多執行緒的方法，就是定義一個函數，然後用threading建立任務。客戶端連線成功，接收客戶端的請求訊息，就是下載的檔案名稱。所以需要判斷，有輸出檔案位元組數。然後在問使用者是不是要下載，得到資訊就使用 while True: 讀檔案的內容，再一個send。看程式碼是不是就是這麼回事。

import socketimport osimport threading# 获取本机ipdef get_host_ip(): try: s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) s.connect(('8.8.8.8', 80)) ip = s.getsockname()[0] finally: s.close() return ip# 处理客户端请求下载文件的操作（从主线程提出来的代码）def deal_client_request(ip_port, service_client_socket): # 连接成功后，输出“客户端连接成功”和客户端的ip和端口 print("客户端连接成功", ip_port) # 接收客户端的请求信息【recv】 file_name = service_client_socket.recv(1024) # 解码 file_name_data = file_name.decode("utf-8") # 判断文件是否存在 if os.path.exists(file_name_data): #输出文件字节数 fsize = os.path.getsize(file_name_data) #转化为兆单位 fmb = fsize/float(1024*1024) #要传输的文件信息 senddata = "文件名：%s 文件大小：%.2fMB"%(file_name_data,fmb) #发送和打印文件信息【send】 service_client_socket.send(senddata.encode("utf-8")) print("请求文件名：%s 文件大小：%.2f MB"%(file_name_data,fmb)) #接受客户是否需要下载【recv】 options = service_client_socket.recv(1024) if options.decode("utf-8") == "y": # 打开文件 with open(file_name_data, "rb") as f: #　计算总数据包数目 nums = fsize/1024 #　当前传输的数据包数目 cnum = 0 while True: file_data = f.read(1024) cnum = cnum + 1 #progress = cnum/nums*100 #print("当前已下载：%.2f%%"%progress,end = "\r") if file_data: # 只要读取到数据，就向客户端进行发送【send】 service_client_socket.send(file_data) # 数据读完，退出循环 else: print("请求的文件数据发送完成") break else: print("下载取消！") else: print("下载的文件不存在！") # 关闭服务当前客户端的套接字【close】 service_client_socket.close()if __name__ == '__main__': # 获取本机ip print("TCP文件传输服务器，本机IP:" + get_host_ip()) # 把工作目录切换到data目录下 os.chdir("./data") # 创建套接字【socket】 tcp_server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM) # 绑定端口号【bind】 tcp_server_socket.bind(("", 3356)) # 设置监听，将主动套接字变为被动套接字【listen】 tcp_server_socket.listen(128) # 循环调用【accept】，可以支持多个客户端同时连接，和多个客户端同时下载文件 while True: service_client_socket, ip_port = tcp_server_socket.accept() # 连接成功后打印套接字号 #print(id(service_client_socket)) # 创建子线程 sub_thread = threading.Thread(target=deal_client_request, args=(ip_port, service_client_socket)) # 启动子线程 sub_thread.start()复制代码

登入後複製

客戶端

客戶端更簡單，連接服務端，發送下載檔案的請求，定義一個寫入的資料夾，就是小兒科東西。不寫了，看程式碼。

# -*- coding:utf-8 -*-# 多任务文件下载器客户端import socketimport osif __name__ == '__main__': # 创建套接字【socket】 tcp_client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM) # 和服务端连接【connect】 server_ip = input("输入服务器IP：") tcp_client_socket.connect((server_ip, 3356)) # 发送下载文件的请求 file_name = input("请输入要下载的文件名：") # 编码 file_name_data = file_name.encode("utf-8") # 发送文件下载请求数据【send】 tcp_client_socket.send(file_name_data) # 接收要下载的文件信息【recv】 file_info = tcp_client_socket.recv(1024) # 文件信息解码 info_decode = file_info.decode("utf-8") print(info_decode) #获取文件大小 fileszie = float(info_decode.split('：')[2].split('MB')[0]) fileszie2 = fileszie*1024 # 是否下载？输入ｙ　确认　输入ｑ 取消 opts = input("是否下载？(y 确认　q 取消)") if opts == 'q': print("下载取消！程序退出") else: print("正在下载　>>>>>>") #向服务器确认正在下载【send】 tcp_client_socket.send(b'y') recvpath = "./receive/" if not os.path.exists(recvpath): os.mkdir(recvpath) # 把数据写入到文件里 with open(recvpath + file_name, "wb") as file: #目前接收到的数据包数目 cnum = 0 while True: # 循环接收文件数据【recv】 file_data = tcp_client_socket.recv(1024) # 接收到数据 if file_data: # 写入数据 file.write(file_data) cnum = cnum+1 #progress =cnum/fileszie2*100 #print("当前已下载：%.2f%%"%progress,end = "\r") # 接收完成 else: print("下载结束！") break # 关闭套接字【close】 tcp_client_socket.close()复制代码

登入後複製