X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/54b3d528298bc6387e88dd48b634c94d350f1c7b..5cb736f7ca55179d06965cc9874d0b059299d933:/src/partners/models.py

diff --git a/src/partners/models.py b/src/partners/models.py
index 32e8a1bb2..583c34c79 100644
--- a/src/partners/models.py
+++ b/src/partners/models.py
@@ -16,6 +16,14 @@ class Partner(models.Model):
             return None
         return price_obj.price
 
+    def get_audio_price(self, minutes):
+        price_obj = self.audiopricelevel_set.exclude(
+            min_minutes__gt=minutes
+        ).order_by('-price').first()
+        if price_obj is None:
+            return None
+        return price_obj.price
+        
 
 class PriceLevel(models.Model):
     partner = models.ForeignKey(Partner, models.CASCADE)
@@ -24,3 +32,12 @@ class PriceLevel(models.Model):
 
     class Meta:
         ordering = ('price',)
+
+
+class AudioPriceLevel(models.Model):
+    partner = models.ForeignKey(Partner, models.CASCADE)
+    min_minutes = models.IntegerField(null=True, blank=True)
+    price = models.DecimalField(max_digits=10, decimal_places=2)
+
+    class Meta:
+        ordering = ('price',)