Django框架中的檔案上傳技巧-Python教學-PHP中文網

Django框架中的檔案上傳技巧

WBOY

發布： 2023-06-18 08:24:06

原創

4548 人瀏覽過

近年來，網路應用程式逐漸流行，而其中許多應用程式都需要檔案上傳功能。在 Django 框架中，實現上傳文件功能並不困難，但是在實際開發中，我們還需要處理上傳的文件，其他操作包括更改文件名、限製文件大小等問題。本文將分享一些 Django 框架中的檔案上傳技巧。

一、設定檔上傳項目
在 Django 專案中，要設定檔上傳需要在 settings.py 檔案中進行設定。下面我們介紹其中的幾個重要配置項。

檔案上傳路徑
在settings.py 中，我們可以定義儲存上傳檔案的路徑，如下：
```
MEDIA_ROOT = os.path.join(BASE_DIR, 'uploads')
```
登入後複製
在這個設定項中，我們定義了儲存上傳檔案的根目錄為uploads，媒體檔案的路徑是相對於settings.py 檔案所在的目錄計算的。我們也可以在 Django 專案的 urls.py 中新增對應的 URL 規則，這樣使用者就可以透過造訪該 URL 來取得上傳的檔案。
支援檔案格式
檔案格式的控制是一個非常重要的內容，我們可以透過下面的設定項指定上傳檔案的格式。
```
ALLOWED_FILE_TYPES = ['jpg', 'jpeg', 'png', 'gif', 'pdf']
```
登入後複製
注意，我們必須將所有的檔案格式轉換成小寫字母。
檔案大小限制
Django 框架預設限制所有檔案的大小都不能超過 2.5MB。如果我們需要修改這個限制值，可以在 settings.py 配置項目中進行修改。
```
MAX_UPLOAD_SIZE = 10485760 # 10MB
```
登入後複製
在這個設定中，我們將最大上傳檔案大小限制為 10MB。

二、檔案上傳
我們可以透過 Django 表單元件實作檔案上傳。以下是一個簡單的檔案上傳表單範例：

<form method="POST" enctype="multipart/form-data">
    {% csrf_token %}
    <input type="file" name="document" accept=".pdf">
    <input type="submit" value="Submit">
</form>

登入後複製

注意，enctype 屬性必須為 multipart/form-data 才能支援檔案上傳。

在視圖函數中，我們可以使用 request.FILES.get() 方法來取得上傳的檔案。這個方法回傳的是一個文件物件。以下是一個簡單的範例：

def upload(request):
    if request.method == 'POST':
        file = request.FILES.get('document')
        if file:
            handle_uploaded_file(file)
            messages.success(request, '上传成功')
            return redirect('home')
        else:
            messages.error(request, '上传失败')
            return redirect('upload')
    else:
        return render(request, 'upload.html')

登入後複製

在這個範例中，我們先取得上傳的檔案對象，然後呼叫 handle_uploaded_file() 函數將其儲存到指定的路徑。在沒有上傳檔案的情況下，我們可以透過 messages 模組將錯誤訊息提示給使用者。

三、文件命名
為了更好地管理上傳到伺服器的文件，我們建議將上傳的文件命名為含有時間戳記的唯一文件名稱。下面是一個簡單的實例：

def handle_uploaded_file(file):
    date = datetime.now().strftime("%Y%m%d%H%M%S")
    filename = f"{date}_{file.name}"
    path = os.path.join(MEDIA_ROOT, filename)
    with open(path, 'wb+') as destination:
        for chunk in file.chunks():
            destination.write(chunk)

登入後複製

首先，我們使用 datetime 模組來取得目前日期和時間，並將其轉換為字串格式。然後，我們將這個時間戳記和原始檔案名稱結合產生一個新的檔案名稱。最後，我們將新檔案名稱與上傳檔案的路徑結合，並使用檔案物件的 chunks() 方法將上傳的檔案內容寫入檔案中。

四、檔案大小
為了防止伺服器出現問題，我們需要對上傳的檔案大小進行限制。下面是一個簡單的檔案大小檢查函數：

def check_file_size(file):
    if file.size > MAX_UPLOAD_SIZE:
        raise ValidationError(f'上传文件大小最大为 {MAX_UPLOAD_SIZE / 1048576} MB')

登入後複製

在這個函數中，我們檢查上傳檔案的大小是否超過最大上傳大小。如果超過了，我們就拋出 ValidationError 異常並提示使用者。在視圖函數中處理這個異常：

def upload(request):
    if request.method == 'POST':
        file = request.FILES.get('document')
        if file:
            try:
                check_file_size(file)
            except ValidationError as e:
                messages.error(request, e)
                return redirect('upload')
            else:
                handle_uploaded_file(file)
                messages.success(request, '上传成功')
                return redirect('home')
        else:
            messages.error(request, '上传失败')
            return redirect('upload')
    else:
        return render(request, 'upload.html')

登入後複製

透過這種方式，我們可以更好地控制上傳檔案的大小。

總結：