src/partners/models.py

   1 from django.db import models
   2
   3
   4 class Partner(models.Model):
   5     name = models.CharField(max_length=255)
   6     key = models.CharField(max_length=255)
   7
   8     def __str__(self):
   9         return self.name
  10
  11     def get_price(self, pages):
  12         price_obj = self.pricelevel_set.exclude(
  13             min_pages__gt=pages
  14         ).order_by('-price').first()
  15         if price_obj is None:
  16             return None
  17         return price_obj.price
  18
  19
  20 class PriceLevel(models.Model):
  21     partner = models.ForeignKey(Partner, models.CASCADE)
  22     min_pages = models.IntegerField(null=True, blank=True)
  23     price = models.DecimalField(max_digits=10, decimal_places=2)
  24
  25     class Meta:
  26         ordering = ('price',)