如何解决如何添加到默认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)
这似乎可行,我现在可以访问用户的收藏夹,但是我也许这不是一个好习惯。使用我的方法,创建的新模型中没有“收藏夹”对象。只有当用户决定添加喜欢的食谱时,才会创建该视图,并且如果该视图尚不存在,则上述视图将创建一个视图。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。