我有一个 Django DRF 后端,实现了令牌用户身份验证。应用程序本身是一个聊天,由公共大厅和私人线程(特定用户之间的线程)组成。目前,我的模型文件包括与 Lobby 和 Thread 模型具有外键关系的 Message 模型,并将这些字段之一保留为 Null,以确定特定消息是否涉及私有 Thread 或公共 Lobby。
到目前为止,我一直专注于开发公共大厅,用户根据他当前所在的大厅查询消息。
在我完成 Lobbies 的开发并启用身份验证后,前端停止按 Lobby 标题过滤消息,只是简单地将所有与当前经过身份验证的用户相关的消息返回给我——显然忽略了与消息相关的 Lobbies 之间的任何区别。在检查了实际的 API 之后,我发现当我启用身份验证时,过滤选项完全从页面中消失了。
how can I combat this issue?
编辑:我用一些代码更新了帖子
我宁愿保留一个消息模型,而不是为大厅和线程创建单独的消息模型。
这是我的模型:
class Thread(models.Model):
user1 = models.ForeignKey(Account, on_delete=models.PROTECT, related_name="user1")
user2 = models.ForeignKey(Account, on_delete=models.PROTECT, related_name="user2")
date = models.DateTimeField(auto_now_add=True, blank=False)
def __repr__(self):
return f"user1: {self.user1} user2: {self.user2}"
class Message(models.Model):
content = models.TextField(max_length=255, default="", blank=False, null=False)
user = models.ForeignKey(Account, on_delete=models.PROTECT, related_name="user")
date = models.DateTimeField(auto_now_add=True, blank=False)
lobby = models.ForeignKey(Lobby, on_delete=models.CASCADE, related_name="messages", blank=True, null=True )
thread = models.ForeignKey(Thread, on_delete=models.CASCADE, related_name="messages", blank=True,
null=True )
class Lobby(models.Model):
title = models.TextField(max_length=90, blank=False, null=False)
thumb = models.ImageField(default='lobby.jpg')
这是序列化程序:
class ThreadSerializer(serializers.ModelSerializer):
user1 = serializers.CharField(source="user1.username")
user1thumb = serializers.CharField(source="user1.accountimage.image.url")
user2 = serializers.CharField(source="user2.username")
user2thumb = serializers.CharField(source="user2.accountimage.image.url")
class Meta:
model= Thread
fields = ["user1", "user2", "user1thumb", "user2thumb"]
class MessageSerializer(serializers.ModelSerializer):
username = serializers.CharField(source='user.username')
thumb = serializers.CharField(source='user.accountimage.image.url')
class Meta:
model= Message
fields=['id','content', 'username', 'thumb']
class LobbySerializer(serializers.ModelSerializer):
class Meta:
model=Lobby
fields="__all__"
depth=2
最后是意见:
class ThreadView(viewsets.ModelViewSet):
queryset = Thread.objects.all()
serializer_class = ThreadSerializer
def get_queryset(self):
print(self.request.user)
return Thread.objects.filter(Q(user1= self.request.user) | Q(user2=self.request.user))
class MessageView(viewsets.ModelViewSet):
queryset = Message.objects.all()
serializer_class = MessageSerializer
filter_fields=('lobby__title',)
class LobbyView(viewsets.ModelViewSet):
queryset = Lobby.objects.all()
serializer_class = LobbySerializer
在我的 settings.py 中,我全局启用了身份验证:
REST_FRAMEWORK = {
'DEFAULT_AUTHENTICATION_CLASSES':[
'rest_framework.authentication.TokenAuthentication',
'rest_framework.authentication.SessionAuthentication'
],
'DEFAULT_PERMISSION_CLASSES': (
'rest_framework.permissions.IsAuthenticated',
)
}