优秀的编程知识分享平台

网站首页 > 技术文章 正文

解决python web cgi文件下载功能的bug

nanyue 2024-10-01 13:07:44 技术文章 6 ℃


最近我在用python cgi做web开发,写文件下载功能时,参考了菜鸟教程的代码。

代码做了一些修改,代码如下:

#!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

假如1.txt文件里的内容是:

人生苦短,我用python

那么下载以后就是这样子的:

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

问:为什么会这个样子?

答:原因是因为print()把字节流转成字符串输出了,而且下载的文件跟源文件的大小也是不一样的,源文件大小是:27Byte(27字节),下载的文件大小是:95Byte(95字节)。

描述:如果是下载二进制文件,比如图片、视频、音乐,下载之后,文件会变大好几倍。

问:为什么会这样子呢?

答:就是因为print()的关系,所以不管源文件是什么文件,下载之后都会变成文本文件。

问:那怎么解决这个bug呢?

答:答案就是不用print()进行输出

问:有的人脑海中就会有新的问题飘出来了,不用print()怎么输出?

答:可以使用sys模块的stdout,stdout是标准输出流,只支持字节流输出,输出的内容就是这种的:b'内容',print()也是对标准输出流进行了封装。

废话不多说,看代码:

#!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
最近发表
标签列表