python BytesIO 中 read 用法示例詳解
在用 Flask 寫一個項目，后臺管理用的插件暫時是 flask-admin。想實現(xiàn)的效果：在后臺管理頁面中，把提交到后端的圖片不保存在 static 文件夾下面，而是通過后端代碼把這個文件對象上傳到 AWS 的 S3中存儲。
通過flask-admin 上傳到后端的文件對象的類型是：
FileStorage # werkzeug.datastructures.FileStorage # flask 中的 request.files 獲取到的類型也是 FileStorage
所以先從提交到后端的 form 表單中獲取到該文件對象，例如為： img_obj。 現(xiàn)在剛需要把類型為 FileStorage 的 img_obj 轉(zhuǎn)化為 file-like object （AWS S3 boto3 中的 upload_fileobj 接口需要這樣的參數(shù)）。轉(zhuǎn)化的過程用到了 shutil 的copyfileobj 和 BytesIO， 如下：
from shutil import copyfileobj temp_file = BytesIO() copyfileobj(img_obj.stream, temp_file) # img_obj.stream 應(yīng)該是能獲取到對象中的數(shù)據(jù)流； 然后把 imb_obj 中的數(shù)據(jù)流copy到 temp_file 中
然后，問題來了。 利用下面的 S3 upload_fileobj接口把文件上傳到 S3后，對應(yīng)的文件一直都是 0 比特。
代碼如下：
from shutil import copyfileobj temp_file = BytesIO() copyfileobj(img_obj.stream, temp_file) client.upload_fileobj(temp_file, "bucket-name", Key="static/%s" % img_obj.filename) # 利用這個接口把文件上傳到服務(wù)器后一直都是0比特
查詢資料發(fā)現(xiàn)原因。
我們先來看下 shutil.copyfileobj 的源碼：
def copyfileobj(fsrc, fdst, length=16*1024):
"""copy data from file-like object fsrc to file-like object fdst"""
while 1:
buf = fsrc.read(length)
if not buf:
break
fdst.write(buf)
"""
從上述代碼的最后一行看，fdst.write(buf) ，此時寫“文件”的游標(biāo)已經(jīng)到“文件”的最后
"""我們再來看下面有關(guān) BytesIO 的的一些用法：
In [1]: from io import BytesIO In [2]: f = BytesIO() In [3]: f.write(b'abc') # 把byte 寫入到 f 中，此時 游標(biāo)已經(jīng)到f的最后位置 Out[3]: 3 In [4]: f.read() # 由于此時游標(biāo)是從f 的 最后的位置開始 read，那么后面的內(nèi)容肯定是空 Out[4]: b'' In [5]: f.tell() Out[5]: 3 # 說明游標(biāo)是在f最后的位置 In [6]: f.seek(0) # 利用 seek(0) 把游標(biāo)的位置放到f的 0 位置處 Out[6]: 0 In [7]: f.read() # 此時再 read 就能看到全部內(nèi)容 Out[7]: b'abc' """ getvalue() 是獲取全部內(nèi)容； read() 是從游標(biāo)的當(dāng)前位置往后讀 """
所以上面問題的原因也是：
copyfileobj 中的 fdst.write(buf) 寫完后，此時游標(biāo)在“文件”最后一個位置；而由于 S3 的 upload_fileobj 接口中的第一個參數(shù)是file-like object, 而且upload_fileobj會調(diào)用 這個 file-like object 的 read() 方法，read 出來的內(nèi)容會上傳到 S3 上。 所以，解決辦法就是利用 seek(0) 把游標(biāo)位置再次放到 0 處
正確代碼如下：
from shutil import copyfileobj temp_file = BytesIO() copyfileobj(img_obj.stream, temp_file) temp_file.seek(0) # 讓游標(biāo)回到0處 client.upload_fileobj(temp_file, "bucket-name", Key="static/%s" % img_obj.filename)
或者直接把利用 FileStorage 的 stream 屬性把文件上傳到 S3，代碼如下：
client.upload_fileobj(img_obj.stream, "bucket-name", Key="static/%s" % img_obj.filename)
到此這篇關(guān)于python BytesIO 中 read 用法的文章就介紹到這了,更多相關(guān)python BytesIO read 用法內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
Python使用百度翻譯開發(fā)平臺實現(xiàn)英文翻譯為中文功能示例
這篇文章主要介紹了Python使用百度翻譯開發(fā)平臺實現(xiàn)英文翻譯為中文功能,結(jié)合實例形式分析了Python使用request請求與百度翻譯API接口交互實現(xiàn)翻譯功能相關(guān)操作技巧,需要的朋友可以參考下2019-08-08
這篇文章主要介紹了Pandas直接讀取sql腳本的方法,本文給大家介紹的非常詳細(xì)，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下2021-01-01
使用Pandoc與Python實現(xiàn)本地化文檔轉(zhuǎn)換全攻略
在DeepSeek等AI工具深度融入工作的今天,我們每天都會生成大量的Markdown格式內(nèi)容,本教程將教你如何使用?Pandoc配合一段簡單的?Python腳本,在本地電腦上免費、安全、完美地完成轉(zhuǎn)換,感興趣的小伙伴可以了解下2026-03-03
Python調(diào)用高德API實現(xiàn)批量地址轉(zhuǎn)經(jīng)緯度并寫入表格的功能
這篇文章主要介紹了Python調(diào)用高德API實現(xiàn)批量地址轉(zhuǎn)經(jīng)緯度并寫入表格的功能,本文給大家介紹的非常詳細(xì)，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下2021-01-01
Python實現(xiàn)返回數(shù)組中第i小元素的方法示例
這篇文章主要介紹了Python實現(xiàn)返回數(shù)組中第i小元素的方法,結(jié)合實例形式分析了Python針對數(shù)組的遍歷、排序、運(yùn)算等相關(guān)操作技巧,需要的朋友可以參考下2017-12-12
本文主要介紹了PyQt5頁面跳轉(zhuǎn)問題及解決方式，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2023-01-01
Python?pycharm讀取文件相對路徑與絕對路徑的方法
這篇文章主要給大家介紹了關(guān)于Python?pycharm讀取文件相對路徑與絕對路徑的方法,絕對路徑就是文件的真正存在的路徑,是指從硬盤的根目錄(盤符)開始,進(jìn)行一級級目錄指向文件,相對路徑就是以當(dāng)前文件為基準(zhǔn)進(jìn)行一級級目錄指向被引用的資源文件,需要的朋友可以參考下2023-12-12
今天小編就為大家分享一篇pyqt5 實現(xiàn)多窗口跳轉(zhuǎn)的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-06-06
在本篇文章里小編給大家整理了關(guān)于pycharm中顯示CSS提示的知識點內(nèi)容，需要的朋友們可以參考學(xué)習(xí)下。2019-07-07
這篇文章主要介紹了使用Python對文件進(jìn)行批量改名的方法,本文給大家介紹的非常詳細(xì)，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下2023-03-03