Django Rest Framework ViewSet 基于外键用户字段的过滤问题

Question

我有一个我正在做的django项目。此项目中有两个模型。有一个user和account模型。我正在集成 django rest 框架视图集。我将在下面包括它们。我现在正在将Django Rest框架集成到项目中。我试图挖掘如何做两件事。

2 型号：

默认 django 用户

帐户模型：

class Account(models.Model):
    user = models.ForeignKey(User, on_delete=models.CASCADE)
    _id = models.CharField(max_length=45)
    name = models.CharField(max_length=140)
    balance = models.DecimalField(max_digits=100, decimal_places=2)
    currency = models.CharField(max_length=12)
    bank_name = models.CharField(max_length=120)
    routing = models.CharField(max_length=8)
    _class = models.CharField(max_length=22)
    type = models.CharField(max_length=22)
    active = models.BooleanField(default=True)
    main = models.BooleanField(default=False)
    synapse = models.BooleanField(default=False)
    create_at = models.DateTimeField(auto_now_add=True)
    def __str__(self):
        return self.user.username + ' - ' + self.name + ': ' + str(self.balance)

1 => 我希望能够输入端点/api/users/accounts/omarjandlai并使用 omarjandali 用户外键获取单个或多个计数

2 =>我希望能够输入以下内容 api/users/accounts/并返回数据库中的所有帐户

我尝试了 4 - 5 种不同的方法来让它工作，但我无法让它工作。

这是我的序列化程序和视图

序列化程序：

class AccountSerializer(serializers.ModelSerializer):
    class Meta:
        model = Account
        fields = ('user', '_id', 'name', 'balance', 'currency', 'bank_name',
                  'routing', '_class', 'type', 'active', 'main', 'synapse')

视图：

class AccountViewSet(viewsets.ModelViewSet):
    queryset = Account.objects.all();
    serializer_class = AccountSerializer
    filter_backends = (filters.DjangoFilterBackend,)
    filter_fields = ('user',)

正在发生的事情是，当我做

api/users/account我看到所有帐户

api/users/account/omarjandali我得到detail not found

网址：

router = DefaultRouter()
router.register(r'users', UserViewSet, basename='user')
router.register(r'accounts', AccountViewSet, basename='account')
urlpatterns = router.urls

Answer 1

我可以看到您正在使用DefaultRouter的默认功能。这样，您只能使用其主键访问记录详细信息。由于你没有明确指定任何字段作为主键（primary=True），Django 设置了字段 id，它PositiveIntegerField为模型的主键。因此，如果要访问记录的详细信息，则必须使用 api/users/account/112/ .

您可以使用过滤器按名称访问记录：api/users/account/?search='omarjandali'，但您需要添加SearchFilter 。

from rest_framework.filters import SearchFilter 
class AccountViewSet(viewsets.ModelViewSet):
    queryset = Account.objects.all();
    serializer_class = AccountSerializer
    filter_backends = (filters.DjangoFilterBackend, SearchFilter)
    filter_fields = ('user',)
    search_fields = ('name',)

如果你想坚持 url 模式api/users/account/omarjandali/，你必须将其添加到 urls.py 并使用另一个视图集，以便让原始视图根据 DefaultRouter 定义运行：

urls.py：

router = DefaultRouter()
router.register(r'users', UserViewSet, basename='user')
router.register(r'accounts', AccountViewSet, basename='account')
urlpatterns = [
    path('', include(router.urls)),
    path('api/users/account/<str:username>/', views.GetRecordsByNameViewSet.as_view({'get': 'list'}),
]

views.py

from rest_framework.response import Response
class GetRecordsByNameViewSet(viewsets.ViewSet):
    def list(self, request, username):
        accounts = Account.objects.filter(user__username=username)
        accounts_to_return = AccountSerializer(accounts, many=True).data
        return Response(accounts_to_return)

Answer 2

您可以尝试覆盖

get_queryset（方法）

这样，无需在URL中提供用户名/电子邮件。您将执行localhost8000/帐户，它只会显示相应用户的帐户。

class AccountViewSet(viewsets.ModelViewSet):
    queryset = Account.objects.all();
    serializer_class = AccountSerializer
    filter_backends = (filters.DjangoFilterBackend,)
    filter_fields = ('user',)
    def get_queryset(self):
        queryset = super().get_queryset()
        queryset = queryset.filter(user=self.request.user)
        return queryset

注意：如果在项目中设置了身份验证，这将起作用。对于令牌身份验证，您需要在授权标头中提供令牌。