From 921d06a8e01e2613abdd07ce4d1a88d4957fe0d1 Mon Sep 17 00:00:00 2001
From: Tom Willemsen
Date: Thu, 22 Mar 2012 00:04:01 +0100
Subject: Add Category model

Each feed can belong to multiple categories. A category is only a slug.
---
 aggregator/models.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

(limited to 'aggregator/models.py')

diff --git a/aggregator/models.py b/aggregator/models.py
index b5c4f3c..4936821 100644
--- a/aggregator/models.py
+++ b/aggregator/models.py
@@ -1,5 +1,11 @@
 from django.db import models
 
+class Category(models.Model):
+    name = models.SlugField()
+
+    def __unicode__(self):
+        return self.name.capitalize()
+
 class Feed(models.Model):
     name = models.CharField(max_length=200)
     base_url = models.URLField(max_length=255)
@@ -11,6 +17,7 @@ class Feed(models.Model):
     uses_title = models.BooleanField(default=False)
     br2nl = models.BooleanField(default=False)
     with_markdown = models.BooleanField(default=False)
+    categories = models.ManyToManyField(Category)
 
     def get_profile_url(self):
         return self.base_url + self.profile_url
@@ -21,6 +28,9 @@ class Feed(models.Model):
     def get_favicon_url(self):
         return self.base_url + 'favicon.' + self.favicon_ext
 
+    def __unicode__(self):
+        return self.name
+
     class Meta:
         ordering = [ '-updated' ]
 
-- 
cgit v1.2.3-54-g00ecf