解決python web cgi文件下載功能的bug

最近我在用python cgi做web開發，寫文件下載功能時，參考了菜鳥教程的代碼。

代碼做了一些修改，代碼如下：

<code>#!C:/Users/Administrator/AppData/Local/Programs/Python/Python37/python.exe
#第一行是python解釋器的路徑，絕對不能省，!也不能省略，路徑根據自己安裝的修改
import os

file_path='1.txt'    #把文件路徑'1.txt'賦值給file_path
file_name=os.path.basename(file_path)    #把文件名賦值給file_name
# HTTP 頭部開始
print ('Content-Disposition: attachment; filename=%s'%(file_name))  
#Content-Disposition: attachment是彈出文件下載的對話框
#filename=%s等號後面是文件下載對話框顯示的文件名，%s是格式化輸出，%(file_name)用括號中的變量傳參給%s
print ()    #輸出1行空行，告訴瀏覽器HTTP頭部結束
# HTTP 頭部結束

fd = open(file_path, "rb")    #以讀取二進制的方式打開一個文件，返回的文件描述符，賦值給fd
read_bytes = fd.read()    #通過文件描述符fd讀取打開文件的數據，並賦值給read_bytes
print (read_bytes)    #輸出讀取到的數據
fd.close()    #關閉文件描述符fd
/<code>

假如1.txt文件裡的內容是：

<code>人生苦短，我用python/<code>

那麼下載以後就是這樣子的：

<code>b'\\\\xe4\\\\xba\\\\xba\\\\xe7\\\\x94\\\\x9f\\\\xe8\\\\x8b\\\\xa6\\\\xe7\\\\x9f\\\\xad\\\\xef\\\\xbc\\\\x8c\\\\xe6\\\\x88\\\\x91\\\\xe7\\\\x94\\\\xa8python'/<code>

問：為什麼會這個樣子？

答：原因是因為print()把字節流轉成字符串輸出了，而且下載的文件跟源文件的大小也是不一樣的，源文件大小是：27Byte（27字節），下載的文件大小是：95Byte（95字節）。

描述：如果是下載二進制文件，比如圖片、視頻、音樂，下載之後，文件會變大好幾倍。

問：為什麼會這樣子呢？

答：就是因為print()的關係，所以不管源文件是什麼文件，下載之後都會變成文本文件。

問：那怎麼解決這個bug呢？

答：答案就是不用print()進行輸出

答：可以使用sys模塊的stdout，stdout是標準輸出流，只支持字節流輸出，輸出的內容就是這種的：b'內容'，print()也是對標準輸出流進行了封裝。

廢話不多說，看代碼：

<code>#!C:/Users/Administrator/AppData/Local/Programs/Python/Python37/python.exe
#第一行是python解釋器的路徑，絕對不能省，!也不能省略，路徑根據自己安裝的修改
import os,sys

file_path='1.txt'    #把文件路徑'1.txt'賦值給file_path
file_name=os.path.basename(file_path)    #獲取文件名，並賦值給file_name
file_size=os.path.getsize(file_path)    #獲取文件大小，並賦值給file_size
# HTTP 頭部開始
print ('Content-Disposition: attachment; filename=%s'%(file_name))  
#Content-Disposition: attachment是彈出文件下載的對話框
#filename=%s等號後面是文件下載對話框顯示的文件名，%s是格式化輸出，%(file_name)用括號中的變量傳參給%s
print('Accept-Length:%s'%(file_size))    #這個是文件下載框顯示的文件大小
print()    #輸出1行空行，告訴瀏覽器HTTP頭部結束
# HTTP 頭部結束

fd = open(file_path, "rb")    #以讀取二進制的方式打開一個文件，返回的文件描述符，賦值給fd
read_bytes = fd.read()    #通過文件描述符fd讀取打開文件的數據，並賦值給read_bytes 

sys.stdout.flush()    #刷新標準輸出流的緩衝區，如果不刷新的話，客戶端就接收不到數據
sys.stdout.buffer.write (read_bytes)    #把讀取到的數據寫入到標準輸出流的緩衝區
fd.close()    #關閉文件描述符fd
/<code>

閱讀更多 張榮殿 的文章

關鍵字: 瀏覽器 Python 下載