python django 下载文件怎么自定义文件名

原创

mob64ca12d2dee8 2024-07-31 08:57:07 ©著作权

文章标签 自定义 django 文件名 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者mob64ca12d2dee8的原创作品，请联系作者获取转载授权，否则将追究法律责任

Django项目方案：自定义文件下载名称

项目背景

在很多Django应用中，我们经常需要为用户提供文件下载的功能。默认情况下，浏览器下载文件时会使用服务器上文件的原始名称。然而，在某些特定场景下，可能希望用户下载的文件名与实际存储在服务器上的文件名不同。比如，为了给用户提供更好的体验，可能希望根据用户信息、时间戳或其他动态数据生成自定义的文件名。

本方案将详细描述如何在Django中实现自定义文件下载名的功能，并提供相关的代码示例。

系统设计

以下是本项目的系统设计概要，包括文件下载的基本流程与数据结构设计。

数据库设计

erDiagram
    User {
        int id
        string username
        string email
    }
    Document {
        int id
        string file_path
        string original_name
        string user_id
    }
    User ||--o{ Document : upload

下载流程设计

用户通过前端页面上传文件。
文件被保存在服务器上，元数据记录到数据库中。
用户请求下载文件。
服务器生成自定义文件名，并将文件提供给用户下载。

journey
    title 用户下载文件流程
    section 上传文件
      用户上传文件: 5: 用户
      保存文件和元数据: 5: 服务器
    section 下载文件
      用户发起下载请求: 5: 用户
      生成自定义文件名: 5: 服务器
      提供文件下载: 5: 服务器

Django实现细节

1. 项目结构

假设我们创建一个名为file_download的Django项目，并包含一个应用documents。

项目结构如下：

file_download/
│
├── manage.py
├── file_download/
│   ├── __init__.py
│   ├── settings.py
│   ├── urls.py
│   └── wsgi.py
└── documents/
    ├── migrations/
    ├── __init__.py
    ├── admin.py
    ├── apps.py
    ├── models.py
    ├── tests.py
    ├── urls.py
    └── views.py

2. 安装依赖

确保你已经安装了Django和相关库。

pip install django

3. 模型设计

在documents/models.py文件中定义Document模型：

from django.db import models
from django.contrib.auth.models import User

class Document(models.Model):
    user = models.ForeignKey(User, on_delete=models.CASCADE)
    file_path = models.FileField(upload_to='documents/')
    original_name = models.CharField(max_length=255)

    def __str__(self):
        return self.original_name

4. 视图逻辑

在documents/views.py中实现文件上传和下载的功能：

from django.shortcuts import render, redirect
from django.http import HttpResponse
from .models import Document
from django.utils.http import urlquote
import os

def upload_file(request):
    if request.method == 'POST':
        file = request.FILES['file']
        document = Document(user=request.user, file_path=file, original_name=file.name)
        document.save()
        return redirect('document_list')
    return render(request, 'upload.html')

def download_file(request, document_id):
    document = Document.objects.get(id=document_id)
    file_path = document.file_path.path  # 获取文件实际路径
    response = HttpResponse(open(file_path, 'rb'), content_type='application/octet-stream')
    
    # 自定义文件名
    custom_filename = f"custom_name_{document.id}_{document.original_name}"
    response['Content-Disposition'] = f'attachment; filename="{urlquote(custom_filename)}"'
    
    return response

5. URL路由

在documents/urls.py中设置URL路由：

from django.urls import path
from .views import upload_file, download_file

urlpatterns = [
    path('upload/', upload_file, name='upload_file'),
    path('download/<int:document_id>/', download_file, name='download_file'),
]

在file_download/urls.py中包含应用的URL：

from django.urls import path, include

urlpatterns = [
    path('documents/', include('documents.urls')),
]

6. 前端界面

在templates/upload.html中创建上传文件的表单：

<form method="post" enctype="multipart/form-data">
  {% csrf_token %}
  <input type="file" name="file" required>
  <button type="submit">上传</button>
</form>