Added support for a custom slugify.

taggit/models.py

@@ -2,7 +2,7 @@ import django
 from django.contrib.contenttypes.models import ContentType
 from django.contrib.contenttypes.generic import GenericForeignKey
 from django.db import models, IntegrityError, transaction
-from django.template.defaultfilters import slugify
+from django.template.defaultfilters import slugify as default_slugify
 from django.utils.translation import ugettext_lazy as _, ugettext
 
 
@@ -18,7 +18,7 @@ class TagBase(models.Model):
 
     def save(self, *args, **kwargs):
         if not self.pk and not self.slug:
-            self.slug = slug = slugify(self.name)
+            self.slug = self.slugify(self.name)
             if django.VERSION >= (1, 2):
                 from django.db import router
                 using = kwargs.get("using") or router.db_for_write(
@@ -39,11 +39,17 @@ class TagBase(models.Model):
                     return res
                 except IntegrityError:
                     transaction.savepoint_rollback(sid, **trans_kwargs)
-                    i += 1
-                    self.slug = "%s_%d" % (slug, i)
+                    self.slug = self.slugify(self.name, i)
         else:
             return super(TagBase, self).save(*args, **kwargs)
 
+    def slugify(self, tag, i=None):
+        slug = default_slugify(tag)
+        if i is not None:
+            slug += "_%d" % i
+        return slug
+
+
 class Tag(TagBase):
     class Meta:
         verbose_name = _("Tag")
@@ -137,4 +143,3 @@ class TaggedItem(GenericTaggedItemBase, TaggedItemBase):
     class Meta:
         verbose_name = _("Tagged Item")
         verbose_name_plural = _("Tagged Items")
-

taggit/tests/models.py

 from django.db import models
 
 from taggit.managers import TaggableManager
-from taggit.models import TaggedItemBase, GenericTaggedItemBase, TagBase
+from taggit.models import (TaggedItemBase, GenericTaggedItemBase, TaggedItem,
+    TagBase, Tag)
 
 
 class Food(models.Model):
@@ -115,3 +116,27 @@ class Photo(Media):
 
 class Movie(Media):
     pass
+
+
+class ArticleTag(Tag):
+    class Meta:
+        proxy = True
+
+    def slugify(self, tag, i=None):
+        slug = "category-%s" % tag
+        if i is not None:
+            slug += "-%d" % i
+        return slug
+
+class ArticleTaggedItem(TaggedItem):
+    class Meta:
+        proxy = True
+
+    # Basically we want to overide the tag ForeignKey to point at the proxy
+    # inherited ArticleTag so we can get the right slugify, unfortunately I
+    # can't seem to figure out how to do this, so we're on hold ATM.
+
+class Article(models.Model):
+    title = models.CharField(max_length=100)
+
+    tags = TaggableManager(through=ArticleTaggedItem)

taggit/tests/tests.py

@@ -12,13 +12,13 @@ from taggit.tests.forms import (FoodForm, DirectFoodForm, CustomPKFoodForm,
 from taggit.tests.models import (Food, Pet, HousePet, DirectFood, DirectPet,
     DirectHousePet, TaggedPet, CustomPKFood, CustomPKPet, CustomPKHousePet,
     TaggedCustomPKPet, OfficialFood, OfficialPet, OfficialHousePet,
-    OfficialThroughModel, OfficialTag, Photo, Movie)
+    OfficialThroughModel, OfficialTag, Photo, Movie, Article)
 from taggit.utils import parse_tags, edit_string_for_tags
 
 
 class BaseTaggingTest(object):
-    def assert_tags_equal(self, qs, tags, sort=True):
-        got = map(lambda tag: tag.name, qs)
+    def assert_tags_equal(self, qs, tags, sort=True, attr="name"):
+        got = map(lambda tag: getattr(tag, attr), qs)
         if sort:
             got.sort()
             tags.sort()
@@ -61,6 +61,15 @@ class TagModelTestCase(BaseTaggingTransactionTestCase):
         yummy = self.tag_model.objects.create(name="yummy")
         apple.tags.add(yummy)
 
+    def test_slugify(self):
+        a = Article.objects.create(title="django-taggit 1.0 Released")
+        a.tags.add("awesome", "release", "AWESOME")
+        self.assert_tags_equal(a.tags.all(), [
+            "category-awesome",
+            "category-release",
+            "category-awesome-1"
+        ], attr="slug")
+
 class TagModelDirectTestCase(TagModelTestCase):
     food_model = DirectFood
     tag_model = Tag