1. 程式人生 > >Django自定義圖片和檔案上傳路徑(upload_to)的2種方式

Django自定義圖片和檔案上傳路徑(upload_to)的2種方式

最近在做一個仿知乎網站的專案了,裡面涉及很多圖片和檔案上傳。趁此機會我給大家總結下Django自定義圖片和檔案上傳路徑的2種方式吧。

方法1: 在Django模型中定義upload_to選項。

Django模型中的ImageField和FileField的upload_to選項是必填項,其儲存路徑是相對於MEIDA_ROOT而來的。

我們來看一個簡單案例(如下所示)。如果你的MEDIA_ROOT是/media/資料夾,而你的上傳資料夾upload_to=“avatar", 那麼你上傳的檔案會自動儲存到/media/avatar/資料夾。

class UserProfile(models.Model):

   user = models.OneToOneField(User,
on_delete=models.CASCADE, related_name='profile')
   avatar = models.ImageField(upload_to='avatar', verbose_name="頭像")
 

如果你的檔名是sky.jpg, 那麼圖片上傳後資料庫中的avatar欄位為avatar/sky.jpg, 該欄位指向圖片物件,而非絕對路徑。要在模板中使用該圖片,應該使用avatar.url (即/media/avatar/sky.jpg)。

但在實際應用中,請千萬別這麼做。這裡有2個嚴重問題。

  • 所有使用者都把頭像上傳到了同一個avatar檔案夾了

  • 原檔名是什麼,那麼新檔名就是什麼

試想使用者很多,很可能發生檔案重名問題,造成後來使用者上傳的檔案把前面使用者上傳的頭像覆蓋了,造成了使用者A掛使用者B頭像的狀況。

正確的做法是動態定義上傳路徑,把圖片儲存到使用者自己的資料夾下,並對其重新命名。如下圖所示。這樣圖片就會儲存在/media/1/avatar/裡了,而且檔案以uuid命名。

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

# Create your models here.

def user_directory_path(instance, filename):
   ext = filename.split('.'
)[-1]
   filename = '{}.{}'.format(uuid.uuid4().hex[:8], ext)
   # return the whole path to the file
   
return "{0}/{1}/{2}".format(instance.user.id, "avatar", filename)

class UserProfile(models.Model):
   user = models.OneToOneField(User, on_delete=models.CASCADE, related_name='profile')
   avatar = models.ImageField(upload_to=user_directory_path, verbose_name="頭像")


上述案例顯然還有一個問題,不同系統路徑分隔符/和\是不一樣的,為保證程式碼在不同系統中能重用,更好的方式是使用python的os模組來拼接路徑。如下圖所示。

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

# Create your models here.

def user_directory_path(instance, filename):
   ext = filename.split('.')[-1]
   filename = '{}.{}'.format(uuid.uuid4().hex[:10], ext)
   # return the whole path to the file
   
return os.path.join(instance.user.id, "avatar", filename)

class UserProfile(models.Model):
   user = models.OneToOneField(User, on_delete=models.CASCADE, related_name='profile')
   avatar = models.ImageField(upload_to=user_directory_path, verbose_name="頭像")
 

使用者上傳檔案可能是圖片,也可能是pdf檔案,我們如何把它們放在同一使用者的不同資料夾下呢?實現這個很簡單,如下圖所示。


def user_directory_path(instance, filename):
   ext = filename.split('.')[-1]
   filename = '{}.{}'.format(uuid.uuid4().hex[:8], ext)
   sub_folder = 'file'
   
if ext.lower() in ["jpg", "png", "gif"]:
       sub_folder = "avatar"
   
if ext.lower() in ["pdf", "docx"]:
       sub_folder = "document"
   
return os.path.join(instance.user.id, sub_folder, filename)

方法2: 在檢視中自定義上傳圖片或檔案路徑

方法1最簡單直白,但有一個較大缺陷,檔案上傳後未經處理就直接儲存了。假如使用者上傳了圖片,我們希望先對其壓縮或裁剪,然後再儲存,或者我們不希望上傳圖片或檔案到預設的路徑,這時我們就有必要在檢視中自定義圖片或檔案路徑了。例子如下。

@login_required
def ajax_avatar_upload(request):
   user = request.user
   user_profile = get_object_or_404(UserProfile, user=user)

   if request.method == "POST":
       form = AvatarUploadForm(request.POST, request.FILES)
       if form.is_valid():

           img = request.FILES['avatar_file']  # 獲取上傳圖片
           cropped_avatar = crop_image(img, user.id)
           user_profile.avatar = cropped_avatar  # 將圖片路徑修改到當前會員資料庫
       
user_profile.save()
   return HttpResponseRedirect(reverse('myaccount:profile'))


def crop_image(file, uid):

   # 隨機生成新的圖片名,自定義路徑。
 
ext = file.name.split('.')[-1]
   file_name = '{}.{}'.format(uuid.uuid4().hex[:10], ext)
   cropped_avatar = os.path.join(uid, "avatar", file_name)
   # 相對根目錄路徑
 
file_path = os.path.join("media", uid, "avatar", file_name)

   # 裁剪圖片,壓縮尺寸為200*200
 
img = Image.open(file)
   crop_im = img.crop((50,50,300, 300)).resize((200, 200), Image.ANTIALIAS)
   crop_im.save(file_path)

   return cropped_avatar