如何添加到默认Django用户模型的ManyToManyField扩展?

问题描述

对于我的应用程序,我想向认的用户模型(django.contrib.auth.models.User)添加一个额外的ManyToManyField。这个额外的字段称为“收藏夹”,用户喜欢的帖子应该去那里。这就是我所拥有的:

class Favorite(models.Model):
    user = models.OnetoOneField(User,related_name='favorites',on_delete=models.CASCADE)
    favorites = models.ManyToManyField(Recipe,related_name='favorited_by')

这就是我尝试从shell添加到“收藏夹”时得到的。

# imported Recipe,Favorite,User(default)
>>> recipe1 = Recipe.objects.all()[0]
>>> me = User.objects.all()[0]
>>> me.favorites.add(recipe1)
django.contrib.auth.models.User.favorites.RelatedobjectDoesNotExist: User has no favorites.

# Just checking if the the User object,me,has a 'favorites' attribute
>>> 'favorites' in dir(me)
True

向此“收藏夹”字段添加配方对象的正确方法是什么?

作为更多参考,我做了类似的处理用户之间友谊的方法,但是由于我没有扩展用户模型,所以这有点简单。下面的代码可以正常工作:

class Friend(models.Model):
    users = models.ManyToManyField(User)
    current_user = models.ForeignKey(User,related_name='owner',null=True,on_delete=models.CASCADE)

    @classmethod
    def make_friend(cls,current_user,new_friend):
        friend,created = cls.objects.get_or_create(
            current_user=current_user
        )
        friend.users.add(new_friend)

    @classmethod
    def lose_friend(cls,created = cls.objects.get_or_create(
            current_user=current_user
        )
        friend.users.remove(new_friend)

解决方法

已解决。我的解决方案在下面,但是我不确定这是否是好习惯。

django.contrib.auth.models.User.favorites.RelatedObjectDoesNotExist: User has no favorites.

用户模型可能具有“收藏夹”字段,但实际上需要用“收藏夹”对象填充它。我是通过在views.py中编写一个函数来实现的:

def add_favorite(request,pk):
    # Check if the user has a favorites field. If not create one and add. If yes,just add
    user_favorites,created = Favorite.objects.get_or_create(
        user=request.user
        )
    recipe = get_object_or_404(Recipe,pk=pk)
    user_favorites.favorites.add(recipe)

这似乎可行,我现在可以访问用户的收藏夹,但是我也许这不是一个好习惯。使用我的方法,创建的新模型中没有“收藏夹”对象。只有当用户决定添加喜欢的食谱时,才会创建该视图,并且如果该视图尚不存在,则上述视图将创建一个视图。