tickets: Add a (virtual) Booking.price property.

[matthijs/projects/dorestad-bookings.git] / tickets / models.py

diff --git a/tickets/models.py b/tickets/models.py
index 3d174dd3fa4143fb9a89a73b5357083ba2a0472a..5c56b0aa82b0d99f2ef88e79215c77f43270e613 100644 (file)
--- a/tickets/models.py
+++ b/tickets/models.py
@@ -1,3 +1,5 @@
+# coding=utf-8
+
 from django.db import models
 
 # Create your models here.
@@ -9,6 +11,8 @@ SHOW_CHOICES = [ ("vrijdag", "vrijdag 17 december, 20:00")
 
 TICKETS_CHOICES = [ (n, n) for n in range(1, 6)]
 
+TICKET_PRICE=10
+
 class Booking(models.Model):
     created = models.DateTimeField(auto_now_add=True, editable=False, verbose_name="Datum")
     name = models.CharField(max_length=100, verbose_name="Naam")
@@ -17,3 +21,8 @@ class Booking(models.Model):
     email = models.EmailField(verbose_name="Email adres")
     tickets = models.IntegerField(choices=TICKETS_CHOICES, verbose_name="Aantal kaarten")
     show = models.CharField(max_length=50, choices=SHOW_CHOICES, verbose_name="Voorstelling")
+    payment = models.DateTimeField(blank=True, null=True, verbose_name="Betaling vewerkt op")
+
+    def _get_price(self):
+        return self.tickets * TICKET_PRICE
+    price = property(_get_price)