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 2a7c3f783a39abd62a16ac23cf8c5f98e36ee67f..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,10 +11,18 @@ 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")
     address1 = models.CharField(max_length=50, verbose_name="Adres")
     address2 = models.CharField(max_length=50, verbose_name="")
     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)