python實現(xiàn)ftp文件傳輸功能

更新時間：2020年03月20日 09:12:17 作者：zjtMeng

這篇文章主要為大家詳細介紹了python實現(xiàn)ftp文件傳輸功能，文中示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下

本文實例為大家分享了python實現(xiàn)ftp文件傳輸?shù)木唧w代碼，供大家參考，具體內(nèi)容如下

主要步驟可以分為以下幾步：

1.讀取文件名
2.檢測文件是否存在
3.打開文件
4.檢測文件大小
5.發(fā)送文件大小和 md5值給客戶端
6.等客戶端確認
7.開始邊讀邊發(fā)數(shù)據(jù)

服務(wù)器端代碼：

import socket,os,time
import hashlib
 
server =socket.socket()
server.bind(('0.0.0.0',6666))
server.listen()
print("等待....")
while True:
 conn,addr = server.accept()
 print("new conn:",conn)
 while True:
 data = conn.recv(1024)
 if not data:
 print("client is disconnection")
 break
 cmd,filename = data.decode().split() #記錄指令和文件名
 print(filename)
 #判斷當(dāng)前目錄是否存在該文件，而且必須是文件，而不是目錄
 if os.path.isfile(filename):
 f = open(filename,'rb')
 #m = hashlib.md5() # 創(chuàng)建md5
 file_size = os.stat(filename).st_size #stat() 可以返回文件的大小值
 conn.send((str(file_size)).encode()) # 發(fā)送文件大小
 conn.recv(1024) #等待返回信息
 for line in f:
 # m.updata(line) 
 conn.send(line)
 #print("file md5",m.hexdigest()) #打印md5值
 f.close()

客戶端代碼：

# Author: zjt
import socket
 
client = socket.socket()
 
client.connect(("0.0.0.0",6666))
 
 
while True:
 cmd = input(">>>:").strip()
 
 if len(cmd)==0 :continue
 
 if cmd.startswith("get"):
 client.send(cmd.encode())
 server_response = client.recv(1024)
 print("server response: ",server_response)
 client.send(b"ready to recv file")
 
 # 開始接收文件
 file_total_size = int(server_response.decode())
 received_size = 0 # 記錄接收文件的大小
 filename = cmd.split()[1]
 # 因為兩個目錄一致，接收的文件名不能與原文件相同
 f = open(filename+".new","wb")
 while received_size < file_total_size:
 data = client.recv(1024)
 received_size += len(data)
 f.write(data)
 print("total：",file_total_size," present: ",received_size)
 else:
 print("file has received done!")
 f.close()
 
client.close()

用80M的文件傳輸測試，效果如下：

程序升級：

前面的代碼還沒添加md5進行驗證，現(xiàn)在對代碼進行升級

服務(wù)器端代碼：

import socket,os,time
import hashlib
 
server =socket.socket()
server.bind(('0.0.0.0',8888))
server.listen()
print("等待....")
while True:
 conn,addr = server.accept()
 print("new conn:",conn)
 while True:
 data = conn.recv(1024)
 if not data:
 print("client is disconnection")
 break
 cmd,filename = data.decode().split() #記錄指令和文件名
 print(filename)
 #判斷當(dāng)前目錄是否存在該文件，而且必須是文件，而不是目錄
 if os.path.isfile(filename):
 f = open(filename,'rb')
 m = hashlib.md5() # 創(chuàng)建md5
 file_size = os.stat(filename).st_size #stat() 可以返回文件的大小值
 conn.send((str(file_size)).encode()) # 發(fā)送文件大小
 conn.recv(1024) #等待返回信息
 for line in f:
 m.update(line)
 conn.send(line)
 print("file md5",m.hexdigest()) #打印md5值
 
 f.close()
 conn.send(m.hexdigest().encode()) # 發(fā)送md5
 
 print("我真的已經(jīng)發(fā)過去了",m.hexdigest().encode())
 
 print("send done")
 
server.close()

客戶端代碼：

import socket
import hashlib
client = socket.socket()
client.connect(("0.0.0.0",8888))
while True:
 cmd = input(">>>:").strip()
 if len(cmd)==0 :continue
 if cmd.startswith("get"):
 client.send(cmd.encode())
 server_response = client.recv(1024)
 print("server response: ",server_response)
 client.send(b"ready to recv file")
 # 開始接收文件
 file_total_size = int(server_response.decode())
 received_size = 0 # 記錄接收文件的大小
 filename = cmd.split()[1]
 # 因為兩個目錄一致，接收的文件名不能與原文件相同
 f = open(filename+".new","wb")
 m = hashlib.md5()
 while received_size < file_total_size:
 data = client.recv(1024)
 received_size += len(data)
 m.update(data)
 f.write(data)
 #print("total：",file_total_size," present: ",received_size)
 else:
 new_file_md5 = m.hexdigest()
 print("client file md5:",new_file_md5)
 print("file has received done!")
 print("total：",file_total_size," present: ",received_size)
 f.close()
 sever_file_md5 = client.recv(1024)
 print("client file md5:",new_file_md5)
 print("server file md5:",sever_file_md5)
client.close()

兩個程序在linux 環(huán)境下運行，結(jié)果如下：

可以看到傳輸后文件大小變大了一點點，而且md5前后值也不同，說明文件傳輸發(fā)生了改變。

現(xiàn)在講程序在windows環(huán)境下運行，結(jié)果如下：

此時可以看到windows上沒有問題，文件大小相同，且md5值也一致。

原因分析：

之所以會發(fā)生這種情況，是因為在linux上運行時，最后一次傳輸文件與發(fā)送md5值的時候，發(fā)生可粘包，導(dǎo)致最后一次接收文件的時候，連同md5的數(shù)據(jù)一并發(fā)送了。而客戶端也當(dāng)作一條接收信息，全部接收了。所以客戶端出現(xiàn)沒有收到來自服務(wù)器端的md5值，多出來的那一點點，就是md5值的大小。

解決方法：

在接收文件的時候，判斷當(dāng)前剩余多少文件需要接收，如果大于1024，就接收1024大小的文件，否則就只接收剩下全部的文件，防止最后一次接收多余的數(shù)據(jù)。

只需要對客戶端代碼進行修改，修改后代碼如下：

import socket
import hashlib
client = socket.socket()
client.connect(("0.0.0.0",8888))
while True:
 cmd = input(">>>:").strip()
 if len(cmd)==0 :continue
 if cmd.startswith("get"):
 client.send(cmd.encode())
 server_response = client.recv(1024)
 print("server response: ",server_response)
 client.send(b"ready to recv file")
 # 開始接收文件
 file_total_size = int(server_response.decode())
 received_size = 0 # 記錄接收文件的大小
 filename = cmd.split()[1]
 f = open(filename+".new","wb")
 m = hashlib.md5()
 while received_size < file_total_size:
 #添加一次判斷，使最后一次剩多少就接收多少，避免發(fā)生粘包
 if file_total_size - received_size > 1024:
  size = 1024
 else: # 最后一次，剩多少收多少
  size = file_total_size - received_size
 data = client.recv(size)
 received_size += len(data)
 m.update(data)
 f.write(data)
 else:
 new_file_md5 = m.hexdigest() 
 print("client file md5:",new_file_md5)
 print("file has received done!")
 print("total：",file_total_size," present: ",received_size)
 print("下一句關(guān)閉文件")
 f.close()
 print("開始接收md5 ") 
 sever_file_md5 = client.recv(1024)
 print("client file md5:",new_file_md5)
 print("server file md5:",sever_file_md5)
client.close()

以上就是本文的全部內(nèi)容，希望對大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。

您可能感興趣的文章: