Django实现文件上传及下载的方式,FileSystemStorage和StreamingHttpResponse的使用

文件保存的路径配置

python 复制代码
import platform

system = platform.system()  # 获取当前系统类型
if system == "Windows":  # 测试环境
    MEDIA_ROOT = "F:/Download/media/"
    DOMAIN = "http://127.0.0.1/"
elif system == "Linux":  # 线上环境
    MEDIA_ROOT = "/mnt/media/"
    DOMAIN = "http://xxxx.com/"
else:
    MEDIA_ROOT = os.path.join(BASE_DIR, 'media/')
    DOMAIN = "http://127.0.0.1/"

# print(MEDIA_ROOT)

MEDIA_URL = '/media/'

一、文件上传

1,使用自带的FileSystemStorage类

简单功能实现

python 复制代码
from django.views import View
from django.core.files.storage import FileSystemStorage


class SingleMediaView(View):
    def post(self, request):
        file = request.FILES.get('file', None)
        # 保存上传的文件到指定的绝对目录
        fs = FileSystemStorage()
        real_path = settings.MEDIA_ROOT + format_filename
        fs.save(real_path, file)
        return JsonResponse({"status": 200, "msg": "上传成功"})
2,使用open函数
python 复制代码
from django.views import View


class SingleMediaView(View):
    def post(self, request):
        file = request.FILES.get('file', 'None')
       
        with open('path/file.txt', 'wb+') as destination:
            for chunk in file.chunks():
                destination.write(chunk)
        return JsonResponse({"status": 200, "msg": "上传成功"})

open函数的第一方法写具体保存文件的路径

二、下载文件

使用StreamingHttpResponse类

python 复制代码
class DownloadFileView(View):
    """文件下载"""
    def get(self, request):
        file_id = request.GET.get("file_id", None)  # 文件id

        if file_id:
            media = Media.objects.get(id=file_id)  # 查询数据库
            real_path = settings.MEDIA_ROOT + media.url  # 获取文件路径

            # 使用StreamingHttpResponse进行大文件下载,不占内存
            def file_iterator(file_path, chunk_size=8192):
                with open(file_path, 'rb') as file:
                    while True:
                        data = file.read(chunk_size)
                        if not data:
                            break
                        yield data

            file_format = media.format
            response = StreamingHttpResponse(file_iterator(real_path))
            if media.type == 1:  # 图片下载
                response['Content-Type'] = f'image/{file_format}'
            elif media.type == 2:  # 视频下载
                response['Content-Type'] = f'video/{file_format}'
            elif media.type == 3:  # 其他类型文件下载
                response['Content-Type'] = 'application/octet-stream'

            # 使用quote对下载的中文文件名进行url编码,否则在Content-Disposition里的filename无法正确识别
            encoded_string = quote(f'{media.title}.{file_format.lower()}')
            response['Content-Disposition'] = f'attachment; filename="{encoded_string}"'

            return response

为DownloadFileView类视图配置url后请求时就可以直接下载了

相关推荐
秋难降39 分钟前
优雅的代码是什么样的?🫣
java·python·代码规范
现在就干41 分钟前
Spring事务基础:你在入门时踩过的所有坑
java·后端
该用户已不存在1 小时前
Gradle vs. Maven,Java 构建工具该用哪个?
java·后端·maven
JohnYan1 小时前
Bun技术评估 - 23 Glob
javascript·后端·bun
二闹1 小时前
聊天怕被老板发现?摩斯密码来帮你
后端·python
用户298698530141 小时前
# C#:删除 Word 中的页眉或页脚
后端
mit6.8241 小时前
[RestGPT] OpenAPI规范(OAS)
人工智能·python
David爱编程1 小时前
happens-before 规则详解:JMM 中的有序性保障
java·后端
小张学习之旅1 小时前
ConcurrentHashMap
java·后端
360安全应急响应中心2 小时前
Python代码保护之重置操作码映射的攻与防探究(一)
python·逆向