tickets: Remove postal address fields from Booking.

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

diff --git a/tickets/models.py b/tickets/models.py
index f022e6f771afc3068c7396cb3d0b08d2aa9f73b6..ab80b5f807a1fe9df0064be7c3b9adf90d603b5c 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,12 +11,23 @@ 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)
+
+    def __unicode__(self):
+        return u'#%s: %s - %s - %s kaart%s (€ %.2d)' % (
+            self.pk, self.name,
+            self.get_show_display(),
+            self.tickets, "en" if self.tickets != 1 else "",
+            self.price)