Django 从多个函数创建类

Question

我的 utils.py 文件中有几个函数，我的视图使用这些函数来计算交易统计数据。下面我只添加了 3 个，但有 8 个，很快就会有更多。我目前设置它的方式有效，但效率非常低。你可以看到每个函数都是从头开始调用其他设置类似变量的函数。

把它变成一个类的正确方法是什么？

这是正确的方法吗？

---

utils.py |多种功能

def max_amount_function (pk):
    trade = get_object_or_404(Trade, pk=pk)
    entries_buy = Entry.objects.filter(trade=trade, entry_type="entry")
    max_amount_function = entries_buy.aggregate(max_amount_function=Sum('amount'))['max_amount_function']
    if max_amount_function is None:
        return 0
    else:
        return max_amount_function

def fees_function (pk):
    trade = get_object_or_404(Trade, pk=pk)
    entries = Entry.objects.filter(trade=trade)
    fees_function = entries.aggregate(fees_function=Sum('fee'))['fees_function']
    return fees_function

def entry_cpu_function (pk):
    if max_amount_function(pk) > 0:
        trade = get_object_or_404(Trade, pk=pk)
        entries_buy = Entry.objects.filter(trade=trade, entry_type="entry")
        entry_cpu_function = entries_buy.annotate(
            s=F('amount') * F('price')
        ).aggregate(
            entry_cpu_function=ExpressionWrapper(
                Sum(
                    Cast('s', output_field=models.FloatField())
                ) / Sum('amount'),
                output_field=models.FloatField()
            )
        )['entry_cpu_function']
        return entry_cpu_function
    else:
        return 0

utils.py |一节课

class TradeStats:
    trade = get_object_or_404(Trade)
    entries_buy = Entry.objects.filter(trade=trade, entry_type="entry")

    def __init__(self):
        pass

    def max_amount_functio(pk):
        pass

    def fees_function(pk):
        pass

    def entry_cpu_function(self):
        pass

编辑添加了 models.py，因为它可能与条目相关，因为它是一个不同的类。

模型.py

class Trade(models.Model):
    user = models.ForeignKey(User, on_delete=models.CASCADE, blank=True)
    status = models.CharField(max_length=2, choices=STATUS_CHOICES, default='op')
    broker = models.ForeignKey(Broker, on_delete=models.CASCADE, blank=True, null=True)
    asset = models.ForeignKey(Asset, on_delete=models.CASCADE, null=True)
    ...

class Entry(models.Model):

    ENTRY = 'entry'
    EXIT = 'exit'
    ENTRY_TYPE_CHOICES = [
        (ENTRY, 'Entry'),
        (EXIT, 'Exit'),

    trade = models.ForeignKey(Trade, on_delete=models.CASCADE)
    amount = models.FloatField(null=True)
    price = models.FloatField(null=True, blank=True)
    entry_type = models.CharField(max_length=5, choices=ENTRY_TYPE_CHOICES, default=ENTRY)
    ...

Answer 1

您可以将所有这些函数移动到模型或自定义 QuerySet 下。

如果你转向模型，你可以做类似的事情。 （代码可能行不通，仅供参考）

# MODEL
class Trade(models.Model):

    # Model fields here

    def get_entries(self):
        # NB: assuming you didn't specify a related_name
        # in the `Entry.trade` ForeignKey field. Else,
        # replace `entry_set` with said related_name.
        return self.entry_set.filter(entry_type="entry")


    def max_amount_function(self):
        max_amount_res = self.get_entries().aggregate(max_amount_function=Sum('amount'))['max_amount_function']
        if max_amount_res is None:
            return 0
        else:
            return max_amount_res

    def fees_function(self):
        return self.get_entries().aggregate(fees_function=Sum('fee'))['fees_function']

    def entry_cpu_function(self, ):

        if self.max_amount_function() <= 0:
            return 0

        entry_cpu_function = self.get_entries().annotate(
            s=F('amount') * F('price')
        ).aggregate(
            entry_cpu_function=ExpressionWrapper(
                Sum(
                    Cast('s', output_field=models.FloatField())
                ) / Sum('amount'),
                output_field=models.FloatField()
            )
        )['entry_cpu_function']
        return entry_cpu_function

在views.py中，你可以像下面这样引用这个函数 -

trade = get_object_or_404(Trade, pk=pk)
trade.max_amount_function()
trade.fees_function()
trade.entry_cpu_function()

Answer 2

Rockstar 已经发布了在这种情况下的最佳解决方案 - 您正在使用 Django 模型，并且您的所有代码都依赖于特定的模型实例，因此将这些功能作为方法添加到您的 Trade 模型中是非常明显的解决方案。

话虽如此，这个问题本身（如何将一堆依赖于相同变量的函数变成一个类）也可以用更通用的答案来解决，该答案也适用于其他上下文：

首先，您要修复此类定义，因此 trades 和 entries_buy 是 instance 属性，而不是类属性 - 否则它们将在模块导入时仅初始化一次，您将结束加上陈旧的数据。为此，您还需要将 pk 参数添加到您的类初始化程序中：

class TradeStats:
    def __init__(self, pk):
        self.trade = get_object_or_404(Trade, pk=pk)
        self.entries_buy = Entry.objects.filter(trade=self.trade, entry_type="entry")

然后您想将self 作为第一个参数添加到您的所有方法（否则当前实例无法传递 - 并删除现在无用的 'pk' 参数：

def max_amount_function(self):
    pass

def fees_function(self):
    pass

def entry_cpu_function(self):
    pass

现在您可以向后移植您的方法实现，将相关部分替换为对self.trade 和self.entries_buy 的访问，即：

def max_amount_function(self):
    result = self.entries_buy.aggregate(max_amount_function=Sum('amount'))['max_amount_function']
    return 0 if result is None else result

等等……

此外，“xxx_yyy_function”是一个非常糟糕的命名。作为一般规则，您希望对对象使用名词（变量、值、类名等），对函数或方法使用动词，所以人们会期望像 get_max_amount、get_fees 等这样的东西。

最后，鉴于您的问题（以及您第二次 sn-p 的样子），我强烈建议您花一些时间在 the official Python tutorial - the part about classes 当然，但您可能会学到很多其他有用的东西教程。