python BytesIO 中 read 用法示例详解
在用 Flask 写一个项目，后台管理用的插件暂时是 flask-admin。想实现的效果：在后台管理页面中，把提交到后端的图片不保存在 static 文件夹下面，而是通过后端代码把这个文件对象上传到 AWS 的 S3中存储。
通过flask-admin 上传到后端的文件对象的类型是：
FileStorage # werkzeug.datastructures.FileStorage # flask 中的 request.files 获取到的类型也是 FileStorage
所以先从提交到后端的 form 表单中获取到该文件对象，例如为： img_obj。 现在刚需要把类型为 FileStorage 的 img_obj 转化为 file-like object （AWS S3 boto3 中的 upload_fileobj 接口需要这样的参数）。转化的过程用到了 shutil 的copyfileobj 和 BytesIO， 如下：
from shutil import copyfileobj temp_file = BytesIO() copyfileobj(img_obj.stream, temp_file) # img_obj.stream 应该是能获取到对象中的数据流； 然后把 imb_obj 中的数据流copy到 temp_file 中
然后，问题来了。 利用下面的 S3 upload_fileobj接口把文件上传到 S3后，对应的文件一直都是 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) # 利用这个接口把文件上传到服务器后一直都是0比特
查询资料发现原因。
我们先来看下 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) ，此时写“文件”的游标已经到“文件”的最后
"""我们再来看下面有关 BytesIO 的的一些用法：
In [1]: from io import BytesIO In [2]: f = BytesIO() In [3]: f.write(b'abc') # 把byte 写入到 f 中，此时 游标已经到f的最后位置 Out[3]: 3 In [4]: f.read() # 由于此时游标是从f 的 最后的位置开始 read，那么后面的内容肯定是空 Out[4]: b'' In [5]: f.tell() Out[5]: 3 # 说明游标是在f最后的位置 In [6]: f.seek(0) # 利用 seek(0) 把游标的位置放到f的 0 位置处 Out[6]: 0 In [7]: f.read() # 此时再 read 就能看到全部内容 Out[7]: b'abc' """ getvalue() 是获取全部内容； read() 是从游标的当前位置往后读 """
所以上面问题的原因也是：
copyfileobj 中的 fdst.write(buf) 写完后，此时游标在“文件”最后一个位置；而由于 S3 的 upload_fileobj 接口中的第一个参数是file-like object, 而且upload_fileobj会调用 这个 file-like object 的 read() 方法，read 出来的内容会上传到 S3 上。 所以，解决办法就是利用 seek(0) 把游标位置再次放到 0 处
正确代码如下：
from shutil import copyfileobj temp_file = BytesIO() copyfileobj(img_obj.stream, temp_file) temp_file.seek(0) # 让游标回到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)
到此这篇关于python BytesIO 中 read 用法的文章就介绍到这了,更多相关python BytesIO read 用法内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！
相关文章
这篇文章主要介绍了Python使用百度翻译开发平台实现英文翻译为中文功能,结合实例形式分析了Python使用request请求与百度翻译API接口交互实现翻译功能相关操作技巧,需要的朋友可以参考下2019-08-08
这篇文章主要介绍了Pandas直接读取sql脚本的方法,本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2021-01-01
在DeepSeek等AI工具深度融入工作的今天,我们每天都会生成大量的Markdown格式内容,本教程将教你如何使用 Pandoc配合一段简单的 Python脚本,在本地电脑上免费、安全、完美地完成转换,感兴趣的小伙伴可以了解下2026-03-03
Python调用高德API实现批量地址转经纬度并写入表格的功能
这篇文章主要介绍了Python调用高德API实现批量地址转经纬度并写入表格的功能,本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2021-01-01
这篇文章主要介绍了Python实现返回数组中第i小元素的方法,结合实例形式分析了Python针对数组的遍历、排序、运算等相关操作技巧,需要的朋友可以参考下2017-12-12
本文主要介绍了PyQt5页面跳转问题及解决方式，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2023-01-01
Python pycharm读取文件相对路径与绝对路径的方法
这篇文章主要给大家介绍了关于Python pycharm读取文件相对路径与绝对路径的方法,绝对路径就是文件的真正存在的路径,是指从硬盘的根目录(盘符)开始,进行一级级目录指向文件,相对路径就是以当前文件为基准进行一级级目录指向被引用的资源文件,需要的朋友可以参考下2023-12-12
今天小编就为大家分享一篇pyqt5 实现多窗口跳转的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-06-06
在本篇文章里小编给大家整理了关于pycharm中显示CSS提示的知识点内容，需要的朋友们可以参考学习下。2019-07-07
这篇文章主要介绍了使用Python对文件进行批量改名的方法,本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2023-03-03
最新评论