Serve partner audiobooks with isbns

[wolnelektury.git] / src / partners / models.py

diff --git a/src/partners/models.py b/src/partners/models.py
index 32e8a1b..583c34c 100644 (file)
--- 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',)