扩展django的user表

最新推荐文章于 2023-05-01 16:30:52 发布

原创最新推荐文章于 2023-05-01 16:30:52 发布 · 1k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#django

django笔记专栏收录该内容

16 篇文章

订阅专栏

本文介绍如何通过定义新的Profile模型并使用一对一关系映射到Django的内置User模型来扩展User表，同时还包括了如何设置多媒体文件上传路径、定义表单以便在前端编辑数据以及实现编辑视图函数的方法。

扩张django的user表来增加user中的字段

定义模型

from django.db import models
from django.conf import settings


class Profile(models.Model):
    #一对一的关系映射
    user = models.OneToOneField(settings.AUTH_USER_MODEL)
    date_of_birth = models.DateField(blank=True, null=True)
    photo = models.ImageField(upload_to='users/%Y/%m/%d', blank=True)

    def __str__(self):
        return self.user.username

为了Django能在开发服务中管理用户上传的多媒体文件，在项目setting.py文件中添加如下设置：

MEDIA_URL = '/media/'
MEDIA_ROOT = os.path.join(BASE_DIR, 'media/')

MEDIA_URL 是管理用户上传的多媒体文件的主URL，MEDIA_ROOT是这些文件在本地保存的路径。我们动态的构建这些路径相对我们的项目路径来确保我们的代码更通用化。


在主urls 配置 文件的访问路径
from django.conf import settings
from django.conf.urls.static import static
if settings.DEBUG:
    urlpatterns += static(settings.MEDIA_URL,
                        document_root=settings.MEDIA_ROOT)

定义modelform 让在前端能修改user模型进行校验:

from .models import Profile

class UserEditForm(forms.ModelForm):
    class Meta:
        model = User
        fields = ('first_name', 'last_name', 'email')

class ProfileEditForm(forms.ModelForm):
    class Meta:
        model = Profile
        fields = ('date_of_birth', 'photo')

UserEditForm：允许用户编辑它们的first name,last name, e-mail 这些储存在User模型（model）中的内置字段。
ProfileEditForm：允许用户编辑我们存储在定制的Profile模型（model）中的额外数据。用户可以编辑他们的生日数据以及为他们的profile上传一张照片。

编辑视图函数

from .forms import LoginForm, UserRegistrationForm, \
UserEditForm, ProfileEditForm
@login_required
def edit(request):
    if request.method == 'POST':
        user_form = UserEditForm(instance=request.user,
                                data=request.POST)
        profile_form = ProfileEditForm(instance=request.user.profile,
                                        data=request.POST,
                                        files=request.FILES)
        if user_form.is_valid() and profile_form.is_valid():
            user_form.save()
            profile_form.save()
    else:
        user_form = UserEditForm(instance=request.user)
        profile_form = ProfileEditForm(instance=request.user.profile)
    return render(request,
                 'account/edit.html',
                 {'user_form': user_form,
                 'profile_form': profile_form})