models.Manager でQueryのだしわけ

Question

Djangoで 次のようなモデルがあるのですが、適切なqueryが思いつかず、悩んでいます。

お客様は複数の契約を持っています。

class Customer(models.Model):
customer_number = models.CharField(max_length=200)
customer_name = models.CharField(max_length=200)

class Contract(models.Model):
customer = models.ForeignKey(Customer, on_delete=models.CASCADE, blank=True, null=True)
contract_name = models.CharField(max_length=200)

class ProductMaster(models.Model):
product_id = models.CharField(max_length=200)
product_name = models.CharField(max_length=200)

各月の請求金額のモデル
class Billing(models.Model):
billing_product = models.ForeignKey(BillingProduct, on_delete=models.CASCADE)
value = models.FloatField()

契約によりプロダクトが決まるのですが、同じプロダクトでも契約ごとに他のプロダクトと
合算で請求するのか個別で請求すのか変わってきます。

class BillingProduct(models.Model):
contract = models.ForeignKey(Contract, on_delete=models.CASCADE)
product = models.ForeignKey(ProductMaster, on_delete=models.CASCADE)
billing_with = models.ForeignKey('self', on_delete=models.CASCADE, blank=True, null=True)

今は出しわけがうまく思いつかずfor文でぐるぐるして解決しています。

単体のプロダクト のパターン -> 自身の名称ででBilling.value を返す

この処理がすっきりしないのですが...
代表にぶら下がるパターン -> 代表の名称ででBilling.value を返す
代表になるパターン -> 自身の名称ででBilling.value を返す

代表の名称で合計してその値を返す

契約で請求対象になっているリストに対して
BillingProduct.objects.get().billingproduct_set.all()
を実行すると請求の代表を返します

契約で請求対象になっているリストに対して
BillingProduct.objects.filter().exclude(billing_with= None)
を実行するとぶら下がっているプロダクトの返します

ここまではDjangoのQueryすっきり結果が返ってきたのですが、models.Managerとかですっきり
できないでしょうか?

コンピュートプロパティーを返す別モデルを作ればいいのでしょうか??

Answer 1

Billingの中に計算を返すフィールドを作成することで少しはすっきりして
modelの中だけで閉じるようにできました。

def _get_BP_billing_set(self):
if len(self.billing_product.billingproduct_set.all()) == 0:
return None
else:
return self.billing_product.billingproduct_set.all()

def _get_kakin_with_children(self):
if self.billing_with == None and self.billing_set == None:
return self.kakin
elif self.billing_set:
children = [Billing.objects.get(billing_product=k, billing_date=self.billing_date, statisticmaste=self.statisticmaste) for k in self.billing_set]
return self.kakin + sum([k.kakin for k in children])
elif self.billing_with == None:
return None

kakin = property(_get_kakin)
new_kakin = property(_get_kakin_with_children)