【发布时间】:2021-01-04 05:21:21
【问题描述】:
我在 Django 中创建了一个好友系统。我为每个用户创建了一个配置文件模型,它有一个朋友字段。我还创建了一个friendrequest 模型来发送和接收请求。一切正常。在每个用户的个人资料(模型)上,我可以看到三个选项:添加朋友、删除朋友和取消请求。但是,如果我所在的个人资料的用户不是我的朋友,我只想显示添加朋友链接。删除好友和取消请求也是如此(如果 FriendRequest 对象存在则显示取消请求)。我试图检查我所在个人资料的用户是否是我的朋友,如果是,则个人资料应显示删除朋友选项,但该操作不起作用。这是我的代码:
我的models.py:
class profile(models.Model):
user = models.OneToOneField(User, on_delete=models.CASCADE, related_name='profile')
desc = models.TextField(null=True)
pfp = models.ImageField(default='social_media/images/profile.jpg', upload_to='profile_pics/', null=True, blank=True)
cover = models.ImageField(default='social_media/images/cover.jpg', upload_to='cover_pics/', null=True, blank=True)
occupation = models.CharField(max_length=250, default='')
friend = models.ManyToManyField(User, blank=True, null=True, related_name='friended')
def __str__(self):
return self.user.username
class FriendRequest(models.Model):
to_user = models.ForeignKey(User, related_name='to_user', on_delete=models.CASCADE)
from_user = models.ForeignKey(User, related_name='from_user', on_delete=models.CASCADE)
timestamp = models.DateTimeField(auto_now_add=True) # set when created
def __str__(self):
return "From {}, to {}".format(self.from_user.username, self.to_user.username)
我的个人资料显示视图:
@login_required
def userprofile(request, user_id=None):
button_status = 'none'
user = get_object_or_404(User, id=user_id)
if user in request.user.friended.all():
button_status = 'not_friend'
if user:
user = User.objects.get(id=user_id)
Post = post.objects.order_by('-created')
return render(request,'social_media/profile.html', {'Post':Post,'User':user})
@login_required
def send_request(request,id):
from_user=request.user
to_user=User.objects.get(id=id)
freq=FriendRequest.objects.get_or_create(from_user=from_user,to_user=to_user)
return redirect(request.META['HTTP_REFERER'])
@login_required
def remove_friend(request,id):
user1=request.user
user2=User.objects.get(id=id)
user1.profile.friend.remove(user2)
user2.profile.friend.remove(user1)
return redirect(request.META['HTTP_REFERER'])
@login_required
def accept_request(request,id):
freq=FriendRequest.objects.get(id=id)
user1=request.user
user2=freq.from_user
user1.profile.friend.add(user2)
user2.profile.friend.add(user1)
freq.delete()
return redirect(request.META['HTTP_REFERER'])
我的个人资料模板:
<div class="right_details_user">
<div class="profile_status">
<div>Status: {{ User.status.status }}</div>
</div>
{% if not user.username == User.username %}
{% if button_status == 'not_friend' %}
<a href="{% url 'add-friend' User.id %}">Add Friend</a>
<br>
{% else %}
<a href="{% url 'remove-friend' User.id %}">Remove Friend</a>
<br>
{% endif %}
<a href="{% url 'cancel-request' User.id %}">Cancel Request</a>
<br>
{% endif %}
{{ User.friended.all }}
</div>
【问题讨论】:
标签: django django-views django-templates django-queryset django-filter