如何在django模型中获取用户id？

Question

我需要从登录用户获取 id 来过滤模型中的查询。如何在 django 模型中获取用户 id？ 感谢

models.py：

class Portfo(models.Model):
    product = models.ForeignKey(Product, on_delete=models.CASCADE, null=True)
    sell = models.ForeignKey(Sell, on_delete=models.CASCADE, null=True)
    customer = models.ForeignKey(Customer, on_delete=models.CASCADE, null=True)
    profit = models.PositiveIntegerField(null=True)
    final_amount = models.IntegerField(null=True)

    def __str__(self):
        return self.product.name

    def final_amount(self):
        final_amount = 0
        buy = Buy.objects.filter(product_id=self.product.id, owner__exact=...)
        sell = Sell.objects.filter(product_id=self.product.id)

Answer 1

我需要从登录用户获取 ID 来过滤模型中的查询。

模型被设计为不感知请求。如果要检索用户 ID，则需要通过参数传递它，因此：

class Portfo(models.Model):
    # …

    def final_amount(self, user_id):
        final_amount = 0
        buy = Buy.objects.filter(product_id=self.product_id, owner_id=user_id)
        sell = Sell.objects.filter(product_id=self.product_id, owner_id=user_id)
        # …

在视图中，您可以使用 .final_amount(…) 方法，例如：

from django.contrib.auth.decorators import login_required
from django.shortcuts import get_object_or_404

@login_required
def some_view(request, pk):
    my_porfolio = get_object_or_404(Portfolio, pk=pk)
    final_amount = my_portfolio.final_amount(request.user.id)
    # …