Django：如何创建一个由给定外键分组（排名）的辅助ID字段？

Question

输出可能看起来像这样，例如：

ID	Secondary_id	FK
1	1	1 1
2	2	1
3	3	1
4	1	2
5	2	2

在上下文中：（

请参见下面的模型）

我有一个佣金结构，该结构将取决于多少用户一个月内收入。 理想情况下，我需要在我的佣金支架模型中知道，这是给定结构的支架索引。

这是我的模型。

class CommissionStructure(APIBaseModel):
    advisor = models.ManyToManyField(AdviserDetail)
    name = models.CharField(max_length=20, blank=True, null=True, default='default')
    company = models.ForeignKey(Company, on_delete=models.CASCADE)
    start_dt = models.DateTimeField(auto_now_add=True)
    end_dt = models.DateTimeField(default=timezone.datetime.max)

    objects = CommissionStructureManager()



class CommissionBracket(APIBaseModel):
    <secondary_id ???>
    commission_structure = models.ForeignKey(CommissionStructure, on_delete=models.CASCADE, related_name="brackets")
    lower_bound = models.DecimalField(decimal_places=2, default=0.00, max_digits=20, null=True, blank=True)
    upper_bound = models.DecimalField(decimal_places=2, default=0.00, max_digits=20, null=True, blank=True)

请注意，如果可以在聚合套件中添加注释，我可能不必将其存储在模型上，但我的偏爱是擦干。

谢谢

Answer 1

我的建议是。您可以将辅助ID添加为 commissionBracket 中的整数字段。然后，您可以实现以下内容：

from django.db import connection

def sample_view(request):
   ...

   with connection.cursor() as cursor:
      cursor.execute('''
         INSERT INTO appname_commissionbracket (
            secondary_id, 
            commission_structure_id
         )
         SELECT CASE 
            WHEN MAX(secondary_id)
               THEN MAX(secondary_id) + 1
               ELSE 1
            END AS new_secid, %s
         FROM appname_commissionbracket 
         WHERE commission_structure_id = %s''',
         [1, 1] # Sample foreign key value
      )

   return render(...)

在这里我们使用 ins select 从同一表中基于新记录的 secondary_id 。我们还添加a case ，以便我们可以在没有记录的情况下拥有后退值使用 commission_structure_id 值返回1。

如果您需要在创建过程中填充其他列，则可以简单地将它们包括在内：

INSERT INTO (secondary_id, commission_structure_id, lower_bound, upper_bound)
SELECT CASE ... END AS new_secid, <fk_value>, <lower_bound_value>, <upper_bound_value>

Answer 2

我找到了一种注释QuerySet的方法，但是为了兴趣，我最初的问题仍然存在：如何添加外键分区的另一个字段？

brackets = CommissionBracket.objects.select_related("commission_structure")\
            .prefetch_related(
            'commission_structure__advisor',
            'commission_structure__start_dt__gte',
            'commission_structure__end_dt__lte',
            'commission_structure__company',
            'bracket_values'
            ).filter(
                commission_structure__advisor=advisor,
                commission_structure__start_dt__lte=date,
                commission_structure__end_dt__gte=date,
                commission_structure__company=advisor.user.company,
            ).annotate(index=Window(
                    expression=Count('id'),
                    partition_by="commission_structure",
                    order_by=F("lower_bound").asc()))