快捷導(dǎo)航

詳解Django自定義圖片和文件上傳路徑(upload_to)的2種方式

更新時間：2020年12月01日 10:07:04 作者：大江狗

這篇文章主要介紹了詳解Django自定義圖片和文件上傳路徑(upload_to)的2種方式，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧

最近在做一個仿知乎網(wǎng)站的項目了，里面涉及很多圖片和文件上傳。趁此機會我給大家總結(jié)下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, 那么圖片上傳后數(shù)據(jù)庫中的avatar字段為avatar/sky.jpg, 該字段指向圖片對象，而非絕對路徑。要在模板中使用該圖片，應(yīng)該使用avatar.url (即/media/avatar/sky.jpg）。

但在實際應(yīng)用中，請千萬別這么做。這里有2個嚴(yán)重問題。

所有用戶都把頭像上傳到了同一個avatar文件夾了
原文件名是什么，那么新文件名就是什么

試想用戶很多，很可能發(fā)生文件重名問題，造成后來用戶上傳的文件把前面用戶上傳的頭像覆蓋了，造成了用戶A掛用戶B頭像的狀況。

正確的做法是動態(tài)定義上傳路徑，把圖片存儲到用戶自己的文件夾下，并對其重命名。如下圖所示。這樣圖片就會保存在/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="頭像")

上述案例顯然還有一個問題，不同系統(tǒng)路徑分隔符/和\是不一樣的，為保證代碼在不同系統(tǒng)中能重用，更好的方式是使用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文件，我們?nèi)绾伟阉鼈兎旁谕挥脩舻牟煌募A下呢？實現(xiàn)這個很簡單，如下圖所示。

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最簡單直白，但有一個較大缺陷，文件上傳后未經(jīng)處理就直接存儲了。假如用戶上傳了圖片，我們希望先對其壓縮或裁剪，然后再存儲，或者我們不希望上傳圖片或文件到默認(rèn)的路徑，這時我們就有必要在視圖中自定義圖片或文件路徑了。例子如下。

@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 # 將圖片路徑修改到當(dāng)前會員數(shù)據(jù)庫
     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

到此這篇關(guān)于詳解Django自定義圖片和文件上傳路徑(upload_to)的2種方式的文章就介紹到這了,更多相關(guān)Django 上傳路徑內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: