1 files changed, 17 insertions, 3 deletions
diff --git a/app/expenses/models.py b/app/expenses/models.py
index ef9352f..bca23bf 100644
--- a/app/expenses/models.py
+++ b/app/expenses/models.py
@@ -1,7 +1,6 @@
 from django.db import models
+from django.core.urlresolvers import reverse
 from django.utils import timezone
-from django.template.defaultfilters import slugify
-import datetime
 
 CATS = (
     ('1', "Travco"),
@@ -13,12 +12,28 @@ CATS = (
     ('7', "Misc"),
 )
 
+
+class Trip(models.Model):
+    name = models.CharField(max_length=200)
+    slug = models.SlugField()
+    start = models.DateTimeField(default=timezone.now, blank=True, null=True)
+    end = models.DateTimeField(default=timezone.now, blank=True, null=True)
+    dek = models.TextField(null=True, blank=True)
+
+    def __str__(self):
+        return self.name
+
+    def get_absolute_url(self):
+        return reverse("expenses:list_expense", kwargs={"slug": self.slug})
+
+
 class Expense(models.Model):
     name = models.CharField(max_length=200)
     amount = models.DecimalField(max_digits=8, decimal_places=2)
     date = models.DateTimeField(default=timezone.now)
     notes = models.TextField(null=True, blank=True)
     category = models.CharField(max_length=2, choices=CATS, default=1)
+    trip = models.ForeignKey(Trip, null=True)
 
     class Meta:
         ordering = ('-date',)
@@ -28,4 +43,3 @@ class Expense(models.Model):
 
     def date_month(self):
         return self.date.strftime("%b %Y")
-