X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/9d763695198b109f00b5d88a5cfe0190871271aa..9eeeeab0b4b72b405b2c2a6e69969537706535bf:/src/club/models.py diff --git a/src/club/models.py b/src/club/models.py index 98b4ee00e..676669497 100644 --- a/src/club/models.py +++ b/src/club/models.py @@ -69,6 +69,10 @@ class Club(models.Model): c[f'{tag}_wide_spot'] = wide_spot return c + def get_description_for_amount(self, amount, monthly): + amounts = self.monthlyamount_set if monthly else self.singleamount_set + amount = amounts.all().filter(amount__lte=amount).last() + return amount.description if amount is not None else '' class SingleAmount(models.Model): @@ -146,6 +150,10 @@ class Schedule(models.Model): super(Schedule, self).save(*args, **kwargs) self.update_contact() + def get_description(self): + club = Club.objects.first() + return club.get_description_for_amount(self.amount, self.monthly) + def initiate_payment(self, request): return self.get_payment_method().initiate(request, self)