Django从好友列表中获取用户点赞

from django.db import models
from django.contrib.auth.models import User

class Friendship(models.Model):
    user = models.ForeignKey(User, related_name='friends', on_delete=models.CASCADE)
    friend = models.ForeignKey(User, related_name='friend_of', on_delete=models.CASCADE)

class Post(models.Model):
    author = models.ForeignKey(User, on_delete=models.CASCADE)
    content = models.TextField()

class Like(models.Model):
    user = models.ForeignKey(User, on_delete=models.CASCADE)
    post = models.ForeignKey(Post, on_delete=models.CASCADE)

def get_likes_from_friends(user):
    # 获取用户的好友列表
    friends = Friendship.objects.filter(user=user).values_list('friend', flat=True)
    
    # 获取好友的点赞
    likes = Like.objects.filter(user__in=friends)
    
    return likes

likes = Like.objects.filter(user__in=friends).select_related('post')

from django.db import transaction

@transaction.atomic
def add_like(user, post):
    Like.objects.create(user=user, post=post)