关注我们

Django 上传头像详解

到目前为止我们的博客处理的都是文字。现代互联网早就进入了“读图”时代，图片的维护、展示也就相当重要。

上一章中预留了avatar字段，用来保存用户上传的头像，现在我们来实现这个功能。

必要的设置

图片属于一种媒体文件，它与静态文件类似，需要设置一个统一的目录，便于集中存储和访问。

这类需要框架统一设置的参数，当然应该在/my_blog/settings.py中。在底部加上：

/my_blog/settings.py

...

# 媒体文件地址
MEDIA_URL = '/media/'
MEDIA_ROOT = os.path.join(BASE_DIR, 'media/')

MEDIA_ROOT和MEDIA_URL是用户上传文件保存、访问的位置：

Django框架擅长的是对逻辑的处理，而对图片这类文件的处理则非常的耗时。因此在实际的生产环境中（即产品上线之后），通常是有专门的Web服务器来处理文件的访问。

而在开发阶段我们不会在意效率问题，所以Django也提供了方法，让开发服务器能够处理图片。

在/my_blog/urls.py添加下面的语句：

/my_blog/urls.py

...
# 新引入的模块
from django.conf import settings
from django.conf.urls.static import static

urlpatterns = [
    ...
]

#添加这行
urlpatterns += static(settings.MEDIA_URL, document_root=settings.MEDIA_ROOT)

这样就为以后上传的图片配置好了URL路径。

编写MTV

回顾一下，avatar的字段已经在上一章写好了：

/userprofile/models.py

...
class Profile(models.Model):
    ...
    avatar = models.ImageField(upload_to='avatar/%Y%m%d/', blank=True)
    ...

upload_to指定了图片上传的位置，即/media/avatar/%Y%m%d/。%Y%m%d是日期格式化的写法，会最终格式化为系统时间。比如说图片上传是2018年12月5日，则图片会保存在/media/avatar/2018205/中。

注意ImageField字段不会存储图片本身，而仅仅保存图片的地址。

记得用pip指令安装Pillow。

表单类在前面也写好了，不用修改：

/userprofile/forms.py

...
class ProfileForm(forms.ModelForm):
    class Meta:
        model = Profile
        fields = ('phone', 'avatar', 'bio')

接着需要修改视图，使之能够对图片进行处理：

/userprofile/views.py

...
@login_required(login_url='/userprofile/login/')
def profile_edit(request, id):
    ...

    # 修改本行
    # 上传的文件保存在 request.FILES 中，通过参数传递给表单类
    profile_form = ProfileForm(request.POST, request.FILES)

    if profile_form.is_valid():
        ...

        # 添加在 profile.bio = profile_cd['bio'] 后面
        # 如果 request.FILES 存在文件，则保存
        if 'avatar' in request.FILES:
            profile.avatar = profile_cd["avatar"]

        ...

表单上传的文件对象存储在类字典对象request.FILES中，因此需要修改表单类的参数，将它一并传递进去。
如果request.FILES中存在键为avatar的元素，则将其赋值给profile.avatar（注意保存的是图片地址）；否则不进行处理。

修改模板文件，添加代码显示、处理用户的头像：

/templates/userprofile/edit.html

...

{% if profile.avatar %}
    <div class="col-md-4">头像</div>
    <img src="{{ profile.avatar.url }}" style="max-width: 20%; border-radius: 15%;" class="col-md-4">
{% else %}
    <h5 class="col-md-4">暂无头像</h5>
{% endif %}
<br>
<br>
<form ... enctype="multipart/form-data">{% csrf_token %}
    <!-- avatar -->
    <div class="form-group">
        <label for="avatar">上传头像</label>
        <input type="file" class="form-control-file" name="avatar" id="avatar">
    </div>

    ...