From 669cf1cb7041c144ec5bc5dd1eff9dc9fc2cbfe5 Mon Sep 17 00:00:00 2001
From: Daniel Quinn <code@danielquinn.org>
Date: Sat, 23 Jan 2016 04:40:35 +0000
Subject: [PATCH] Add labels (#9)

---
 src/documents/admin.py                        | 19 ++++++++--
 .../migrations/0006_auto_20160123_0430.py     | 37 +++++++++++++++++++
 src/documents/models.py                       | 33 +++++++++++++++--
 3 files changed, 82 insertions(+), 7 deletions(-)
 create mode 100644 src/documents/migrations/0006_auto_20160123_0430.py

diff --git a/src/documents/admin.py b/src/documents/admin.py
index b2944426b..613e7b292 100644
--- a/src/documents/admin.py
+++ b/src/documents/admin.py
@@ -2,15 +2,15 @@
 from django.core.urlresolvers import reverse
 from django.templatetags.static import static
 
-from .models import Sender, Document
+from .models import Sender, Tag, Document
 
 
 class DocumentAdmin(admin.ModelAdmin):
 
     search_fields = ("sender__name", "title", "content",)
-    list_display = ("edit", "created", "sender", "title", "pdf")
-    list_filter = ("created", "sender")
-    save_on_top = True
+    list_display = ("edit", "created", "sender", "title", "tags_", "pdf")
+    list_filter = ("created", "tags", "sender")
+    list_per_page = 25
 
     def edit(self, obj):
         return '<img src="{}" width="22" height="22" alt="Edit icon" />'.format(
@@ -26,5 +26,16 @@ def pdf(self, obj):
                 )
     pdf.allow_tags = True
 
+    def tags_(self, obj):
+        r = ""
+        for tag in obj.tags.all():
+            r += '<span style="padding: 0 0.5em; background-color: {}; color: #ffffff; border-radius: 0.2em; margin: 1px; display: inline-block;">{}</span>'.format(
+                tag.get_colour_display(),
+                tag.slug
+            )
+        return r
+    tags_.allow_tags = True
+
 admin.site.register(Sender)
+admin.site.register(Tag)
 admin.site.register(Document, DocumentAdmin)
diff --git a/src/documents/migrations/0006_auto_20160123_0430.py b/src/documents/migrations/0006_auto_20160123_0430.py
new file mode 100644
index 000000000..b8d9979db
--- /dev/null
+++ b/src/documents/migrations/0006_auto_20160123_0430.py
@@ -0,0 +1,37 @@
+# -*- coding: utf-8 -*-
+# Generated by Django 1.9 on 2016-01-23 04:30
+from __future__ import unicode_literals
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('documents', '0005_auto_20160123_0313'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='Tag',
+            fields=[
+                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('name', models.CharField(max_length=128, unique=True)),
+                ('slug', models.SlugField(blank=True)),
+                ('colour', models.PositiveIntegerField(choices=[(1, '#a6cee3'), (2, '#1f78b4'), (3, '#b2df8a'), (4, '#33a02c'), (5, '#fb9a99'), (6, '#e31a1c'), (7, '#fdbf6f'), (8, '#ff7f00'), (9, '#cab2d6'), (10, '#6a3d9a'), (11, '#ffff99'), (12, '#b15928'), (13, '#000000'), (14, '#cccccc')], default=1)),
+            ],
+            options={
+                'abstract': False,
+            },
+        ),
+        migrations.AlterField(
+            model_name='sender',
+            name='slug',
+            field=models.SlugField(blank=True),
+        ),
+        migrations.AddField(
+            model_name='document',
+            name='tags',
+            field=models.ManyToManyField(related_name='documents', to='documents.Tag'),
+        ),
+    ]
diff --git a/src/documents/models.py b/src/documents/models.py
index c8c7b08f6..88faa079e 100644
--- a/src/documents/models.py
+++ b/src/documents/models.py
@@ -6,13 +6,13 @@
 from django.utils import timezone
 
 
-class Sender(models.Model):
+class SluggedModel(models.Model):
 
     name = models.CharField(max_length=128, unique=True)
-    slug = models.SlugField()
+    slug = models.SlugField(blank=True)
 
     class Meta(object):
-        ordering = ("name",)
+        abstract = True
 
     def save(self, *args, **kwargs):
         if not self.slug:
@@ -23,12 +23,39 @@ def __str__(self):
         return self.name
 
 
+class Sender(SluggedModel):
+
+    class Meta(object):
+        ordering = ("name",)
+
+
+class Tag(SluggedModel):
+    
+    COLOURS = (
+        (1, "#a6cee3"),
+        (2, "#1f78b4"),
+        (3, "#b2df8a"),
+        (4, "#33a02c"),
+        (5, "#fb9a99"),
+        (6, "#e31a1c"),
+        (7, "#fdbf6f"),
+        (8, "#ff7f00"),
+        (9, "#cab2d6"),
+        (10, "#6a3d9a"),
+        (11, "#b15928"),
+        (12, "#000000"),
+        (13, "#cccccc")
+    )
+    colour = models.PositiveIntegerField(choices=COLOURS, default=1)
+
+
 class Document(models.Model):
 
     sender = models.ForeignKey(
         Sender, blank=True, null=True, related_name="documents")
     title = models.CharField(max_length=128, blank=True, db_index=True)
     content = models.TextField(db_index=True)
+    tags = models.ManyToManyField(Tag, related_name="documents")
     created = models.DateTimeField(default=timezone.now, editable=False)
     modified = models.DateTimeField(auto_now=True, editable=False)