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

diff --git a/src/partners/models.py b/src/partners/models.py
index 9e58e2a88..583c34c79 100644
--- a/src/partners/models.py
+++ b/src/partners/models.py
@@ -16,11 +16,28 @@ 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)
     min_pages = models.IntegerField(null=True, blank=True)
-    price = models.IntegerField()
+    price = models.DecimalField(max_digits=10, decimal_places=2)
+
+    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',)