用于经过身份验证的用户的条件上下文处理器

Question

我有一个上下文处理器返回用户朋友列表。我希望它仅在用户登录时返回朋友的字典，因为目前我有干净的数据库，没有任何用户，并且我收到错误：

异常类型：DoesNotExist at /
异常值：UserProfile 匹配查询不存在。

这是我尝试过的两个版本，但没有任何运气。为什么它不应该搜索匹配的用户？

def friends_list(request):
    if request.user.is_authenticated:
        userprofile = UserProfile.objects.get(user=request.user)
        friends = Friend.objects.filter(friend_of=userprofile)
    else:
        friends = {}
    return {'friends': friends}

def friends_list(request):
    userprofile = UserProfile.objects.get(user=request.user)
    if userprofile:
        friends = Friend.objects.filter(friend_of=userprofile)
    else:
        friends = {}
    return {'friends': friends}

Answer 1

我不知道您的系统如何创建 UserProfiles，但看起来即使用户登录，也可能没有他们的 UserProfiles。您应该假设 UserProfile 可能不存在来编写代码：

def friends_list(request):
    userprofile = None
    if request.user.is_authenticated:
        try:
            userprofile = UserProfile.objects.get(user=request.user)
        except DoesNotExist:
            pass
    if userprofile:
        friends = Friend.objects.filter(friend_of=userprofile)
    else:
        friends = []
    return {'friends': friends}

Answer 2

您收到错误，因为同步时创建管理员不会创建您的 UserProfile，因此 UserProfile.get() 会引发DoesNotExist 异常。

将其更改

userprofile = UserProfile.objects.get(user=request.user)

为

userprofile,created = UserProfile.objects.get_or_create(user=request.user)

：就可以了。